feat(api): add support for DPM SDE scheduler
This commit is contained in:
parent
50db19922a
commit
2fc5ec930c
|
@ -30,6 +30,7 @@ from .version_safe_diffusers import (
|
|||
DDPMScheduler,
|
||||
DEISMultistepScheduler,
|
||||
DPMSolverMultistepScheduler,
|
||||
DPMSolverSDEScheduler,
|
||||
DPMSolverSinglestepScheduler,
|
||||
EulerAncestralDiscreteScheduler,
|
||||
EulerDiscreteScheduler,
|
||||
|
@ -71,6 +72,7 @@ pipeline_schedulers = {
|
|||
"ddpm": DDPMScheduler,
|
||||
"deis-multi": DEISMultistepScheduler,
|
||||
"dpm-multi": DPMSolverMultistepScheduler,
|
||||
"dpm-sde": DPMSolverSDEScheduler,
|
||||
"dpm-single": DPMSolverSinglestepScheduler,
|
||||
"euler": EulerDiscreteScheduler,
|
||||
"euler-a": EulerAncestralDiscreteScheduler,
|
||||
|
|
|
@ -12,6 +12,11 @@ try:
|
|||
except ImportError:
|
||||
from ..diffusers.stub_scheduler import StubScheduler as DEISMultistepScheduler
|
||||
|
||||
try:
|
||||
from diffusers import DPMSolverSDEScheduler
|
||||
except:
|
||||
from ..diffusers.stub_scheduler import StubScheduler as DPMSolverSDEScheduler
|
||||
|
||||
try:
|
||||
from diffusers import LCMScheduler
|
||||
except ImportError:
|
||||
|
|
|
@ -287,6 +287,7 @@ export const I18N_STRINGS_EN = {
|
|||
'ddpm': 'DDPM',
|
||||
'deis-multi': 'DEIS Multistep',
|
||||
'dpm-multi': 'DPM Multistep',
|
||||
'dpm-sde': 'DPM SDE (Turbo)',
|
||||
'dpm-single': 'DPM Singlestep',
|
||||
'euler': 'Euler',
|
||||
'euler-a': 'Euler Ancestral',
|
||||
|
|
Loading…
Reference in New Issue