Tiny wan vae optimizations. (#9136)

2025-08-01 02:25:38 -07:00
parent 4696d74305
commit 1e638a140b
2 changed files with 10 additions and 5 deletions
--- a/comfy/ldm/wan/vae2_2.py
+++ b/comfy/ldm/wan/vae2_2.py
@@ -151,7 +151,7 @@ class ResidualBlock(nn.Module):
                        ],
                        dim=2,
                    )
-                x = layer(x, feat_cache[idx])
+                x = layer(x, cache_list=feat_cache, cache_idx=idx)
                feat_cache[idx] = cache_x
                feat_idx[0] += 1
            else: