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

apply config correctly, adjust defaults

Browse source
This commit is contained in:
mertalev 2025-06-16 16:24:26 -04:00
parent 22690fa096
commit 585d093baf
No known key found for this signature in database
GPG key ID: DF6ABC77AAD98C95
10 changed files with 43 additions and 35 deletions
Download patch file Download diff file Expand all files Collapse all files

2
machine-learning/immich_ml/models/clip/textual.py
View file

@ -19,7 +19,7 @@ class BaseCLIPTextualEncoder(InferenceModel):
depends = []
identity = (ModelType.TEXTUAL, ModelTask.SEARCH)
def _predict(self, inputs: str, language: str | None = None, **kwargs: Any) -> str:
def _predict(self, inputs: str, language: str | None = None) -> str:
tokens = self.tokenize(inputs, language=language)
res: NDArray[np.float32] = self.session.run(None, tokens)[0][0]
return serialize_np_array(res)