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"