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

We name it InferenceSessionProvider

Browse Source
This commit is contained in:
henryruhs
2025-01-22 22:51:52 +01:00
parent 5270bd679c
commit dbbf3445b6
4 changed files with 20 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/test_execution.py
View File

@@ -1,4 +1,4 @@
from facefusion.execution import create_inference_execution_providers, get_available_execution_providers, has_execution_provider
from facefusion.execution import create_inference_session_providers, get_available_execution_providers, has_execution_provider
def test_has_execution_provider() -> None:
@@ -10,8 +10,8 @@ def test_get_available_execution_providers() -> None:
assert 'cpu' in get_available_execution_providers()
def test_create_inference_execution_providers() -> None:
execution_providers =\
def test_create_inference_session_providers() -> None:
inference_session_providers =\
[
('CUDAExecutionProvider',
{
@@ -21,4 +21,4 @@ def test_create_inference_execution_providers() -> None:
'CPUExecutionProvider'
]
assert create_inference_execution_providers('1', [ 'cpu', 'cuda' ]) == execution_providers
assert create_inference_session_providers('1', [ 'cpu', 'cuda' ]) == inference_session_providers