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,
|
||||
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]
|
||||
|
||||
|
|
Loading…
Reference in New Issue