apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

fix(api): use pipeline class name in cache key

This commit is contained in:
Sean Sube 2023-03-05 21:54:09 -06:00
parent 55576ae87c
commit 339868de4d
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/diffusers/load.py
View File

@ -147,7 +147,7 @@ def load_pipeline(
lpw: bool,
inversion: Optional[str],
):
pipe_key = (pipeline, model, device.device, device.provider, lpw, inversion)
pipe_key = (pipeline.__name__, model, device.device, device.provider, lpw, inversion)
scheduler_key = (scheduler_name, model)
scheduler_type = get_pipeline_schedulers()[scheduler_name]