helpy4/immich
1
0
Fork 0
mirror of https://github.com/immich-app/immich synced 2025-11-07 17:27:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 7

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
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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: