From d7173b482df6d7ebe423ff77a4b2b30030e7058a Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Fri, 14 Mar 2025 04:40:29 -0700
Subject: [PATCH] Update vision.py

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

diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py
index da1588e848..c80ea984e3 100644
--- a/unsloth/models/vision.py
+++ b/unsloth/models/vision.py
@@ -192,6 +192,7 @@ class FastBaseModel:
             if (disable_name.lower() == model_type_arch.lower() or \
                 disable_name.lower() in model_name.lower()) and \
                 dtype == torch.float16:
+                break
 
                 print(f"Unsloth: Using float16 precision for {model_type_arch} won't work! Using float32.")
                 os.environ["UNSLOTH_FORCE_FLOAT32"] = "1"