Zengtudor/Wan2.1
1
0
Fork 0
mirror of https://github.com/Wan-Video/Wan2.1.git synced 2025-06-05 14:54:54 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix dist vace fwd hook error

Browse Source
This commit is contained in:
wangang.wa 2025-05-13 15:02:22 +08:00
parent 3fc0ac9adc
commit 58059485c7
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
wan/distributed/xdit_context_parallel.py
View File

@ -87,9 +87,10 @@ def usp_dit_forward_vace(
c, get_sequence_parallel_world_size(),
dim=1)[get_sequence_parallel_rank()]
hints = []
for block in self.vace_blocks:
c = block(c, **new_kwargs)
hints = torch.unbind(c)[:-1]
c, c_skip = block(c, **new_kwargs)
hints.append(c_skip)
return hints