wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-02-10 05:22:34 +08:00
Code Issues Actions 14 Packages Projects Releases Wiki Activity

Fix.

Browse Source
This commit is contained in:
comfyanonymous 2025-11-20 22:32:38 -05:00 committed by GitHub
parent 8b10e0bdaa
commit 1840bc5226
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
comfy/ldm/hunyuan_video/vae_refiner.py
View File

@ -327,7 +327,7 @@ class Decoder(nn.Module):
x = self.mid.block_2(self.mid.attn_1(self.mid.block_1(x))) x = self.mid.block_2(self.mid.attn_1(self.mid.block_1(x)))
if self.refiner_vae: if self.refiner_vae:
x = torch.split(x, 1, dim=2) x = torch.split(x, 2, dim=2)
else: else:
x = [ x ] x = [ x ]
out = [] out = []