xinyun/ComfyUI-CogVideoXWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-CogVideoXWrapper.git synced 2026-05-03 01:53:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

tile decode by default

Browse Source
This commit is contained in:
kijai 2024-11-16 19:36:13 +02:00
parent bececf0189
commit f21432bea1
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
nodes.py
View File

@ -945,7 +945,7 @@ class CogVideoDecode:
return {"required": { return {"required": {
"pipeline": ("COGVIDEOPIPE",), "pipeline": ("COGVIDEOPIPE",),
"samples": ("LATENT", ), "samples": ("LATENT", ),
"enable_vae_tiling": ("BOOLEAN", {"default": False, "tooltip": "Drastically reduces memory use but may introduce seams"}), "enable_vae_tiling": ("BOOLEAN", {"default": True, "tooltip": "Drastically reduces memory use but may introduce seams"}),
}, },
"optional": { "optional": {
"tile_sample_min_height": ("INT", {"default": 240, "min": 16, "max": 2048, "step": 8, "tooltip": "Minimum tile height, default is half the height"}), "tile_sample_min_height": ("INT", {"default": 240, "min": 16, "max": 2048, "step": 8, "tooltip": "Minimum tile height, default is half the height"}),
@ -995,7 +995,14 @@ class CogVideoDecode:
except: except:
pass pass
frames = vae.decode(latents[:, :, additional_frames:]).sample try:
frames = vae.decode(latents[:, :, additional_frames:]).sample
except:
mm.soft_empty_cache()
log.warning("Failed to decode, retrying with tiling")
vae.enable_tiling()
frames = vae.decode(latents[:, :, additional_frames:]).sample
vae.disable_tiling() vae.disable_tiling()
if not pipeline["cpu_offloading"]: if not pipeline["cpu_offloading"]:
vae.to(offload_device) vae.to(offload_device)