3.5.1 (#985)
* Fix type for device id * Fix type for device id * Fix order * Remove comma * Replace Generator typing * Replace Generator typing * Conditional kill myself (#984) * Introduce conditional remove mask * fix --------- Co-authored-by: harisreedhar <h4harisreedhar.s.s@gmail.com> --------- Co-authored-by: harisreedhar <h4harisreedhar.s.s@gmail.com>
This commit is contained in:
@@ -15,10 +15,10 @@ def test_create_inference_session_providers() -> None:
|
||||
[
|
||||
('CUDAExecutionProvider',
|
||||
{
|
||||
'device_id': '1',
|
||||
'device_id': 1,
|
||||
'cudnn_conv_algo_search': 'EXHAUSTIVE'
|
||||
}),
|
||||
'CPUExecutionProvider'
|
||||
]
|
||||
|
||||
assert create_inference_session_providers('1', [ 'cpu', 'cuda' ]) == inference_session_providers
|
||||
assert create_inference_session_providers(1, [ 'cpu', 'cuda' ]) == inference_session_providers
|
||||
|
||||
@@ -18,7 +18,7 @@ def before_all() -> None:
|
||||
subprocess.run([ 'ffmpeg', '-i', get_test_example_file('source.jpg'), '-vf', 'crop=iw*0.8:ih*0.8', get_test_example_file('source-80crop.jpg') ])
|
||||
subprocess.run([ 'ffmpeg', '-i', get_test_example_file('source.jpg'), '-vf', 'crop=iw*0.7:ih*0.7', get_test_example_file('source-70crop.jpg') ])
|
||||
subprocess.run([ 'ffmpeg', '-i', get_test_example_file('source.jpg'), '-vf', 'crop=iw*0.6:ih*0.6', get_test_example_file('source-60crop.jpg') ])
|
||||
state_manager.init_item('execution_device_ids', [ '0' ])
|
||||
state_manager.init_item('execution_device_ids', [ 0 ])
|
||||
state_manager.init_item('execution_providers', [ 'cpu' ])
|
||||
state_manager.init_item('download_providers', [ 'github' ])
|
||||
state_manager.init_item('face_detector_angles', [ 0 ])
|
||||
|
||||
@@ -9,7 +9,7 @@ from facefusion.inference_manager import INFERENCE_POOL_SET, get_inference_pool
|
||||
|
||||
@pytest.fixture(scope = 'module', autouse = True)
|
||||
def before_all() -> None:
|
||||
state_manager.init_item('execution_device_ids', [ '0' ])
|
||||
state_manager.init_item('execution_device_ids', [ 0 ])
|
||||
state_manager.init_item('execution_providers', [ 'cpu' ])
|
||||
state_manager.init_item('download_providers', [ 'github' ])
|
||||
content_analyser.pre_check()
|
||||
|
||||
Reference in New Issue
Block a user