Update default model name
Browse files- trainer.py +2 -2
trainer.py
CHANGED
@@ -83,8 +83,8 @@ class Trainer:
|
|
83 |
resolution = int(resolution_s)
|
84 |
|
85 |
if not output_model_name:
|
86 |
-
|
87 |
-
|
88 |
output_model_name = slugify.slugify(output_model_name)
|
89 |
|
90 |
repo_dir = pathlib.Path(__file__).parent
|
|
|
83 |
resolution = int(resolution_s)
|
84 |
|
85 |
if not output_model_name:
|
86 |
+
timestamp = datetime.datetime.now().strftime('%Y-%m-%d-%H-%M-%S')
|
87 |
+
output_model_name = f'lora-dreambooth-{timestamp}'
|
88 |
output_model_name = slugify.slugify(output_model_name)
|
89 |
|
90 |
repo_dir = pathlib.Path(__file__).parent
|