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

Update nodes.py

Browse Source
This commit is contained in:
kijai 2024-09-18 17:57:07 +03:00
parent aad80fa1ae
commit 7cf10c64d6
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
nodes.py
View File

@ -455,6 +455,8 @@ class CogVideoXFunSampler:
"DPM++", "DPM++",
"PNDM", "PNDM",
"DDIM", "DDIM",
"CogVideoXDDIM",
"CogVideoXDPMScheduler",
], ],
{ {
"default": 'DDIM' "default": 'DDIM'
@ -502,7 +504,11 @@ class CogVideoXFunSampler:
elif scheduler == "PNDM": elif scheduler == "PNDM":
noise_scheduler = PNDMScheduler.from_pretrained(base_path, subfolder= 'scheduler') noise_scheduler = PNDMScheduler.from_pretrained(base_path, subfolder= 'scheduler')
elif scheduler == "DDIM": elif scheduler == "DDIM":
noise_scheduler = DDIMScheduler.from_pretrained(base_path, subfolder= 'scheduler') noise_scheduler = CogVideoXDDIMScheduler.from_pretrained(base_path, subfolder= 'scheduler')
elif scheduler == "CogVideoXDDIM":
noise_scheduler = CogVideoXDDIMScheduler.from_pretrained(base_path, subfolder= 'scheduler')
elif scheduler == "CogVideoXDPMScheduler":
noise_scheduler = CogVideoXDPMScheduler.from_pretrained(base_path, subfolder= 'scheduler')
pipe.scheduler = noise_scheduler pipe.scheduler = noise_scheduler
#if not pipeline["cpu_offloading"]: #if not pipeline["cpu_offloading"]: