Update finetune.py

1a9e39eb · dengjb · 768d526a · 1a9e39eb
Commit 1a9e39eb authored Jun 13, 2024 by dengjb
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

finetune.py finetune.py +2 -2

No files found.
--- a/finetune.py
+++ b/finetune.py
@@ -130,8 +130,8 @@ def main(args):

    print("Saving the last checkpoint of the model")
    model.save_pretrained(os.path.join(args.output_dir, "final_checkpoint/"))
-    if args.push_to_hub:
-        trainer.push_to_hub("Upload model")
+    #if args.push_to_hub:
+    #    trainer.push_to_hub("Upload model")
    print("Training Done! 💥")