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)
|