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,
|