add LCM to available scheduler list

2023-11-21 23:23:46 -06:00 · 2023-11-21 23:23:46 -06:00 · 66dfa7206a
parent fd97eebfa2
commit 66dfa7206a
1 changed files with 1 additions and 0 deletions
--- a/api/onnx_web/diffusers/load.py
+++ b/api/onnx_web/diffusers/load.py
@ -85,6 +85,7 @@ pipeline_schedulers = {
    "k-dpm-2-a": KDPM2AncestralDiscreteScheduler,
    "k-dpm-2": KDPM2DiscreteScheduler,
    "karras-ve": KarrasVeScheduler,
+    "lcm": LCMScheduler,
    "lms-discrete": LMSDiscreteScheduler,
    "pndm": PNDMScheduler,
    "unipc-multi": UniPCMultistepScheduler,