inoid commited on
Commit
97003da
1 Parent(s): 29d78f2

Fix bug when TrainingArguments and Trainer

Browse files
Files changed (1) hide show
  1. spanish_medica_llm.py +1 -0
spanish_medica_llm.py CHANGED
@@ -497,6 +497,7 @@ def configAndRunTraining(basemodel, dataset, eval_dataset, tokenizer):
497
  else:
498
  tokenizer.pad_token = tokenizer.eos_token
499
  data_collator_pretrain = DataCollatorForLanguageModeling(tokenizer, mlm = False)
 
500
  training_args = TrainingArguments(
501
  output_dir=output_dir,
502
  push_to_hub = True,
 
497
  else:
498
  tokenizer.pad_token = tokenizer.eos_token
499
  data_collator_pretrain = DataCollatorForLanguageModeling(tokenizer, mlm = False)
500
+
501
  training_args = TrainingArguments(
502
  output_dir=output_dir,
503
  push_to_hub = True,