Update cross_entropy_loss.py

This commit is contained in:
Daniel Han 2024-09-24 23:52:54 -07:00
commit 805d343b7f

View file

@ -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