diff --git a/api/onnx_web/diffusion/load.py b/api/onnx_web/diffusion/load.py index eb9d5e14..da0bc13a 100644 --- a/api/onnx_web/diffusion/load.py +++ b/api/onnx_web/diffusion/load.py @@ -139,7 +139,7 @@ def load_pipeline( last_pipeline_options = options last_pipeline_scheduler = scheduler_type - if last_pipeline_scheduler != scheduler: + if last_pipeline_scheduler != scheduler_type: logger.debug("loading new diffusion scheduler") scheduler = scheduler_type.from_pretrained( model,