diff --git a/unsloth/kernels/cross_entropy_loss.py b/unsloth/kernels/cross_entropy_loss.py
index 266ece4ca2..66d5046deb 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
+# @torch._disable_dynamo
 def fast_cross_entropy_loss(
     logits,
     labels,