fix(api): use pipeline class name in cache key
This commit is contained in:
parent
55576ae87c
commit
339868de4d
|
@ -147,7 +147,7 @@ def load_pipeline(
|
||||||
lpw: bool,
|
lpw: bool,
|
||||||
inversion: Optional[str],
|
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_key = (scheduler_name, model)
|
||||||
scheduler_type = get_pipeline_schedulers()[scheduler_name]
|
scheduler_type = get_pipeline_schedulers()[scheduler_name]
|
||||||
|
|
||||||
|
|
Loading…
Reference in New Issue