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

Removing checking for device_ids for openvino since cpu will always be available

Browse source
This commit is contained in:
Aleksander 2025-10-15 07:33:09 -07:00
parent 5a57c99f98
commit 62d2974e00
1 changed files with 0 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -62,13 +62,6 @@ class OrtSession:
def _providers_default(self) -> list[str]: def _providers_default(self) -> list[str]:
available_providers = set(ort.get_available_providers()) available_providers = set(ort.get_available_providers())
log.debug(f"Available ORT providers: {available_providers}") log.debug(f"Available ORT providers: {available_providers}")
if (openvino := "OpenVINOExecutionProvider") in available_providers:
device_ids: list[str] = ort.capi._pybind_state.get_available_openvino_device_ids()
log.debug(f"Available OpenVINO devices: {device_ids}")
if not device_ids:
log.warning("No device found in OpenVINO. Falling back to CPU.")
available_providers.remove(openvino)
return [provider for provider in SUPPORTED_PROVIDERS if provider in available_providers] return [provider for provider in SUPPORTED_PROVIDERS if provider in available_providers]
@property @property