fix pre-check issue

This commit is contained in:
harisreedhar
2026-02-23 15:52:46 +05:30
committed by henryruhs
parent f83ff29116
commit 47bdcd67f1
2 changed files with 7 additions and 0 deletions
+3
View File
@@ -54,6 +54,9 @@ 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)
+4
View File
@@ -9,6 +9,7 @@ from starlette.testclient import TestClient
from facefusion import metadata, session_manager, state_manager
from facefusion.apis import asset_store
from facefusion.apis.core import create_api
from facefusion.core import common_pre_check, processors_pre_check
from facefusion.download import conditional_download
from .helper import get_test_example_file, get_test_examples_directory
@@ -43,6 +44,9 @@ def test_client() -> Iterator[TestClient]:
state_manager.init_item('face_swapper_pixel_boost', '256x256')
state_manager.init_item('face_swapper_weight', 0.5)
common_pre_check()
processors_pre_check()
with TestClient(create_api()) as test_client:
yield test_client