diff --git a/wanvideo/modules/model.py b/wanvideo/modules/model.py index 6d90338..977c705 100644 --- a/wanvideo/modules/model.py +++ b/wanvideo/modules/model.py @@ -671,7 +671,7 @@ class WanT2VCrossAttention(WanSelfAttention): x = self.normalized_attention_guidance(b, n, d, q, context, nag_context, nag_params) else: if is_longcat: - k = self.norm_k(self.k(context).to(self.norm_k.weight.dtype)).to(x.dtype).view(b, -1, n, d) + k = self.norm_k(self.k(context).to(self.norm_k.weight.dtype).view(b, -1, n, d)).to(x.dtype) else: k = self.norm_k(self.k(context).to(self.norm_k.weight.dtype)).to(x.dtype).view(b, -1, n, d)