From 25ec3d96a323c8455c6ee69e43bdd7a5599d3cc0 Mon Sep 17 00:00:00 2001
From: "Reiner \"Tiles\" Prokein" <ReinerBforartists@users.noreply.github.com>
Date: Sat, 28 Feb 2026 01:03:45 +0100
Subject: [PATCH] Class WanVAE, def encode, feat_map is using self.decoder
 instead of self.encoder (#12682)

---
 comfy/ldm/wan/vae.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/comfy/ldm/wan/vae.py b/comfy/ldm/wan/vae.py
index 7903c7690..71f73c64e 100644
--- a/comfy/ldm/wan/vae.py
+++ b/comfy/ldm/wan/vae.py
@@ -485,7 +485,7 @@ class WanVAE(nn.Module):
         iter_ = 1 + (t - 1) // 4
         feat_map = None
         if iter_ > 1:
-            feat_map = [None] * count_conv3d(self.decoder)
+            feat_map = [None] * count_conv3d(self.encoder)
         ## 对encode输入的x，按时间拆分为1、4、4、4....
         for i in range(iter_):
             conv_idx = [0]