From 805d343b7f9af17b24aaeede108145fe52ca34d0 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Tue, 24 Sep 2024 23:52:54 -0700 Subject: [PATCH] Update cross_entropy_loss.py --- unsloth/kernels/cross_entropy_loss.py | 1 + 1 file changed, 1 insertion(+) diff --git a/unsloth/kernels/cross_entropy_loss.py b/unsloth/kernels/cross_entropy_loss.py index 74ee4ee66a..59d0f3b694 100644 --- a/unsloth/kernels/cross_entropy_loss.py +++ b/unsloth/kernels/cross_entropy_loss.py @@ -443,6 +443,7 @@ def patch_llama_for_causal_lm(): patched_function = f"class Unsloth_LlamaForCausalLM(LlamaForCausalLM):\n"\ f" {function}\n" + print(patched_function) exec(patched_function) transformers.models.llama.modeling_llama.LlamaForCausalLM = Unsloth_LlamaForCausalLM return