hysts HF staff commited on
Commit
d049258
1 Parent(s): 21bd5b5

Update default model name

Browse files
Files changed (1) hide show
  1. 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
- output_model_name = datetime.datetime.now().strftime(
87
- '%Y-%m-%d-%H-%M-%S')
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