xinyun/ComfyUI-CogVideoXWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-CogVideoXWrapper.git synced 2025-12-08 20:34:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

diffusers backwards compatibility on tiled encode

Browse Source 
for some reason it's tuple in 0.31.0
This commit is contained in:
kijai 2024-10-25 21:49:48 +03:00
parent 7fe4716f2d
commit f9c1e11851
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mz_enable_vae_encode_tiling.py
View File

@ -81,6 +81,8 @@ def tiled_encode(self, x: torch.Tensor) -> torch.Tensor:
] ]
tile = self.encoder(tile) tile = self.encoder(tile)
if not isinstance(tile, tuple):
tile = (tile,)
if self.quant_conv is not None: if self.quant_conv is not None:
tile = self.quant_conv(tile) tile = self.quant_conv(tile)
time.append(tile[0]) time.append(tile[0])