From f45bddbdaecdae01afd501ac12dbda2273ff206d Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Thu, 13 Mar 2025 06:06:24 -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 efdf67a95e..78b8c76ff4 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -185,6 +185,7 @@ class FastBaseModel: os.environ["UNSLOTH_FORCE_FLOAT32"] = "0" bnb_compute_dtype = dtype for disable_name in FORCE_FLOAT32: + print(disable_name, model_type_arch) if disable_name.lower() == model_type_arch.lower() and dtype == torch.float16: print(f"Unsloth: Using float16 precision for {model_type_arch} won't work! Using float32.") os.environ["UNSLOTH_FORCE_FLOAT32"] = "1"