public/immich
1
0
Fork 0
mirror of https://github.com/immich-app/immich.git synced 2025-11-10 14:22:35 +09:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat(ml): conditionally download .armnn models (#6650)

Browse Source
This commit is contained in:
Mert
2024-01-28 10:31:59 -05:00
committed by GitHub
parent fa0913120d
commit a84b6f5fb1
5 changed files with 127 additions and 38 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
machine-learning/app/models/clip.py
View File

@@ -81,11 +81,11 @@ class BaseCLIPEncoder(InferenceModel):
@property
def textual_path(self) -> Path:
return self.textual_dir / "model.onnx"
return self.textual_dir / f"model.{self.preferred_runtime}"
@property
def visual_path(self) -> Path:
return self.visual_dir / "model.onnx"
return self.visual_dir / f"model.{self.preferred_runtime}"
@property
def tokenizer_file_path(self) -> Path: