Fix some code
This commit is contained in:
parent
390a4e8bc6
commit
5a8a95f9f1
@ -49,6 +49,8 @@ def get_face_occluder() -> Any:
|
|||||||
global FACE_OCCLUDER
|
global FACE_OCCLUDER
|
||||||
|
|
||||||
with THREAD_LOCK:
|
with THREAD_LOCK:
|
||||||
|
while process_manager.is_checking():
|
||||||
|
sleep(0.5)
|
||||||
if FACE_OCCLUDER is None:
|
if FACE_OCCLUDER is None:
|
||||||
model_path = MODELS.get('face_occluder').get('path')
|
model_path = MODELS.get('face_occluder').get('path')
|
||||||
FACE_OCCLUDER = onnxruntime.InferenceSession(model_path, providers = apply_execution_provider_options(facefusion.globals.execution_providers))
|
FACE_OCCLUDER = onnxruntime.InferenceSession(model_path, providers = apply_execution_provider_options(facefusion.globals.execution_providers))
|
||||||
|
@ -28,7 +28,6 @@ from facefusion.processors.frame import globals as frame_processors_globals
|
|||||||
from facefusion.processors.frame import choices as frame_processors_choices
|
from facefusion.processors.frame import choices as frame_processors_choices
|
||||||
|
|
||||||
FRAME_PROCESSOR = None
|
FRAME_PROCESSOR = None
|
||||||
MODEL_MATRIX = None
|
|
||||||
THREAD_LOCK : threading.Lock = threading.Lock()
|
THREAD_LOCK : threading.Lock = threading.Lock()
|
||||||
NAME = __name__.upper()
|
NAME = __name__.upper()
|
||||||
MODELS : ModelSet =\
|
MODELS : ModelSet =\
|
||||||
|
Loading…
Reference in New Issue
Block a user