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

Set inswapper according to execution provider

Browse Source
This commit is contained in:
henryruhs
2024-05-19 16:50:46 +02:00
parent 319e3f9652
commit cefd2fa41d
3 changed files with 12 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tests/test_execution.py
View File

@@ -1,4 +1,4 @@
from facefusion.execution import encode_execution_providers, decode_execution_providers, apply_execution_provider_options
from facefusion.execution import encode_execution_providers, decode_execution_providers, has_execution_provider, apply_execution_provider_options
def test_encode_execution_providers() -> None:
@@ -9,6 +9,11 @@ def test_decode_execution_providers() -> None:
assert decode_execution_providers([ 'cpu' ]) == [ 'CPUExecutionProvider' ]
def test_has_execution_provider() -> None:
assert has_execution_provider('CPUExecutionProvider') is True
assert has_execution_provider('InvalidExecutionProvider') is False
def test_multiple_execution_providers() -> None:
execution_provider_with_options =\
[