From 825a00188e5dd4191b3ecaca67d446691de5808d Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Sun, 3 Nov 2024 14:09:47 -0800 Subject: [PATCH] Update cross_entropy_loss.py --- unsloth/kernels/cross_entropy_loss.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/unsloth/kernels/cross_entropy_loss.py b/unsloth/kernels/cross_entropy_loss.py index 5236e51985..f5a015073c 100644 --- a/unsloth/kernels/cross_entropy_loss.py +++ b/unsloth/kernels/cross_entropy_loss.py @@ -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, )