From 57998847541b5e3e9ed16f9927360a6aaa215e85 Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Wed, 30 Oct 2024 16:50:56 -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 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,