From 7333f819e1536842c57268cf565eba346750b365 Mon Sep 17 00:00:00 2001
From: Wing Lian <wing@axolotl.ai>
Date: Thu, 10 Apr 2025 04:50:58 -0400
Subject: [PATCH] guard on model.eval

---
 src/transformers/trainer.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/transformers/trainer.py b/src/transformers/trainer.py
index 5b46763ecc5e..d6e8aab11eda 100755
--- a/src/transformers/trainer.py
+++ b/src/transformers/trainer.py
@@ -4303,7 +4303,8 @@ def evaluation_loop(
             logger.info("  Num examples: Unknown")
         logger.info(f"  Batch size = {batch_size}")
 
-        model.eval()
+        if hasattr(model, "eval") and callable(model.eval):
+            model.eval()
         if hasattr(self.optimizer, "eval") and callable(self.optimizer.eval):
             self.optimizer.eval()