From fde2079debaf08e82541ecf2d715f01669724889 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Mon, 25 Nov 2024 23:03:35 -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 7259923f7a..05f6520ef0 100644 --- a/unsloth/models/_utils.py +++ b/unsloth/models/_utils.py @@ -1179,8 +1179,8 @@ def raise_logits_error(*args, **kwargs): raise NotImplementedError(LOGITS_ERROR_ def warn_logits_error(*args, **kwargs): warnings.warn(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