hotfix for dpo trainer

addresses #914
huggingface · Oct 27, 2023 · 8a272cc · 8a272cc
1 parent 02f5c1d
commit 8a272cc
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/trl/trainer/dpo_trainer.py b/trl/trainer/dpo_trainer.py
@@ -638,6 +638,7 @@ def evaluation_loop(
                     )
                 }
             )
+            self.state.log_history.pop()
 
         # Base evaluation
         initial_output = super().evaluation_loop(