From 0b431c3317110b02a1c131dd78c613badca3a5c2 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Mon, 25 Nov 2024 22:33:18 -0800 Subject: [PATCH] Update _utils.py --- unsloth/models/_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py index 5024204f20..acf411b6c9 100644 --- a/unsloth/models/_utils.py +++ b/unsloth/models/_utils.py @@ -1192,7 +1192,7 @@ for j, function in enumerate(functions): try: exec(f"EMPTY_LOGITS.{function} = warn_logits_error", globals(), locals()) except: continue else: - exec(f"def raise_{j}: print({function})", globals(), locals()) + exec(f"def raise_{j}(*args, **kwargs): print({function})", globals(), locals()) try: exec(f"EMPTY_LOGITS.{function} = raise_{j}", globals(), locals()) except: continue pass