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

don't keep vae tiling enabled forever

Browse Source
This commit is contained in:
kijai 2024-08-30 20:14:39 +03:00
parent 9208585442
commit 248428ccf8
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
nodes.py
View File

@ -353,6 +353,8 @@ class CogVideoDecode:
tile_overlap_factor_height=tile_overlap_factor_height, tile_overlap_factor_height=tile_overlap_factor_height,
tile_overlap_factor_width=tile_overlap_factor_width, tile_overlap_factor_width=tile_overlap_factor_width,
) )
else:
vae.disable_tiling()
latents = latents.to(vae.dtype) latents = latents.to(vae.dtype)
latents = latents.permute(0, 2, 1, 3, 4) # [batch_size, num_channels, num_frames, height, width] latents = latents.permute(0, 2, 1, 3, 4) # [batch_size, num_channels, num_frames, height, width]
latents = 1 / vae.config.scaling_factor * latents latents = 1 / vae.config.scaling_factor * latents