From 92e4d089bf3568c723dd56acff215f1baa1ca5c3 Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Mon, 25 Nov 2024 22:20:16 -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 82e8f97eb3..a4ded5f55b 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}(*args, **kwargs): print({j})", globals(), locals())
+    exec(f"def raise_{j}(*args, **kwargs): print(function, {j})", globals(), locals())
     try: exec(f"EMPTY_LOGITS.{function} = raise_{j}", globals(), locals())
     except: continue
 pass