ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update cross_entropy_loss.py

Browse source
This commit is contained in:
Daniel Han 2024-11-03 14:09:47 -08:00
commit 82db087cab
1 changed files with 2 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

5
unsloth/kernels/cross_entropy_loss.py
View file

@ -20,9 +20,8 @@ from transformers.models.llama.modeling_llama import logger
from packaging.version import Version
from unsloth_zoo.loss_utils import (
causal_loss_function,
transformers_losses_patcher,
patch_loss_function,
patch_loss_functions,
post_patch_loss_function,
)