Class WanVAE, def encode, feat_map is using self.decoder instead of self.encoder (#12682)

This commit is contained in:
Reiner "Tiles" Prokein
2026-02-28 01:03:45 +01:00
committed by GitHub
parent 1f1ec377ce
commit 25ec3d96a3

View File

@@ -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]