inoid commited on
Commit
4d863e3
1 Parent(s): cf94ee5

Fix dataset reference error

Browse files
Files changed (1) hide show
  1. spanish_medica_llm.py +1 -1
spanish_medica_llm.py CHANGED
@@ -525,7 +525,7 @@ def configAndRunTraining(basemodel, dataset, eval_dataset, tokenizer):
525
 
526
  trainer = Trainer(
527
  model= basemodel,
528
- train_dataset = dataset['train'],
529
  eval_dataset = eval_dataset,
530
  args = training_args,
531
  data_collator = data_collator_pretrain
 
525
 
526
  trainer = Trainer(
527
  model= basemodel,
528
+ train_dataset = dataset,
529
  eval_dataset = eval_dataset,
530
  args = training_args,
531
  data_collator = data_collator_pretrain