dkebudi commited on
Commit
bd3d196
·
verified ·
1 Parent(s): ccca96c

syntax error

Browse files
train_dreambooth_lora_sdxl_advanced.py CHANGED
@@ -1322,7 +1322,7 @@ def main(args):
1322
 
1323
  # Generate class images if prior preservation is enabled.
1324
  if args.with_prior_preservation:
1325
- if arg.class_data_dir == None:
1326
  class_folder = str(uuid.uuid4())
1327
  args.class_data_dir = os.path.join("ariadne", class_folder)
1328
  class_images_dir = Path(args.class_data_dir)
 
1322
 
1323
  # Generate class images if prior preservation is enabled.
1324
  if args.with_prior_preservation:
1325
+ if args.class_data_dir is None:
1326
  class_folder = str(uuid.uuid4())
1327
  args.class_data_dir = os.path.join("ariadne", class_folder)
1328
  class_images_dir = Path(args.class_data_dir)