From bd3c5822a1f9fda25163796ba74c139113b963a4 Mon Sep 17 00:00:00 2001 From: Mr_Dwj Date: Fri, 13 Feb 2026 01:13:08 +0800 Subject: [PATCH] fix: WanVAE2.2 decode error --- diffsynth/models/wan_video_vae.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/diffsynth/models/wan_video_vae.py b/diffsynth/models/wan_video_vae.py index b77f75c..19ab6bd 100644 --- a/diffsynth/models/wan_video_vae.py +++ b/diffsynth/models/wan_video_vae.py @@ -509,7 +509,7 @@ class Up_ResidualBlock(nn.Module): x_main, feat_cache, feat_idx = module(x_main, feat_cache, feat_idx) if self.avg_shortcut is not None: x_shortcut = self.avg_shortcut(x, first_chunk) - return x_main + x_shortcut + return x_main + x_shortcut, feat_cache, feat_idx else: return x_main, feat_cache, feat_idx @@ -1336,6 +1336,7 @@ class VideoVAE38_(VideoVAE_): x = self.conv2(z) for i in range(iter_): self._conv_idx = [0] + # breakpoint() if i == 0: out, self._feat_map, self._conv_idx = self.decoder(x[:, :, i:i + 1, :, :], feat_cache=self._feat_map,