update wan2.2-fun

2026-03-19 14:58:12 +00:00 · 2025-08-21 20:08:49 +08:00
parent 46bd05b54d
commit 0d81626fe7
2 changed files with 18 additions and 13 deletions
--- a/diffsynth/models/wan_video_dit.py
+++ b/diffsynth/models/wan_video_dit.py
@@ -294,6 +294,7 @@ class WanModel(torch.nn.Module):
    ):
        super().__init__()
        self.dim = dim
+        self.in_dim = dim
        self.freq_dim = freq_dim
        self.has_image_input = has_image_input
        self.patch_size = patch_size