Update cross_entropy_loss.py

2024-09-25 00:07:49 -07:00 · 2024-09-25 00:07:49 -07:00 · b74a86a2bd
commit b74a86a2bd
parent 06e06f922d
1 changed files with 1 additions and 0 deletions
--- a/unsloth/kernels/cross_entropy_loss.py
+++ b/unsloth/kernels/cross_entropy_loss.py
@ -443,6 +443,7 @@ 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