Update model.py

2025-10-28 02:04:50 +02:00
parent 2633119505
commit eebbcd5ee0
1 changed files with 1 additions and 1 deletions
@@ -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)