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

fix path

Browse Source
This commit is contained in:
yoni13 2025-01-12 01:09:36 +08:00
parent efaf70eb9d
commit 19ee48f6f0
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
machine-learning/app/sessions/rknn.py
View File

@ -24,9 +24,9 @@ class RknnSession:
self.model_path = Path(model_path) self.model_path = Path(model_path)
self.ort_model_path = str(self.model_path).replace(".rknn", ".onnx") self.ort_model_path = str(self.model_path).replace(".rknn", ".onnx")
if 'textual' in self.model_path.name: if 'textual' in str(self.model_path):
self.tpe = settings.rknn_textual_threads self.tpe = settings.rknn_textual_threads
elif 'visual' in self.model_path.name: elif 'visual' in str(self.model_path):
self.tpe = settings.rknn_visual_threads self.tpe = settings.rknn_visual_threads
else: else:
self.tpe = settings.rknn_facial_detection_threads self.tpe = settings.rknn_facial_detection_threads