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

fix directml again

Browse Source
This commit is contained in:
henryruhs
2026-08-04 21:01:35 +02:00
parent a939990e2f
commit ed8756f977
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
facefusion/inference_manager.py
+3 -3
View File
@@ -31,9 +31,6 @@ def get_inference_pool(module_name : str, model_names : List[str], model_source_
has_arena_leak = has_execution_provider('cuda') and get_onnxruntime_version() > (1, 24, 4)
app_context = detect_app_context()
if is_windows() and has_execution_provider('directml'):
INFERENCE_POOL_SET[app_context].clear()
for execution_device_id in execution_device_ids:
inference_context = get_inference_context(module_name, model_names, execution_device_id, execution_providers)
@@ -68,6 +65,9 @@ def clear_inference_pool(module_name : str, model_names : List[str]) -> None:
execution_providers = state_manager.get_item('execution_providers')
app_context = detect_app_context()
if is_windows() and has_execution_provider('directml'):
INFERENCE_POOL_SET[app_context].clear()
for execution_device_id in execution_device_ids:
inference_context = get_inference_context(module_name, model_names, execution_device_id, execution_providers)