From 9dff3fd21ebfd42e1010e30b77e929af166166a8 Mon Sep 17 00:00:00 2001 From: Hansheng Chen Date: Sun, 9 Nov 2025 16:10:34 -0800 Subject: [PATCH] Fix temb typecasting in GMQwen (#6) --- modules/architectures/gm_qwen_image.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/modules/architectures/gm_qwen_image.py b/modules/architectures/gm_qwen_image.py index 6adb73e..36e545f 100644 --- a/modules/architectures/gm_qwen_image.py +++ b/modules/architectures/gm_qwen_image.py @@ -174,7 +174,7 @@ class GMQwenImageTransformer2DModel(QwenImageTransformer2DModel): bs, k, 1, 1, h * self.patch_size, w * self.patch_size )[..., :x.shape[-2], :x.shape[-1]].log_softmax(dim=1) if self.constant_logstd is None: - out_logstds = self.proj_out_logstds(temb.detach()).reshape(bs, 1, 1, 1, 1, 1) + out_logstds = self.proj_out_logstds(temb.detach().float()).reshape(bs, 1, 1, 1, 1, 1) else: out_logstds = hidden_states.new_full((bs, 1, 1, 1, 1, 1), float(self.constant_logstd))