CalvinBackup/facefusion
1
0
Fork 0
mirror of https://github.com/facefusion/facefusion.git synced 2026-06-06 20:53:54 +02:00
Code Issues Packages Projects Releases Wiki Activity

prevent countless importlib.import_module calls

Browse Source
This commit is contained in:
henryruhs
2026-05-26 08:49:55 +02:00
parent c2cb048ac7
commit 25df9aee48
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
facefusion/inference_manager.py
+5 -3
View File
@@ -1,5 +1,6 @@
import importlib
import random
from functools import lru_cache
from time import sleep, time
from typing import List
@@ -25,7 +26,7 @@ def get_inference_pool(module_name : str, model_names : List[str], model_source_
while process_manager.is_checking():
sleep(0.5)
execution_device_ids = state_manager.get_item('execution_device_ids')
execution_providers = resolve_execution_providers(module_name)
execution_providers = resolve_static_execution_providers(module_name)
app_context = detect_app_context()
for execution_device_id in execution_device_ids:
@@ -55,7 +56,7 @@ def create_inference_pool(model_source_set : DownloadSet, execution_device_id :
def clear_inference_pool(module_name : str, model_names : List[str]) -> None:
execution_device_ids = state_manager.get_item('execution_device_ids')
execution_providers = resolve_execution_providers(module_name)
execution_providers = resolve_static_execution_providers(module_name)
app_context = detect_app_context()
if is_windows() and has_execution_provider('directml'):
@@ -87,7 +88,8 @@ def get_inference_context(module_name : str, model_names : List[str], execution_
return inference_context
def resolve_execution_providers(module_name : str) -> List[ExecutionProvider]:
@lru_cache()
def resolve_static_execution_providers(module_name : str) -> List[ExecutionProvider]:
module = importlib.import_module(module_name)
if hasattr(module, 'resolve_execution_providers'):