From ba31a178ec9a99b857b237cacd9b6c72883dbcce Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Wed, 19 Mar 2025 08:37:53 -0700 Subject: [PATCH] Update loader.py --- unsloth/models/loader.py | 1 + 1 file changed, 1 insertion(+) diff --git a/unsloth/models/loader.py b/unsloth/models/loader.py index 86edf154b5..1cd05e1074 100644 --- a/unsloth/models/loader.py +++ b/unsloth/models/loader.py @@ -649,6 +649,7 @@ class FastModel(FastBaseModel): model_type_arch = model_types[1] global FORCE_FLOAT32 for disable_name in FORCE_FLOAT32: + print(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):