CalvinBackup/facefusion
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion.git synced 2026-08-21 08:17:12 +02:00
Code Issues Packages Projects Releases Wiki Activity

Fix broken inference pools part2

Browse Source
This commit is contained in:
henryruhs
2025-04-28 10:03:25 +02: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
facefusion/face_recognizer.py
+4 -3
View File
@@ -40,13 +40,14 @@ def create_static_model_set(download_scope : DownloadScope) -> ModelSet:
def get_inference_pool() -> InferencePool:
model_names = [ 'arcface' ]
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 = [ 'arcface' ]
inference_manager.clear_inference_pool(__name__, model_names)
def get_model_options() -> ModelOptions: