diff --git a/comfy/ldm/hunyuan_video/vae_refiner.py b/comfy/ldm/hunyuan_video/vae_refiner.py index 3fe30089d..75a6a2d18 100644 --- a/comfy/ldm/hunyuan_video/vae_refiner.py +++ b/comfy/ldm/hunyuan_video/vae_refiner.py @@ -327,7 +327,7 @@ class Decoder(nn.Module): x = self.mid.block_2(self.mid.attn_1(self.mid.block_1(x))) if self.refiner_vae: - x = torch.split(x, 1, dim=2) + x = torch.split(x, 2, dim=2) else: x = [ x ] out = []