xinyun/ComfyUI-CogVideoXWrapper
1
0
Fork 0
mirror of https://git.datalinker.icu/kijai/ComfyUI-CogVideoXWrapper.git synced 2026-05-02 07:40:11 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update nodes.py

Browse Source
This commit is contained in:
kijai 2024-10-06 11:54:47 +03:00
parent fc7e7d7c62
commit af2a568c8e
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
nodes.py
View File

@ -1042,8 +1042,8 @@ class CogVideoXFunSampler:
# Load Sampler # Load Sampler
if context_options is not None and context_options["context_schedule"] == "temporal_tiling": if context_options is not None and context_options["context_schedule"] == "temporal_tiling":
logging.info("Temporal tiling enabled, changing scheduler to DDIM_tiled") logging.info("Temporal tiling enabled, changing scheduler to CogVideoXDDIM")
scheduler="DDIM_tiled" scheduler="CogVideoXDDIM"
scheduler_config = pipeline["scheduler_config"] scheduler_config = pipeline["scheduler_config"]
if scheduler in scheduler_mapping: if scheduler in scheduler_mapping:
noise_scheduler = scheduler_mapping[scheduler].from_config(scheduler_config) noise_scheduler = scheduler_mapping[scheduler].from_config(scheduler_config)
@ -1373,8 +1373,8 @@ class CogVideoXFunControlSampler:
# Load Sampler # Load Sampler
scheduler_config = pipeline["scheduler_config"] scheduler_config = pipeline["scheduler_config"]
if context_options is not None and context_options["context_schedule"] == "temporal_tiling": if context_options is not None and context_options["context_schedule"] == "temporal_tiling":
logging.info("Temporal tiling enabled, changing scheduler to DDIM_tiled") logging.info("Temporal tiling enabled, changing scheduler to CogVideoXDDIM")
scheduler="DDIM_tiled" scheduler="CogVideoXDDIM"
if scheduler in scheduler_mapping: if scheduler in scheduler_mapping:
noise_scheduler = scheduler_mapping[scheduler].from_config(scheduler_config) noise_scheduler = scheduler_mapping[scheduler].from_config(scheduler_config)
pipe.scheduler = noise_scheduler pipe.scheduler = noise_scheduler