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

Refactor choices, imports and naming

Browse Source
This commit is contained in:
henryruhs 2024-12-11 22:37:10 +01:00
parent 2836b331cd
commit af04c946d4
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
facefusion/choices.py
View File

@ -37,7 +37,7 @@ execution_provider_set : ExecutionProviderSet =\
'rocm': 'ROCMExecutionProvider', 'rocm': 'ROCMExecutionProvider',
'tensorrt': 'TensorrtExecutionProvider' 'tensorrt': 'TensorrtExecutionProvider'
} }
execution_provider : List[ExecutionProvider] = list(execution_provider_set.keys()) execution_providers : List[ExecutionProvider] = list(execution_provider_set.keys())
download_provider_set : DownloadProviderSet =\ download_provider_set : DownloadProviderSet =\
{ {
'github': 'https://github.com/facefusion/facefusion-assets/releases/download/{base_name}/{file_name}', 'github': 'https://github.com/facefusion/facefusion-assets/releases/download/{base_name}/{file_name}',

2
tests/test_execution.py
View File

@ -20,4 +20,4 @@ def test_multiple_execution_providers() -> None:
'CPUExecutionProvider' 'CPUExecutionProvider'
] ]
assert create_inference_execution_providers('1', ['cpu', 'cuda']) == execution_providers assert create_inference_execution_providers('1', [ 'cpu', 'cuda' ]) == execution_providers