From 27f740375fc89dfdd984d89a03a1fbf432fcd511 Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Mon, 25 Nov 2024 22:18:33 -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 daf0bdf44d..82e8f97eb3 100644
--- a/unsloth/models/_utils.py
+++ b/unsloth/models/_utils.py
@@ -1187,7 +1187,7 @@ pass
 EMPTY_LOGITS = EmptyLogits()
 functions = dir(torch.Tensor)
 for j, function in enumerate(functions):
-    exec(f"def raise_{j}: print({j})", globals(), locals())
+    exec(f"def raise_{j}(*args, **kwargs): print({j})", globals(), locals())
     try: exec(f"EMPTY_LOGITS.{function} = raise_{j}", globals(), locals())
     except: continue
 pass