mirrors/facefusion
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix OpenVINO by aliasing GPU.0 to GPU

Browse Source
This commit is contained in:
henryruhs 2024-10-15 21:32:11 +02:00
parent ac6966d132
commit c75bea3151
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
facefusion/execution.py
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.' + 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' ]: