ComfyUI-Demo / diffusion_webui /utils /scheduler_list.py
kadirnar's picture
Upload 25 files
a22dacf
raw
history blame
1.18 kB
from diffusers import (
DDIMScheduler,
EulerAncestralDiscreteScheduler,
EulerDiscreteScheduler,
HeunDiscreteScheduler,
LMSDiscreteScheduler,
UniPCMultistepScheduler,
)
SCHEDULER_LIST = [
"DDIM",
"EulerA",
"Euler",
"LMS",
"Heun",
"UniPC",
]
def get_scheduler_list(pipe, scheduler):
if scheduler == SCHEDULER_LIST[0]:
pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
elif scheduler == SCHEDULER_LIST[1]:
pipe.scheduler = EulerAncestralDiscreteScheduler.from_config(
pipe.scheduler.config
)
elif scheduler == SCHEDULER_LIST[2]:
pipe.scheduler = EulerDiscreteScheduler.from_config(
pipe.scheduler.config
)
elif scheduler == SCHEDULER_LIST[3]:
pipe.scheduler = LMSDiscreteScheduler.from_config(pipe.scheduler.config)
elif scheduler == SCHEDULER_LIST[4]:
pipe.scheduler = HeunDiscreteScheduler.from_config(
pipe.scheduler.config
)
elif scheduler == SCHEDULER_LIST[5]:
pipe.scheduler = UniPCMultistepScheduler.from_config(
pipe.scheduler.config
)
return pipe