mirror of
https://github.com/facefusion/facefusion.git
synced 2026-05-12 10:31:33 +02:00
fix pre-check issue
This commit is contained in:
@@ -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)
|
||||
|
||||
@@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user