Skip to content

GitLab

Commit 1a9e39eb authored by dengjb's avatar dengjb
Browse files

Update finetune.py

parent 768d526a
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
finetune.py
View file @ 1a9e39eb
...@@ -130,8 +130,8 @@ def main(args): ...@@ -130,8 +130,8 @@ def main(args):
print("Saving the last checkpoint of the model") print("Saving the last checkpoint of the model")
model.save_pretrained(os.path.join(args.output_dir, "final_checkpoint/")) model.save_pretrained(os.path.join(args.output_dir, "final_checkpoint/"))
if args.push_to_hub: #if args.push_to_hub:
trainer.push_to_hub("Upload model") # trainer.push_to_hub("Upload model")
print("Training Done! 💥") print("Training Done! 💥")
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment