Merge branch 'usp' into xdit

2026-03-20 07:18:14 +00:00 · 2025-03-25 11:45:49 +08:00
parent 3dc28f428f 54081bdcbb
commit 64eaa0d76a
6 changed files with 169 additions and 2 deletions
--- a/diffsynth/models/wan_video_dit.py
+++ b/diffsynth/models/wan_video_dit.py
@@ -90,6 +90,7 @@ def rope_apply(x, freqs, num_heads):
    x = rearrange(x, "b s (n d) -> b s n d", n=num_heads)
    x_out = torch.view_as_complex(x.to(torch.float64).reshape(
        x.shape[0], x.shape[1], x.shape[2], -1, 2))
+    
    x_out = torch.view_as_real(x_out * freqs).flatten(2)
    return x_out.to(x.dtype)