jones/facefusion
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix broken inference pools part2

Browse Source
This commit is contained in:
henryruhs
2025-02-09 01:43:58 +01:00
parent b438b899e8
commit 1bdc02014c
18 changed files with 92 additions and 65 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
facefusion/content_analyser.py
View File

@@ -42,13 +42,14 @@ def create_static_model_set(download_scope : DownloadScope) -> ModelSet:
def get_inference_pool() -> InferencePool:
model_names = [ 'yolo_nsfw' ]
model_sources = get_model_options().get('sources')
return inference_manager.get_inference_pool(__name__, model_sources)
return inference_manager.get_inference_pool(__name__, model_names, model_sources)
def clear_inference_pool() -> None:
model_sources = get_model_options().get('sources')
inference_manager.clear_inference_pool(__name__, model_sources)
model_names = [ 'yolo_nsfw' ]
inference_manager.clear_inference_pool(__name__, model_names)
def get_model_options() -> ModelOptions: