diff --git a/unsloth/kernels/cross_entropy_loss.py b/unsloth/kernels/cross_entropy_loss.py index 474c291ed8..9beb2da25c 100644 --- a/unsloth/kernels/cross_entropy_loss.py +++ b/unsloth/kernels/cross_entropy_loss.py @@ -353,7 +353,7 @@ class Fast_CrossEntropyLoss(torch.autograd.Function): pass pass - +@torch._disable_dynamo def fast_cross_entropy_loss( logits, labels,