Spaces:
Runtime error
Runtime error
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
|
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)
|