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: