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,