diff --git a/facefusion/choices.py b/facefusion/choices.py index e0e76a66..6e91560a 100755 --- a/facefusion/choices.py +++ b/facefusion/choices.py @@ -37,7 +37,7 @@ execution_provider_set : ExecutionProviderSet =\ 'rocm': 'ROCMExecutionProvider', 'tensorrt': 'TensorrtExecutionProvider' } -execution_provider : List[ExecutionProvider] = list(execution_provider_set.keys()) +execution_providers : List[ExecutionProvider] = list(execution_provider_set.keys()) download_provider_set : DownloadProviderSet =\ { 'github': 'https://github.com/facefusion/facefusion-assets/releases/download/{base_name}/{file_name}', diff --git a/tests/test_execution.py b/tests/test_execution.py index aa836faf..1d1743c9 100644 --- a/tests/test_execution.py +++ b/tests/test_execution.py @@ -20,4 +20,4 @@ def test_multiple_execution_providers() -> None: 'CPUExecutionProvider' ] - assert create_inference_execution_providers('1', ['cpu', 'cuda']) == execution_providers + assert create_inference_execution_providers('1', [ 'cpu', 'cuda' ]) == execution_providers