From 54b901bc50b7798a78ed862813d4d986dd4c1e24 Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Wed, 30 Oct 2024 17:19:30 -0700
Subject: [PATCH] Update cross_entropy_loss.py

---
 unsloth/kernels/cross_entropy_loss.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

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,