From cb8e9e70803de0442b66c0dd53512a05a0e8544b Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sat, 25 Feb 2023 13:14:34 -0600 Subject: [PATCH] fix scheduler caching --- 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 5e334941..47c35fcd 100644 --- a/api/onnx_web/diffusion/load.py +++ b/api/onnx_web/diffusion/load.py @@ -215,6 +215,6 @@ def load_pipeline( pipe = pipe.to(device.torch_str()) server.cache.set("diffusion", pipe_key, pipe) - server.cache.set("scheduler", scheduler_key, scheduler) + server.cache.set("scheduler", scheduler_key, components["scheduler"]) return pipe