diff --git a/api/onnx_web/diffusers/load.py b/api/onnx_web/diffusers/load.py index 30a878633..2aaf21549 100644 --- 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 index d256d6159..3d51ea4bd 100644 --- 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 index c9d901f97..304c33358 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',