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

add LCM and UniPC samplers to CogVideoXFunSampler

Browse Source
This commit is contained in:
Phr00t 2024-09-21 12:51:28 -04:00
parent 69dbbd3487
commit df3f210287
1 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
nodes.py
View File

@ -3,7 +3,7 @@ import torch
import folder_paths import folder_paths
import comfy.model_management as mm import comfy.model_management as mm
from comfy.utils import ProgressBar, load_torch_file from comfy.utils import ProgressBar, load_torch_file
from diffusers.schedulers import CogVideoXDDIMScheduler, CogVideoXDPMScheduler, DDIMScheduler, PNDMScheduler, DPMSolverMultistepScheduler, EulerDiscreteScheduler, EulerAncestralDiscreteScheduler from diffusers.schedulers import CogVideoXDDIMScheduler, CogVideoXDPMScheduler, DDIMScheduler, PNDMScheduler, DPMSolverMultistepScheduler, EulerDiscreteScheduler, EulerAncestralDiscreteScheduler, LCMScheduler, UniPCMultistepScheduler
from diffusers.models import AutoencoderKLCogVideoX, CogVideoXTransformer3DModel from diffusers.models import AutoencoderKLCogVideoX, CogVideoXTransformer3DModel
from .pipeline_cogvideox import CogVideoXPipeline from .pipeline_cogvideox import CogVideoXPipeline
@ -698,6 +698,8 @@ class CogVideoXFunSampler:
"DPM++", "DPM++",
"PNDM", "PNDM",
"DDIM", "DDIM",
"LCM",
"UniPC",
"CogVideoXDDIM", "CogVideoXDDIM",
"CogVideoXDPMScheduler", "CogVideoXDPMScheduler",
], ],
@ -753,6 +755,10 @@ class CogVideoXFunSampler:
noise_scheduler = PNDMScheduler.from_config(scheduler_config) noise_scheduler = PNDMScheduler.from_config(scheduler_config)
elif scheduler == "DDIM": elif scheduler == "DDIM":
noise_scheduler = DDIMScheduler.from_config(scheduler_config) noise_scheduler = DDIMScheduler.from_config(scheduler_config)
elif scheduler == "LCM":
noise_scheduler = LCMScheduler.from_config(scheduler_config)
elif scheduler == "UniPC":
noise_scheduler = UniPCMultistepScheduler.from_config(scheduler_config)
elif scheduler == "CogVideoXDDIM": elif scheduler == "CogVideoXDDIM":
noise_scheduler = CogVideoXDDIMScheduler.from_config(scheduler_config) noise_scheduler = CogVideoXDDIMScheduler.from_config(scheduler_config)
elif scheduler == "CogVideoXDPMScheduler": elif scheduler == "CogVideoXDPMScheduler":
@ -811,6 +817,8 @@ class CogVideoXFunVid2VidSampler:
"DPM++", "DPM++",
"PNDM", "PNDM",
"DDIM", "DDIM",
"LCM",
"UniPC",
"CogVideoXDDIM", "CogVideoXDDIM",
"CogVideoXDPMScheduler", "CogVideoXDPMScheduler",
], ],
@ -855,6 +863,10 @@ class CogVideoXFunVid2VidSampler:
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 = DDIMScheduler.from_pretrained(base_path, subfolder= 'scheduler')
elif scheduler == "LCM":
noise_scheduler = LCMScheduler.from_pretrained(base_path, subfolder= 'scheduler')
elif scheduler == "UniPC":
noise_scheduler = UniPCMultistepScheduler.from_pretrained(base_path, subfolder= 'scheduler')
elif scheduler == "CogVideoXDDIM": elif scheduler == "CogVideoXDDIM":
noise_scheduler = CogVideoXDDIMScheduler.from_pretrained(base_path, subfolder= 'scheduler') noise_scheduler = CogVideoXDDIMScheduler.from_pretrained(base_path, subfolder= 'scheduler')
elif scheduler == "CogVideoXDPMScheduler": elif scheduler == "CogVideoXDPMScheduler":