From aaf82a42f14a27bb9107d90ac04a799bccbc3f37 Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sun, 12 Feb 2023 12:47:55 -0600 Subject: [PATCH] fix(api): use correct var for diffusion scheduler cache --- api/onnx_web/diffusion/load.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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,