From 4aaa1e269a801629be37cfb9fec220e91a4d7b07 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Mon, 30 Jun 2025 07:06:35 -0700 Subject: [PATCH] Update loader.py --- unsloth/models/loader.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/unsloth/models/loader.py b/unsloth/models/loader.py index c974a4640b..a95a54b59d 100644 --- a/unsloth/models/loader.py +++ b/unsloth/models/loader.py @@ -578,7 +578,7 @@ class FastModel(FastBaseModel): os.environ["UNSLOTH_DISABLE_STATIC_GENERATION"] = "1" os.environ["UNSLOTH_FORCE_CUSTOM_DTYPE"] = \ "float16;torch.float16;torch.float16;"\ - "if name.endswith(('.conv')): module.to(torch.float32);"\ + "if name.endswith(('.conv')): module;"\ "from unsloth_zoo.temporary_patches.gemma3n import patch_Gemma3nConvNormAct_forward; patch_Gemma3nConvNormAct_forward()" if transformers_version < Version("4.53.0"):