diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py
index a50c8be2bb..90cbae2131 100644
--- a/unsloth/models/_utils.py
+++ b/unsloth/models/_utils.py
@@ -1178,8 +1178,8 @@ 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