Zengtudor/Wan2.1
1
0
Fork 0
mirror of https://github.com/Wan-Video/Wan2.1.git synced 2025-07-13 11:10:11 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #4 from songruizz/fix/FA3-duplicated-code

Browse Source 
fix performance regression due to duplicated code
This commit is contained in:
zc8gerard 2025-05-27 16:21:46 +08:00 committed by GitHub
commit 7ae8070fd9
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 0 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
wan/distributed/xdit_context_parallel.py
View File

@ -304,13 +304,6 @@ def usp_attn_forward(self,
value=half(v),
window_size=self.window_size)
x = xFuserLongContextAttention()(
None,
query=half(q),
key=half(k),
value=half(v),
window_size=self.window_size)
# TODO: padding after attention.
# x = torch.cat([x, x.new_zeros(b, s - x.size(1), n, d)], dim=1)