From 2fc5ec930c30021dfdbe5b9e5b8b40da9346ae6f Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Sat, 9 Dec 2023 23:51:09 -0600
Subject: [PATCH] feat(api): add support for DPM SDE scheduler

---
 api/onnx_web/diffusers/load.py                   | 2 ++
 api/onnx_web/diffusers/version_safe_diffusers.py | 5 +++++
 gui/src/strings/en.ts                            | 1 +
 3 files changed, 8 insertions(+)

diff --git a/api/onnx_web/diffusers/load.py b/api/onnx_web/diffusers/load.py
index 30a87863..2aaf2154 100644
--- a/api/onnx_web/diffusers/load.py
+++ b/api/onnx_web/diffusers/load.py
@@ -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,
diff --git a/api/onnx_web/diffusers/version_safe_diffusers.py b/api/onnx_web/diffusers/version_safe_diffusers.py
index d256d615..3d51ea4b 100644
--- a/api/onnx_web/diffusers/version_safe_diffusers.py
+++ b/api/onnx_web/diffusers/version_safe_diffusers.py
@@ -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:
diff --git a/gui/src/strings/en.ts b/gui/src/strings/en.ts
index c9d901f9..304c3335 100644
--- 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',