diff --git a/unsloth/models/_utils.py b/unsloth/models/_utils.py index 3ad6e9d6e2..5c5cc375da 100644 --- a/unsloth/models/_utils.py +++ b/unsloth/models/_utils.py @@ -1186,6 +1186,7 @@ def unsloth_compile_transformers( os.environ["UNSLOTH_FORCE_FLOAT32"] = "0" do_forced_float32 = False model_type_arch = model_types[1] + print("!!!!!!!!!!!!!") for disable_name in FORCE_FLOAT32: print(disable_name, model_type_arch, model_name) if (disable_name.lower() == model_type_arch.lower() or \