From 7f9ecd592f3e9add62c5a3dfd1b448128836caef Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Wed, 25 Sep 2024 00:11:57 -0700 Subject: [PATCH] Update cross_entropy_loss.py --- unsloth/kernels/cross_entropy_loss.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/unsloth/kernels/cross_entropy_loss.py b/unsloth/kernels/cross_entropy_loss.py index 0034448be5..1fec5d7a85 100644 --- a/unsloth/kernels/cross_entropy_loss.py +++ b/unsloth/kernels/cross_entropy_loss.py @@ -443,9 +443,8 @@ function = [function[0]] + [" "*4 + x for x in function[1:]] function = "\n".join(function) function = f"class Unsloth_LlamaForCausalLM(LlamaForCausalLM):\n"\ f" {function}\n" -print(function) exec(function, globals()) -del function, replacement +del function, replacement, inspect, re def patch_llama_for_causal_lm():