From fb4ae0134a07a69fedf6ca6978845a7531120f31 Mon Sep 17 00:00:00 2001 From: pluesclues <136766175+pluesclues@users.noreply.github.com> Date: Fri, 29 Aug 2025 04:32:21 -0400 Subject: [PATCH] Update mistral.py, showed flag to not call cut cross entropy (#3233) * Update mistral.py, showed flag to not call cut cross entropy * Update mistral.py, made it so if its not equal to zero * Update unsloth/models/mistral.py --------- Co-authored-by: Daniel Han --- unsloth/models/mistral.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/unsloth/models/mistral.py b/unsloth/models/mistral.py index 16a23f8853..6274f2e5df 100644 --- a/unsloth/models/mistral.py +++ b/unsloth/models/mistral.py @@ -300,7 +300,7 @@ def MistralForCausalLM_fast_forward( # < 1024 Normal Unsloth uses less VRAM! if bsz * q_len <= 1024: RETURN_LOGITS = True - if not RETURN_LOGITS and HAS_CUT_CROSS_ENTROPY and labels is not None: + if not RETURN_LOGITS and HAS_CUT_CROSS_ENTROPY and os.environ.get("UNSLOTH_ENABLE_CCE", "1") != "0" and labels is not None: n_items = kwargs.get("num_items_in_batch", None) or kwargs.get("n_items", None) logit_softcapping = getattr(self.config, "final_logit_softcapping", 0) loss = fused_linear_cross_entropy(