Cutlery/immich
1
0
Fork 1
mirror of https://github.com/immich-app/immich.git synced 2025-07-09 03:04:16 -04:00
Code Issues Packages Projects Releases Wiki Activity

ignore rknn model if not using it

Browse Source
This commit is contained in:
yoni13 2025-01-13 18:37:01 +08:00
parent f328104e84
commit b6cc2054c5
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
machine-learning/app/models/base.py
View File

@ -69,6 +69,7 @@ class InferenceModel(ABC):
def _download(self) -> None: def _download(self) -> None:
ignore_patterns = [] if self.model_format == ModelFormat.ARMNN else ["*.armnn"] ignore_patterns = [] if self.model_format == ModelFormat.ARMNN else ["*.armnn"]
if self.model_format != ModelFormat.RKNN: ignore_patterns.append("*.rknn")
snapshot_download( snapshot_download(
f"immich-app/{clean_name(self.model_name)}", f"immich-app/{clean_name(self.model_name)}",
cache_dir=self.cache_dir, cache_dir=self.cache_dir,