From 20e182b6068614346cbb38ad26d5933cf802ffef Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Mon, 25 Nov 2024 23:35:36 -0800
Subject: [PATCH] Update _utils.py

---
 unsloth/models/_utils.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py
index 5f0b4e0552..45d32c8bc8 100644
--- a/unsloth/models/_utils.py
+++ b/unsloth/models/_utils.py
@@ -1175,7 +1175,6 @@ LOGITS_ERROR_STRING = \
     "os.environ['UNSLOTH_RETURN_LOGITS'] = '1'\n"\
     "... trainer.train() ..."
 
-def  warn_logits_error(*args, **kwargs): warnings.warn(LOGITS_ERROR_STRING)
 class EmptyLogits:
     def __init__(self): return
     def raise_logits_error(*args, **kwargs): raise NotImplementedError(LOGITS_ERROR_STRING)
@@ -1183,8 +1182,8 @@ class EmptyLogits:
     def raise_getattr_error(self, attr):
         if attr == "to": return self.return_none
         return self.raise_logits_error
-    __getitem__ = self.raise_logits_error
-    __getattr__ = self.raise_getattr_error
+    __getitem__ = raise_logits_error
+    __getattr__ = raise_getattr_error
     def __repr__(self): return LOGITS_ERROR_STRING
     def __str__ (self): return LOGITS_ERROR_STRING
 pass