From 5f04f870aa88385d0db3765c39cba42a1948fa1d Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Mon, 17 Mar 2025 04:45:49 -0700 Subject: [PATCH] Update _utils.py --- unsloth/models/_utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py index dd7b02216a..3ad6e9d6e2 100644 --- a/unsloth/models/_utils.py +++ b/unsloth/models/_utils.py @@ -1187,6 +1187,7 @@ def unsloth_compile_transformers( do_forced_float32 = False model_type_arch = model_types[1] for disable_name in FORCE_FLOAT32: + print(disable_name, model_type_arch, model_name) if (disable_name.lower() == model_type_arch.lower() or \ disable_name.lower() in model_name.lower()) and \ dtype == torch.float16: