From 9ccbc0ed1fb32107a848a3e11f05b89f65107d18 Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Wed, 30 Oct 2024 13:35:01 -0700
Subject: [PATCH] Update _utils.py

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

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