diff --git a/api/onnx_web/diffusers/load.py b/api/onnx_web/diffusers/load.py index 7755ecc5..75dafc2e 100644 --- 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,