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

Update nodes.py

Browse Source
This commit is contained in:
kijai 2024-10-28 18:24:38 +02:00
parent c17750ea0a
commit dec1f8ba0f
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

@ -1104,6 +1104,9 @@ class ToraEncodeTrajectory:
"start_percent": ("FLOAT", {"default": 0.0, "min": 0.0, "max": 1.0, "step": 0.01}), "start_percent": ("FLOAT", {"default": 0.0, "min": 0.0, "max": 1.0, "step": 0.01}),
"end_percent": ("FLOAT", {"default": 1.0, "min": 0.0, "max": 1.0, "step": 0.01}), "end_percent": ("FLOAT", {"default": 1.0, "min": 0.0, "max": 1.0, "step": 0.01}),
}, },
"optional": {
"enable_tiling": ("BOOL", {"default": False}),
}
} }
RETURN_TYPES = ("TORAFEATURES", "IMAGE", ) RETURN_TYPES = ("TORAFEATURES", "IMAGE", )
@ -1111,7 +1114,7 @@ class ToraEncodeTrajectory:
FUNCTION = "encode" FUNCTION = "encode"
CATEGORY = "CogVideoWrapper" CATEGORY = "CogVideoWrapper"
def encode(self, pipeline, width, height, num_frames, coordinates, strength, start_percent, end_percent, tora_model): def encode(self, pipeline, width, height, num_frames, coordinates, strength, start_percent, end_percent, tora_model, enable_tiling=False):
check_diffusers_version() check_diffusers_version()
device = mm.get_torch_device() device = mm.get_torch_device()
offload_device = mm.unet_offload_device() offload_device = mm.unet_offload_device()
@ -1122,7 +1125,11 @@ class ToraEncodeTrajectory:
try: try:
vae._clear_fake_context_parallel_cache() vae._clear_fake_context_parallel_cache()
except: except:
pass pass
if enable_tiling:
from .mz_enable_vae_encode_tiling import enable_vae_encode_tiling
enable_vae_encode_tiling(vae)
if len(coordinates) < 10: if len(coordinates) < 10:
coords_list = [] coords_list = []