Skip to content

Commit ea0f939

Browse files
Fix issue with wan and other attention implementations.
1 parent f37551c commit ea0f939

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

comfy/ldm/wan/model.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -88,9 +88,9 @@ def qkv_fn(x):
8888
q, k = apply_rope(q, k, freqs)
8989

9090
x = optimized_attention(
91-
q=q.view(b, s, n * d),
92-
k=k.view(b, s, n * d),
93-
v=v,
91+
q.view(b, s, n * d),
92+
k.view(b, s, n * d),
93+
v,
9494
heads=self.num_heads,
9595
)
9696

0 commit comments

Comments
 (0)