From 339868de4d5a918e7f8eb4bf8825c64e6d59ce2e Mon Sep 17 00:00:00 2001 From: Sean Sube Date: Sun, 5 Mar 2023 21:54:09 -0600 Subject: [PATCH] fix(api): use pipeline class name in cache key --- api/onnx_web/diffusers/load.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/api/onnx_web/diffusers/load.py b/api/onnx_web/diffusers/load.py index c91151dd..2c8db970 100644 --- a/api/onnx_web/diffusers/load.py +++ b/api/onnx_web/diffusers/load.py @@ -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]