Update finetune.py

finetune.py

@@ -17,6 +17,7 @@ from peft import (
 
 
 # Parameters
+HF_TOKEN = os.environ.get("TRL_TOKEN", None)
 MICRO_BATCH_SIZE = 16
 BATCH_SIZE = 16
 size = "7b"
@@ -152,6 +153,7 @@ 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,
+        use_auth_token=HF_TOKEN,
         push_to_hub=True,
         push_to_hub_model_id="gustavoaq/llama_ft"
     ),