feat(api): add support for DPM SDE scheduler

ssube · Dec 10, 2023 · 2fc5ec9 · 2fc5ec9
1 parent 50db199
commit 2fc5ec9
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 0 deletions.
diff --git a/api/onnx_web/diffusers/load.py b/api/onnx_web/diffusers/load.py
@@ -30,6 +30,7 @@
     DDPMScheduler,
     DEISMultistepScheduler,
     DPMSolverMultistepScheduler,
+    DPMSolverSDEScheduler,
     DPMSolverSinglestepScheduler,
     EulerAncestralDiscreteScheduler,
     EulerDiscreteScheduler,
@@ -71,6 +72,7 @@
     "ddpm": DDPMScheduler,
     "deis-multi": DEISMultistepScheduler,
     "dpm-multi": DPMSolverMultistepScheduler,
+    "dpm-sde": DPMSolverSDEScheduler,
     "dpm-single": DPMSolverSinglestepScheduler,
     "euler": EulerDiscreteScheduler,
     "euler-a": EulerAncestralDiscreteScheduler,

diff --git a/api/onnx_web/diffusers/version_safe_diffusers.py b/api/onnx_web/diffusers/version_safe_diffusers.py
@@ -12,6 +12,11 @@
 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:

diff --git a/gui/src/strings/en.ts b/gui/src/strings/en.ts
@@ -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',