Update finetune.py

finetune.py

@@ -17,7 +17,7 @@ from peft import (
 
 HF_TOKEN = os.environ.get("TRL_TOKEN", None)
 if HF_TOKEN:
-
+    print(HF_TOKEN)
     repo = Repository(
         local_dir="./checkpoints/", clone_from="gustavoaq/llama_ft", use_auth_token=HF_TOKEN, repo_type="models"
     )
@@ -159,9 +159,6 @@ trainer = transformers.Trainer(
         save_total_limit=100,
         load_best_model_at_end=True if VAL_SET_SIZE > 0 else False,
         ddp_find_unused_parameters=False if ddp else None,
-        push_to_hub=True,
-        push_to_hub_model_id='llama_ft'
-        
     ),
     data_collator=transformers.DataCollatorForLanguageModeling(tokenizer, mlm=False),
 )
@@ -175,6 +172,6 @@ print("Training")
 trainer.train()
 
 model.save_pretrained(OUTPUT_DIR)
-trainer.push_to_hub()
+repo.push_to_hub(OUTPUT_DIR, commit_message="Ft model")