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

Add multi GPU support for openvino

Browse Source
This commit is contained in:
henryruhs
2025-03-02 15:31:24 +01:00
parent 6c49a9779b
commit 7bdd084c20
1 changed files with 9 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
facefusion/execution.py
View File

@@ -56,7 +56,7 @@ def create_inference_session_providers(execution_device_id : str, execution_prov
if execution_provider == 'openvino':
inference_session_providers.append((facefusion.choices.execution_provider_set.get(execution_provider),
{
'device_type': 'GPU' if execution_device_id == '0' else 'GPU.' + execution_device_id,
'device_type': resolve_openvino_device_type(execution_device_id),
'precision': 'FP32'
}))
if execution_provider == 'coreml':
@@ -68,6 +68,14 @@ def create_inference_session_providers(execution_device_id : str, execution_prov
return inference_session_providers
def resolve_openvino_device_type(execution_device_id : str) -> str:
if execution_device_id == '0':
return 'GPU'
if execution_device_id == '∞':
return 'MULTI:GPU'
return 'GPU.' + device_id
def is_geforce_16_series() -> bool:
execution_devices = detect_static_execution_devices()
product_names = ('GeForce GTX 1630', 'GeForce GTX 1650', 'GeForce GTX 1660')