From a62651ce699ae663ff39d91285f80ef1c73e7010 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Mon, 25 Nov 2024 21:59:40 -0800 Subject: [PATCH] Update _utils.py --- unsloth/models/_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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