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

ssube · Mar 6, 2023 · 339868d · 339868d
1 parent 55576ae
commit 339868d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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]