gustavoaq commited on
Commit
0f2d082
·
1 Parent(s): ae1f2b4

Update finetune.py

Browse files
Files changed (1) hide show
  1. finetune.py +0 -3
finetune.py CHANGED
@@ -153,9 +153,6 @@ 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
- push_to_hub_token="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",
157
- push_to_hub=True,
158
- push_to_hub_model_id="llama_ft"
159
  ),
160
  data_collator=transformers.DataCollatorForLanguageModeling(tokenizer, mlm=False),
161
  )
 
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
  ),
157
  data_collator=transformers.DataCollatorForLanguageModeling(tokenizer, mlm=False),
158
  )