Skip to content

Commit ddaebf8

Browse files
authored
fix dpo eval override to call grandparent instead of the broken super (axolotl-ai-cloud#2628) [skip ci]
1 parent 6797430 commit ddaebf8

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/axolotl/core/trainers/dpo/trainer.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -247,7 +247,9 @@ def evaluation_loop(
247247
)
248248

249249
# Base evaluation
250-
initial_output = super().evaluation_loop(
250+
initial_output = super( # pylint: disable=bad-super-call
251+
DPOTrainer, self
252+
).evaluation_loop(
251253
dataloader,
252254
description,
253255
prediction_loss_only,

0 commit comments

Comments
 (0)