Skip to content

Commit 73a1b44

Browse files
tastelikefeetJintao-Huang
authored andcommitted
fix (#1575)
1 parent bb017a8 commit 73a1b44

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

swift/trainers/kto_trainer.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -87,7 +87,7 @@ def __init__(self, *args, template: Template, test_oom_error=False, **kwargs):
8787
super().__init__(*args, **kwargs)
8888
train_ds_info = self.stat_dataset(self.train_dataset)
8989
if self.eval_dataset is not None:
90-
val_ds_info = self.stat_dataset(self.eval_dataset, self.is_encoder_decoder)
90+
val_ds_info = self.stat_dataset(self.eval_dataset)
9191
self.dataset_info = {'train_dataset': train_ds_info, 'val_dataset': val_ds_info}
9292
else:
9393
self.dataset_info = {'train_dataset': train_ds_info}

0 commit comments

Comments
 (0)