Spaces:
Runtime error
Runtime error
mkshing
commited on
Commit
•
d309899
1
Parent(s):
47390c8
minor fix
Browse files- 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}
|
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
|
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:
|