2023-05-11 12:40:03 +00:00
|
|
|
from diffusers import DDIMScheduler, DPMSolverMultistepScheduler, KDPM2DiscreteScheduler, \
|
|
|
|
|
KDPM2AncestralDiscreteScheduler, EulerDiscreteScheduler, EulerAncestralDiscreteScheduler, \
|
2023-05-11 14:23:33 +00:00
|
|
|
HeunDiscreteScheduler, LMSDiscreteScheduler, PNDMScheduler, UniPCMultistepScheduler, \
|
2023-05-11 15:18:34 +00:00
|
|
|
DPMSolverSinglestepScheduler, DEISMultistepScheduler, DDPMScheduler
|
2023-05-11 12:40:03 +00:00
|
|
|
|
|
|
|
|
SCHEDULER_MAP = dict(
|
2023-05-11 14:23:33 +00:00
|
|
|
ddim=(DDIMScheduler, dict()),
|
2023-05-11 15:18:34 +00:00
|
|
|
ddpm=(DDPMScheduler, dict()),
|
2023-05-11 15:10:30 +00:00
|
|
|
deis=(DEISMultistepScheduler, dict()),
|
2023-05-11 14:59:43 +00:00
|
|
|
lms=(LMSDiscreteScheduler, dict()),
|
|
|
|
|
pndm=(PNDMScheduler, dict()),
|
2023-05-12 14:43:32 +00:00
|
|
|
heun=(HeunDiscreteScheduler, dict(use_karras_sigmas=False)),
|
|
|
|
|
heun_k=(HeunDiscreteScheduler, dict(use_karras_sigmas=True)),
|
2023-05-11 14:59:43 +00:00
|
|
|
euler=(EulerDiscreteScheduler, dict(use_karras_sigmas=False)),
|
|
|
|
|
euler_k=(EulerDiscreteScheduler, dict(use_karras_sigmas=True)),
|
|
|
|
|
euler_a=(EulerAncestralDiscreteScheduler, dict()),
|
|
|
|
|
kdpm_2=(KDPM2DiscreteScheduler, dict()),
|
|
|
|
|
kdpm_2_a=(KDPM2AncestralDiscreteScheduler, dict()),
|
2023-05-11 14:23:33 +00:00
|
|
|
dpmpp_2s=(DPMSolverSinglestepScheduler, dict()),
|
2023-05-11 14:59:43 +00:00
|
|
|
dpmpp_2m=(DPMSolverMultistepScheduler, dict(use_karras_sigmas=False)),
|
|
|
|
|
dpmpp_2m_k=(DPMSolverMultistepScheduler, dict(use_karras_sigmas=True)),
|
2023-05-11 12:40:03 +00:00
|
|
|
unipc=(UniPCMultistepScheduler, dict(cpu_only=True))
|
|
|
|
|
)
|
