diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py
index a39bc58db0..35b56f26a9 100644
--- a/unsloth/models/_utils.py
+++ b/unsloth/models/_utils.py
@@ -910,7 +910,7 @@ class Unsloth_Offloaded_Gradient_Checkpointer(torch.autograd.Function):
 pass
 
 
-@torch._disable_dynamo
+# @torch._disable_dynamo
 def unsloth_offloaded_gradient_checkpoint(function, *args, use_reentrant = None, **kwargs):
     return Unsloth_Offloaded_Gradient_Checkpointer.apply(function, *args)
 pass