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

Find OpenVINOExecutionProvider index instead of assuming index 0

Browse source
This commit is contained in:
Aleksander 2025-10-15 07:46:59 -07:00
parent 62d2974e00
commit ce268fa92d
1 changed files with 4 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

6
machine-learning/immich_ml/sessions/ort.py
View file

@ -134,7 +134,8 @@ class OrtSession:
elif settings.model_inter_op_threads == 0 and self.providers == ["CPUExecutionProvider"]:
sess_options.inter_op_num_threads = 1
elif settings.model_inter_op_threads == 0 and (
"OpenVINOExecutionProvider" in self.providers and self._provider_options[0].get("device_type") == "CPU"
"OpenVINOExecutionProvider" in self.providers
and self._provider_options[self.providers.index("OpenVINOExecutionProvider")].get("device_type") == "CPU"
):
sess_options.inter_op_num_threads = 1
@ -144,7 +145,8 @@ class OrtSession:
elif settings.model_intra_op_threads == 0 and self.providers == ["CPUExecutionProvider"]:
sess_options.intra_op_num_threads = 2
elif settings.model_intra_op_threads == 0 and (
"OpenVINOExecutionProvider" in self.providers and self._provider_options[0].get("device_type") == "CPU"
"OpenVINOExecutionProvider" in self.providers
and self._provider_options[self.providers.index("OpenVINOExecutionProvider")].get("device_type") == "CPU"
):
sess_options.intra_op_num_threads = 1