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"