From b72cdbd5dd7d2cc92007cbcd2fe96805e60dab18 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Mon, 25 Nov 2024 22:56:24 -0800 Subject: [PATCH] Update _utils.py --- unsloth/models/_utils.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py index fbb68b38f4..7259923f7a 100644 --- a/unsloth/models/_utils.py +++ b/unsloth/models/_utils.py @@ -1187,11 +1187,7 @@ pass EMPTY_LOGITS = EmptyLogits() functions = dir(torch.Tensor) for j, function in enumerate(functions): - # if function.endswith("_") and len(function) > 2 and function[-2] != "_": - # print() - # try: exec(f"EMPTY_LOGITS.{function} = warn_logits_error", globals(), locals()) - # except: continue - # else: + if function == "to": continue exec(f"def raise_{j}(*args, **kwargs): print('{function}')", globals(), locals()) try: exec(f"EMPTY_LOGITS.{function} = raise_{j}", globals(), locals()) except: continue