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):