xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-08-24 20:45:45 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix issue with wan and other attention implementations.

Browse Source
This commit is contained in:
comfyanonymous 2025-02-25 19:13:39 -05:00
parent 65760e20d6
commit f5990fa4c3
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
comfy/ldm/wan/model.py
View File

@ -88,9 +88,9 @@ class WanSelfAttention(nn.Module):
q, k = apply_rope(q, k, freqs) q, k = apply_rope(q, k, freqs)
x = optimized_attention( x = optimized_attention(
q=q.view(b, s, n * d), q.view(b, s, n * d),
k=k.view(b, s, n * d), k.view(b, s, n * d),
v=v, v,
heads=self.num_heads, heads=self.num_heads,
) )