fix(api): use correct var for diffusion scheduler cache

ssube · Feb 12, 2023 · aaf82a4 · aaf82a4
1 parent feb7e7d
commit aaf82a4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,