1
0
Fork 0

feat(api): add support for DPM SDE scheduler

This commit is contained in:
Sean Sube 2023-12-09 23:51:09 -06:00
parent 50db19922a
commit 2fc5ec930c
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
3 changed files with 8 additions and 0 deletions

View File

@ -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,

View File

@ -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:

View File

@ -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',