gitea-mirror/immich
1
0
Fork 0
mirror of https://github.com/immich-app/immich.git synced 2026-02-04 19:12:11 -08:00
Code Issues Packages Projects Releases Wiki Activity

change log message

Browse Source
This commit is contained in:
mertalev
2024-06-08 21:22:38 -04:00
parent 717961ce7b
commit fb4fe5d40b
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
machine-learning/app/models/facial_recognition/recognition.py
View File

@@ -32,7 +32,7 @@ class FaceRecognizer(InferenceModel):
def _load(self) -> ModelSession:
session = self._make_session(self.model_path)
if isinstance(session, ort.InferenceSession) and not ort_has_batch_dim(session):
log.debug(f"Adding batch dimension to model {self.model_path}")
log.info(f"Adding batch dimension to recognition model {self.model_name}")
ort_add_batch_dim(self.model_path, self.model_path)
session = self._make_session(self.model_path)
self.model = ArcFaceONNX(