From 8ce6916d72132f9d9866f35db8670c0d3cf74602 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Mon, 3 Nov 2025 10:36:55 +0200 Subject: [PATCH] Fix for some cases of using comfy_chunked rope --- wanvideo/modules/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/wanvideo/modules/model.py b/wanvideo/modules/model.py index 57db863..e089733 100644 --- a/wanvideo/modules/model.py +++ b/wanvideo/modules/model.py @@ -1003,7 +1003,7 @@ class WanAttentionBlock(nn.Module): return torch.addcmul(shift_msa, norm_x, 1 + scale_msa) def ffn_chunked(self, x, shift_mlp, scale_mlp, num_chunks=4): - modulated_input = torch.addcmul(shift_mlp, self.norm2(x), 1 + scale_mlp) + modulated_input = torch.addcmul(shift_mlp, self.norm2(x.to(shift_mlp.dtype)), 1 + scale_mlp).to(x.dtype) result = torch.empty_like(x) seq_len = modulated_input.shape[1]