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/face_landmarker.py
View File

@@ -79,13 +79,14 @@ def create_static_model_set(download_scope : DownloadScope) -> ModelSet:
def get_inference_pool() -> InferencePool:
model_names = [ state_manager.get_item('face_landmarker_model'), 'fan_68_5' ]
_, 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_landmarker_model'), 'fan_68_5' ]
inference_manager.clear_inference_pool(__name__, model_names)
def collect_model_downloads() -> Tuple[DownloadSet, DownloadSet]: