Update cross_entropy_loss.py
This commit is contained in:
parent
b25b46977f
commit
805d343b7f
1 changed files with 1 additions and 0 deletions
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue