mkshing commited on
Commit
d309899
1 Parent(s): 47390c8
Files changed (1) hide show
  1. train_svdiff.py +2 -2
train_svdiff.py CHANGED
@@ -73,7 +73,7 @@ inference: true
73
  """
74
  model_card = f"""
75
  # SVDiff-pytorch - {repo_id}
76
- These are SVDiff weights for {base_model}. The weights were trained on {prompt} using [DreamBooth](https://dreambooth.github.io/).
77
  """
78
  with open(os.path.join(repo_folder, "README.md"), "w") as f:
79
  f.write(yaml + model_card)
@@ -1032,7 +1032,7 @@ def main(args):
1032
  log_validation(text_encoder, tokenizer, unet, vae, args, accelerator, weight_dtype, epoch)
1033
 
1034
  accelerator.wait_for_everyone()
1035
- # put the last checkpoint to output-dir
1036
  save_weights(global_step, save_path=args.output_dir)
1037
  if accelerator.is_main_process:
1038
  if args.push_to_hub:
 
73
  """
74
  model_card = f"""
75
  # SVDiff-pytorch - {repo_id}
76
+ These are SVDiff weights for {base_model}. The weights were trained on {prompt}.
77
  """
78
  with open(os.path.join(repo_folder, "README.md"), "w") as f:
79
  f.write(yaml + model_card)
 
1032
  log_validation(text_encoder, tokenizer, unet, vae, args, accelerator, weight_dtype, epoch)
1033
 
1034
  accelerator.wait_for_everyone()
1035
+ # put the latest checkpoint to output-dir
1036
  save_weights(global_step, save_path=args.output_dir)
1037
  if accelerator.is_main_process:
1038
  if args.push_to_hub: