Fix OpenVINO by aliasing GPU.0 to GPU

This commit is contained in:
henryruhs 2024-10-16 17:29:32 +02:00
parent 853474bf79
commit 20d2b6a4ea

View File

@ -52,7 +52,7 @@ def create_execution_providers(execution_device_id : str, execution_provider_key
if execution_provider_key == 'openvino':
execution_providers.append((execution_provider_set.get(execution_provider_key),
{
'device_type': 'GPU' if execution_device_id == 0 else 'GPU.' + execution_device_id,
'device_type': 'GPU' if execution_device_id == '0' else 'GPU.' + execution_device_id,
'precision': 'FP32'
}))
if execution_provider_key in [ 'directml', 'rocm' ]: