fix (#1575)

modelscope · Aug 2, 2024 · 73a1b44 · 73a1b44
1 parent bb017a8
commit 73a1b44
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/swift/trainers/kto_trainer.py b/swift/trainers/kto_trainer.py
@@ -87,7 +87,7 @@ def __init__(self, *args, template: Template, test_oom_error=False, **kwargs):
         super().__init__(*args, **kwargs)
         train_ds_info = self.stat_dataset(self.train_dataset)
         if self.eval_dataset is not None:
-            val_ds_info = self.stat_dataset(self.eval_dataset, self.is_encoder_decoder)
+            val_ds_info = self.stat_dataset(self.eval_dataset)
             self.dataset_info = {'train_dataset': train_ds_info, 'val_dataset': val_ds_info}
         else:
             self.dataset_info = {'train_dataset': train_ds_info}