From 96fb0f3afea01ffb773a9d2faed26178ffd18478 Mon Sep 17 00:00:00 2001 From: Mr_Dwj Date: Thu, 12 Feb 2026 23:51:56 +0800 Subject: [PATCH] fix: unpack Resample38 output --- diffsynth/models/wan_video_vae.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/diffsynth/models/wan_video_vae.py b/diffsynth/models/wan_video_vae.py index 3c2181a..b77f75c 100644 --- a/diffsynth/models/wan_video_vae.py +++ b/diffsynth/models/wan_video_vae.py @@ -469,7 +469,7 @@ class Down_ResidualBlock(nn.Module): def forward(self, x, feat_cache=None, feat_idx=[0]): x_copy = x.clone() for module in self.downsamples: - x = module(x, feat_cache, feat_idx) + x, feat_cache, feat_idx = module(x, feat_cache, feat_idx) return x + self.avg_shortcut(x_copy), feat_cache, feat_idx @@ -506,7 +506,7 @@ class Up_ResidualBlock(nn.Module): def forward(self, x, feat_cache=None, feat_idx=[0], first_chunk=False): x_main = x.clone() for module in self.upsamples: - x_main = module(x_main, feat_cache, feat_idx) + 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