From 0487293f4cd666f515caa41d742a62ce0b2ba25c Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Mon, 25 Nov 2024 22:03:08 -0800
Subject: [PATCH] Update _utils.py

---
 unsloth/models/_utils.py | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py
index 90cbae2131..242e30675f 100644
--- a/unsloth/models/_utils.py
+++ b/unsloth/models/_utils.py
@@ -1178,14 +1178,14 @@ LOGITS_ERROR_STRING = \
 def raise_logits_error(*args, **kwargs): raise NotImplementedError(LOGITS_ERROR_STRING)
 class EmptyLogits:
     def __init__(self): return
-    # __getitem__ = raise_logits_error
-    # __getattr__ = raise_logits_error
+    __getitem__ = raise_logits_error
+    __getattr__ = raise_logits_error
     def __repr__(self): return LOGITS_ERROR_STRING
     def __str__ (self): return LOGITS_ERROR_STRING
 pass
 EMPTY_LOGITS = EmptyLogits()
 functions = dir(torch.Tensor)
-for function in functions:
-    try: exec(f"EMPTY_LOGITS.{function} = raise_logits_error", globals(), locals())
-    except: continue
-pass
+# for function in functions:
+#     try: exec(f"EMPTY_LOGITS.{function} = raise_logits_error", globals(), locals())
+#     except: continue
+# pass