From 67d40f3f6d9fbfe1c068378e698316c24d22a2e3 Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Wed, 20 Nov 2024 19:40:08 -0800
Subject: [PATCH] Update _utils.py

---
 unsloth/models/_utils.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py
index f5075a5445..e1214a89e8 100644
--- a/unsloth/models/_utils.py
+++ b/unsloth/models/_utils.py
@@ -1111,7 +1111,9 @@ def unsloth_compile_transformers(
     cudagraphs              = False,
     debug                   = False,
     import_from_cache       = False,
+    disable                 = False,
 ):
+    if disable: return
     model_types = get_transformers_model_type(
         model_name        = model_name,
         token             = token,
@@ -1140,6 +1142,7 @@ def unsloth_compile_transformers(
             cudagraphs             = cudagraphs,
             debug                  = debug,
             import_from_cache      = import_from_cache,
+            disable                = disable,
         )
     pass
     return