mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2025-03-14 21:47:07 +00:00
Fix issue with wan and other attention implementations.
This commit is contained in:
parent
f37551c1d2
commit
ea0f939df3
@ -88,9 +88,9 @@ class WanSelfAttention(nn.Module):
|
||||
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,
|
||||
)
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user