Fix some code

This commit is contained in:
henryruhs 2024-03-27 01:01:41 +01:00
parent 390a4e8bc6
commit 5a8a95f9f1
2 changed files with 2 additions and 1 deletions

View File

@ -49,6 +49,8 @@ def get_face_occluder() -> Any:
global FACE_OCCLUDER
with THREAD_LOCK:
while process_manager.is_checking():
sleep(0.5)
if FACE_OCCLUDER is None:
model_path = MODELS.get('face_occluder').get('path')
FACE_OCCLUDER = onnxruntime.InferenceSession(model_path, providers = apply_execution_provider_options(facefusion.globals.execution_providers))

View File

@ -28,7 +28,6 @@ from facefusion.processors.frame import globals as frame_processors_globals
from facefusion.processors.frame import choices as frame_processors_choices
FRAME_PROCESSOR = None
MODEL_MATRIX = None
THREAD_LOCK : threading.Lock = threading.Lock()
NAME = __name__.upper()
MODELS : ModelSet =\