feat: make additional schedulers available in UI
This commit is contained in:
parent
16c8b54b1a
commit
93e53f6dc3
|
@ -3,8 +3,13 @@ from diffusers import (
|
|||
DDIMScheduler,
|
||||
DDPMScheduler,
|
||||
DPMSolverMultistepScheduler,
|
||||
DPMSolverSinglestepScheduler,
|
||||
EulerDiscreteScheduler,
|
||||
EulerAncestralDiscreteScheduler,
|
||||
HeunDiscreteScheduler,
|
||||
KDPM2AncestralDiscreteScheduler,
|
||||
KDPM2DiscreteScheduler,
|
||||
KarrasVeScheduler,
|
||||
LMSDiscreteScheduler,
|
||||
PNDMScheduler,
|
||||
# onnx
|
||||
|
@ -54,8 +59,13 @@ pipeline_schedulers = {
|
|||
'ddim': DDIMScheduler,
|
||||
'ddpm': DDPMScheduler,
|
||||
'dpm-multi': DPMSolverMultistepScheduler,
|
||||
'dpm-single': DPMSolverSinglestepScheduler,
|
||||
'euler': EulerDiscreteScheduler,
|
||||
'euler-a': EulerAncestralDiscreteScheduler,
|
||||
'heun': HeunDiscreteScheduler,
|
||||
'k-dpm-2-a': KDPM2AncestralDiscreteScheduler,
|
||||
'k-dpm-2': KDPM2DiscreteScheduler,
|
||||
'karras-ve': KarrasVeScheduler,
|
||||
'lms-discrete': LMSDiscreteScheduler,
|
||||
'pndm': PNDMScheduler,
|
||||
}
|
||||
|
|
|
@ -12,8 +12,13 @@ export const SCHEDULER_LABELS: Record<string, string> = {
|
|||
'ddim': 'DDIM',
|
||||
'ddpm': 'DDPM',
|
||||
'dpm-multi': 'DPM Multistep',
|
||||
'dpm-single': 'DPM Singlestep',
|
||||
'euler': 'Euler',
|
||||
'euler-a': 'Euler Ancestral',
|
||||
'lms-discrete': 'LMS Discrete',
|
||||
'heun': 'Heun',
|
||||
'k-dpm-2-a': 'KDPM2 Ancestral',
|
||||
'k-dpm-2': 'KDPM2',
|
||||
'karras-ve': 'Karras Ve',
|
||||
'lms-discrete': 'LMS',
|
||||
'pndm': 'PNDM',
|
||||
};
|
||||
|
|
Loading…
Reference in New Issue