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,