Fix issue with wan and other attention implementations.

comfyanonymous · Feb 26, 2025 · ea0f939 · ea0f939
1 parent f37551c
commit ea0f939
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/comfy/ldm/wan/model.py b/comfy/ldm/wan/model.py
@@ -88,9 +88,9 @@ def qkv_fn(x):
         q, k = apply_rope(q, k, freqs)
 
         x = optimized_attention(
-            q=q.view(b, s, n * d),
-            k=k.view(b, s, n * d),
-            v=v,
+            q.view(b, s, n * d),
+            k.view(b, s, n * d),
+            v,
             heads=self.num_heads,
         )