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,
|
DDPMScheduler,
|
||||||
DEISMultistepScheduler,
|
DEISMultistepScheduler,
|
||||||
DPMSolverMultistepScheduler,
|
DPMSolverMultistepScheduler,
|
||||||
|
DPMSolverSDEScheduler,
|
||||||
DPMSolverSinglestepScheduler,
|
DPMSolverSinglestepScheduler,
|
||||||
EulerAncestralDiscreteScheduler,
|
EulerAncestralDiscreteScheduler,
|
||||||
EulerDiscreteScheduler,
|
EulerDiscreteScheduler,
|
||||||
|
@ -71,6 +72,7 @@ pipeline_schedulers = {
|
||||||
"ddpm": DDPMScheduler,
|
"ddpm": DDPMScheduler,
|
||||||
"deis-multi": DEISMultistepScheduler,
|
"deis-multi": DEISMultistepScheduler,
|
||||||
"dpm-multi": DPMSolverMultistepScheduler,
|
"dpm-multi": DPMSolverMultistepScheduler,
|
||||||
|
"dpm-sde": DPMSolverSDEScheduler,
|
||||||
"dpm-single": DPMSolverSinglestepScheduler,
|
"dpm-single": DPMSolverSinglestepScheduler,
|
||||||
"euler": EulerDiscreteScheduler,
|
"euler": EulerDiscreteScheduler,
|
||||||
"euler-a": EulerAncestralDiscreteScheduler,
|
"euler-a": EulerAncestralDiscreteScheduler,
|
||||||
|
|
|
@ -12,6 +12,11 @@ try:
|
||||||
except ImportError:
|
except ImportError:
|
||||||
from ..diffusers.stub_scheduler import StubScheduler as DEISMultistepScheduler
|
from ..diffusers.stub_scheduler import StubScheduler as DEISMultistepScheduler
|
||||||
|
|
||||||
|
try:
|
||||||
|
from diffusers import DPMSolverSDEScheduler
|
||||||
|
except:
|
||||||
|
from ..diffusers.stub_scheduler import StubScheduler as DPMSolverSDEScheduler
|
||||||
|
|
||||||
try:
|
try:
|
||||||
from diffusers import LCMScheduler
|
from diffusers import LCMScheduler
|
||||||
except ImportError:
|
except ImportError:
|
||||||
|
|
|
@ -287,6 +287,7 @@ export const I18N_STRINGS_EN = {
|
||||||
'ddpm': 'DDPM',
|
'ddpm': 'DDPM',
|
||||||
'deis-multi': 'DEIS Multistep',
|
'deis-multi': 'DEIS Multistep',
|
||||||
'dpm-multi': 'DPM Multistep',
|
'dpm-multi': 'DPM Multistep',
|
||||||
|
'dpm-sde': 'DPM SDE (Turbo)',
|
||||||
'dpm-single': 'DPM Singlestep',
|
'dpm-single': 'DPM Singlestep',
|
||||||
'euler': 'Euler',
|
'euler': 'Euler',
|
||||||
'euler-a': 'Euler Ancestral',
|
'euler-a': 'Euler Ancestral',
|
||||||
|
|
Loading…
Reference in New Issue