CalvinBackup/facefusion
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion.git synced 2026-09-25 00:40:49 +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_detector.py
+4 -3
View File
@@ -78,13 +78,14 @@ def create_static_model_set(download_scope : DownloadScope) -> ModelSet:
def get_inference_pool() -> InferencePool:
model_names = [ state_manager.get_item('face_detector_model') ]
_, model_sources = collect_model_downloads()
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 = collect_model_downloads()
inference_manager.clear_inference_pool(__name__, model_sources)
model_names = [ state_manager.get_item('face_detector_model') ]
inference_manager.clear_inference_pool(__name__, model_names)
def collect_model_downloads() -> Tuple[DownloadSet, DownloadSet]: