From ea0df867e197689949617a8e73cb2387bac119c6 Mon Sep 17 00:00:00 2001 From: harisreedhar Date: Thu, 12 Feb 2026 13:28:10 +0530 Subject: [PATCH] fix cache --- facefusion/execution.py | 7 +++---- facefusion/system.py | 6 +++--- 2 files changed, 6 insertions(+), 7 deletions(-) diff --git a/facefusion/execution.py b/facefusion/execution.py index 2d768c16..b5a02de6 100644 --- a/facefusion/execution.py +++ b/facefusion/execution.py @@ -8,7 +8,6 @@ import pynvml import onnxruntime import facefusion.choices -from facefusion import state_manager from facefusion.system import detect_static_graphic_devices from facefusion.types import ExecutionProvider, InferenceSessionProvider @@ -40,7 +39,7 @@ def create_inference_providers(execution_device_id : int, execution_providers : inference_providers.append((facefusion.choices.execution_provider_set.get(execution_provider), { 'device_id': execution_device_id, - 'cudnn_conv_algo_search': resolve_cudnn_conv_algo_search() + 'cudnn_conv_algo_search': resolve_cudnn_conv_algo_search(execution_providers) })) if execution_provider == 'tensorrt': @@ -113,8 +112,8 @@ def resolve_cache_path() -> str: return os.path.join('.caches', onnxruntime.get_version_string()) -def resolve_cudnn_conv_algo_search() -> str: - execution_devices = detect_static_graphic_devices(state_manager.get_item('execution_providers')) +def resolve_cudnn_conv_algo_search(execution_providers : List[ExecutionProvider]) -> str: + execution_devices = detect_static_graphic_devices(tuple(execution_providers)) product_names = ('GeForce GTX 1630', 'GeForce GTX 1650', 'GeForce GTX 1660') for execution_device in execution_devices: diff --git a/facefusion/system.py b/facefusion/system.py index 61a8438b..7fe1bd7f 100644 --- a/facefusion/system.py +++ b/facefusion/system.py @@ -2,7 +2,7 @@ import importlib import shutil from functools import lru_cache from pathlib import Path -from typing import List +from typing import List, Tuple import psutil @@ -24,11 +24,11 @@ def get_metrics_set() -> Metrics: @lru_cache() -def detect_static_graphic_devices(execution_providers : List[ExecutionProvider]) -> List[GraphicDevice]: +def detect_static_graphic_devices(execution_providers : Tuple[ExecutionProvider, ...]) -> List[GraphicDevice]: return detect_graphic_devices(execution_providers) -def detect_graphic_devices(execution_providers : List[ExecutionProvider]) -> List[GraphicDevice]: +def detect_graphic_devices(execution_providers : Tuple[ExecutionProvider, ...]) -> List[GraphicDevice]: if any(execution_provider in [ 'rocm', 'migraphx' ] for execution_provider in execution_providers): return detect_amd_graphic_devices() if any(execution_provider in [ 'cuda', 'tensorrt' ] for execution_provider in execution_providers):