apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix scheduler caching

This commit is contained in:
Sean Sube 2023-02-25 13:14:34 -06:00
parent 18f59f034d
commit cb8e9e7080
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/onnx_web/diffusion/load.py
View File

@ -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