123456789101112131415161718192021222324 |
- from facefusion.execution import create_execution_providers, get_execution_provider_choices, has_execution_provider
- def test_get_execution_provider_choices() -> None:
- assert 'cpu' in get_execution_provider_choices()
- def test_has_execution_provider() -> None:
- assert has_execution_provider('cpu') is True
- assert has_execution_provider('openvino') is False
- def test_multiple_execution_providers() -> None:
- execution_providers =\
- [
- ('CUDAExecutionProvider',
- {
- 'device_id': '1',
- 'cudnn_conv_algo_search': 'DEFAULT'
- }),
- 'CPUExecutionProvider'
- ]
- assert create_execution_providers('1', [ 'cpu', 'cuda' ]) == execution_providers
|