From 063cca03c85674f190192576dcbf41dc7be16ae2 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Wed, 19 Mar 2025 08:41:40 -0700 Subject: [PATCH] Update loader.py --- unsloth/models/loader.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/unsloth/models/loader.py b/unsloth/models/loader.py index 1cd05e1074..20b96eb2c5 100644 --- a/unsloth/models/loader.py +++ b/unsloth/models/loader.py @@ -646,10 +646,11 @@ class FastModel(FastBaseModel): # Set forced float32 env flag os.environ["UNSLOTH_FORCE_FLOAT32"] = "0" do_forced_float32 = False - model_type_arch = model_types[1] + for model_type_arch in model_types: + if model_type_arch != "siglip": break global FORCE_FLOAT32 for disable_name in FORCE_FLOAT32: - print(disable_name.lower(), model_type_arch.lower(), model_name.lower(), dtype, SUPPORTS_BFLOAT16) + print(model_types, disable_name.lower(), model_type_arch.lower(), model_name.lower(), dtype, SUPPORTS_BFLOAT16) if (disable_name.lower() == model_type_arch.lower() or \ disable_name.lower() in model_name.lower()) and \ ((dtype == torch.float16) or not SUPPORTS_BFLOAT16):