pre_check() on test instead

This commit is contained in:
harisreedhar
2026-02-22 13:45:43 +05:30
committed by henryruhs
parent c35cf5a607
commit 2ee01b38ee
2 changed files with 3 additions and 2 deletions
-2
View File
@@ -54,8 +54,6 @@ def route(args : Args) -> None:
benchmarker.render()
if state_manager.get_item('command') == 'api':
if not common_pre_check() or not processors_pre_check():
hard_exit(2)
logger.info(translator.get('api_started').format(host = state_manager.get_item('api_host'), port = state_manager.get_item('api_port')), __name__)
uvicorn.run(create_api(), host = state_manager.get_item('api_host'), port = state_manager.get_item('api_port'))
hard_exit(1)
+3
View File
@@ -10,6 +10,7 @@ from facefusion import metadata, session_manager, state_manager
from facefusion.apis import asset_store
from facefusion.apis.core import create_api
from facefusion.args_helper import apply_args
from facefusion.core import common_pre_check, processors_pre_check
from facefusion.download import conditional_download
from facefusion.program import collect_step_program
from .helper import get_test_example_file, get_test_examples_directory
@@ -34,6 +35,8 @@ def test_client() -> Iterator[TestClient]:
state_manager.init_item('temp_path', tempfile.gettempdir())
state_manager.init_item('download_providers', [ 'github', 'huggingface' ])
state_manager.init_item('face_selector_mode', 'many')
common_pre_check()
processors_pre_check()
with TestClient(create_api()) as test_client:
yield test_client