From fd97eebfa208f24055ce78a8eec16395ec4e043a Mon Sep 17 00:00:00 2001
From: Sean Sube <seansube@gmail.com>
Date: Tue, 21 Nov 2023 23:14:15 -0600
Subject: [PATCH] feat(api): add LCM scheduler

---
 api/onnx_web/diffusers/load.py                   | 1 +
 api/onnx_web/diffusers/version_safe_diffusers.py | 5 +++++
 2 files changed, 6 insertions(+)

diff --git a/api/onnx_web/diffusers/load.py b/api/onnx_web/diffusers/load.py
index d7ffe749..7755ecc5 100644
--- a/api/onnx_web/diffusers/load.py
+++ b/api/onnx_web/diffusers/load.py
@@ -44,6 +44,7 @@ from .version_safe_diffusers import (
     KarrasVeScheduler,
     KDPM2AncestralDiscreteScheduler,
     KDPM2DiscreteScheduler,
+    LCMScheduler,
     LMSDiscreteScheduler,
     OnnxRuntimeModel,
     OnnxStableDiffusionImg2ImgPipeline,
diff --git a/api/onnx_web/diffusers/version_safe_diffusers.py b/api/onnx_web/diffusers/version_safe_diffusers.py
index 79198b81..d256d615 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 LCMScheduler
+except ImportError:
+    from ..diffusers.stub_scheduler import StubScheduler as LCMScheduler
+
 try:
     from diffusers import UniPCMultistepScheduler
 except ImportError: