Spaces:
Runtime error
Runtime error
Fix bug when TrainingArguments and Trainer
Browse files- 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,
|