From 73a1b44ae6b77eeb4c107c1f81e3d0e8cea4a030 Mon Sep 17 00:00:00 2001 From: tastelikefeet <58414341+tastelikefeet@users.noreply.github.com> Date: Fri, 2 Aug 2024 00:21:12 +0800 Subject: [PATCH] fix (#1575) --- swift/trainers/kto_trainer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/swift/trainers/kto_trainer.py b/swift/trainers/kto_trainer.py index acdfd4d12..53ade6706 100644 --- 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}