align wan tokenizer to official

2026-03-21 16:18:13 +00:00 · 2025-02-28 15:50:07 +08:00
parent 6fa8dbe077
commit 61a30673c2
3 changed files with 7 additions and 0 deletions
--- a/diffsynth/models/wan_video_vae.py
+++ b/diffsynth/models/wan_video_vae.py
@@ -785,6 +785,7 @@ class WanVideoVAE(nn.Module):
                video = self.single_decode(hidden_state, device)
            video = video.squeeze(0)
            videos.append(video)
+        videos = torch.stack(videos)
        return videos