gustavoaq commited on
Commit
25f93cb
·
1 Parent(s): 6736d75

Update finetune.py

Browse files
Files changed (1) hide show
  1. finetune.py +1 -1
finetune.py CHANGED
@@ -153,7 +153,7 @@ trainer = transformers.Trainer(
153
  save_total_limit=100,
154
  load_best_model_at_end=True if VAL_SET_SIZE > 0 else False,
155
  ddp_find_unused_parameters=False if ddp else None,
156
- use_auth_token=HF_TOKEN,
157
  push_to_hub=True,
158
  push_to_hub_model_id="gustavoaq/llama_ft"
159
  ),
 
153
  save_total_limit=100,
154
  load_best_model_at_end=True if VAL_SET_SIZE > 0 else False,
155
  ddp_find_unused_parameters=False if ddp else None,
156
+ push_to_hub_token="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",
157
  push_to_hub=True,
158
  push_to_hub_model_id="gustavoaq/llama_ft"
159
  ),