d'oh

fixie-ai · Oct 18, 2024 · f86258c · f86258c
1 parent 8d3ac96
commit f86258c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ultravox/training/train.py b/ultravox/training/train.py
@@ -256,7 +256,7 @@ def train(args: config_base.TrainConfig):
     trainer = transformers.Seq2SeqTrainer(
         model,
         train_dataset=train_dataset,
-        eval_dataset=val_dataset,
+        eval_dataset=val_datasets,
         data_collator=data_collator,
         tokenizer=text_tokenizer,
         args=transformers.Seq2SeqTrainingArguments(