Update vision.py

2025-07-17 00:33:27 -07:00 · 2025-07-17 00:33:27 -07:00 · ec8a4c6216
commit ec8a4c6216
parent 9d1afec41d
1 changed files with 5 additions and 0 deletions
--- 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