From b6cc2054c523276f7460d00c29eb68e431a4cb5b Mon Sep 17 00:00:00 2001 From: yoni13 Date: Mon, 13 Jan 2025 18:37:01 +0800 Subject: [PATCH] ignore rknn model if not using it --- machine-learning/app/models/base.py | 1 + 1 file changed, 1 insertion(+) diff --git a/machine-learning/app/models/base.py b/machine-learning/app/models/base.py index 8b34bf2b8e..04e2507b3c 100644 --- a/machine-learning/app/models/base.py +++ b/machine-learning/app/models/base.py @@ -69,6 +69,7 @@ class InferenceModel(ABC): def _download(self) -> None: ignore_patterns = [] if self.model_format == ModelFormat.ARMNN else ["*.armnn"] + if self.model_format != ModelFormat.RKNN: ignore_patterns.append("*.rknn") snapshot_download( f"immich-app/{clean_name(self.model_name)}", cache_dir=self.cache_dir,