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

reformat

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

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

@ -69,7 +69,8 @@ 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") 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,