feat(api): add LCM scheduler

ssube · Nov 22, 2023 · fd97eeb · fd97eeb
1 parent 54dc970
commit fd97eeb
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/api/onnx_web/diffusers/load.py b/api/onnx_web/diffusers/load.py
@@ -44,6 +44,7 @@
     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
@@ -12,6 +12,11 @@
 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: