diff --git a/machine-learning/immich_ml/main.py b/machine-learning/immich_ml/main.py index 62d8d1ec8e..35f04d77ef 100644 --- a/machine-learning/immich_ml/main.py +++ b/machine-learning/immich_ml/main.py @@ -184,8 +184,7 @@ async def run_inference(payload: Image | str, entries: InferenceEntries) -> Infe async def _run_inference(entry: InferenceEntry) -> None: model = await model_cache.get( - entry["name"], entry["type"], entry["task"], - ttl=settings.model_ttl, **entry["options"] + entry["name"], entry["type"], entry["task"], ttl=settings.model_ttl, **entry["options"] ) inputs = [payload] for dep in model.depends: