From ec8a4c621668d53815cca125d9634b58ea537c17 Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Thu, 17 Jul 2025 00:33:27 -0700
Subject: [PATCH] Update vision.py

---
 unsloth/models/vision.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py
index 7442f07e73..075498d040 100644
--- a/unsloth/models/vision.py
+++ b/unsloth/models/vision.py
@@ -755,6 +755,11 @@ class FastBaseModel:
         os.environ["UNSLOTH_RETURN_HIDDEN_STATES"] = "0"
         # Must enable returning logits
         os.environ["UNSLOTH_RETURN_LOGITS"] = "1"
+        try:
+            # Turn off skip guards
+            torch.compiler.set_stance(skip_guard_eval_unsafe = False)
+        except:
+            pass
         return model
     pass