Hunyuan refiner vae now works with tiled. (#9836)

2025-09-12 16:46:46 -07:00
parent d7f40442f9
commit a3b04de700
2 changed files with 15 additions and 7 deletions
--- a/comfy/ldm/hunyuan_video/vae_refiner.py
+++ b/comfy/ldm/hunyuan_video/vae_refiner.py
@@ -185,7 +185,6 @@ class Encoder(nn.Module):
        self.regul = comfy.ldm.models.autoencoder.DiagonalGaussianRegularizer()

    def forward(self, x):
-        x = x.unsqueeze(2)
        x = self.conv_in(x)

        for stage in self.down: