Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 17:24:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

fix onnx execution providers

Browse Source 
Signed-off-by: Vladimir Mandic <mandic00@live.com>
This commit is contained in:
Vladimir Mandic
2026-06-22 16:22:11 +02:00
parent cd8d65e5db
commit b697f33d76
2 changed files with 9 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/onnx_impl/execution_providers.py
+7 -1
View File
@@ -12,6 +12,7 @@ class ExecutionProvider(str, Enum):
MIGraphX = "MIGraphXExecutionProvider"
OpenVINO = "OpenVINOExecutionProvider"
Azure = "AzureExecutionProvider"
TensorRT = "TensorrtExecutionProvider"
EP_TO_NAME = {
@@ -32,8 +33,13 @@ TORCH_DEVICE_TO_EP = {
}
from modules.onnx_impl import ort
available_execution_providers = []
if ort is not None:
available_execution_providers: list[ExecutionProvider] = [ExecutionProvider(ep) for ep in ort.get_available_providers()]
for provider in ort.get_available_providers():
try:
available_execution_providers.append(ExecutionProvider(provider))
except Exception as e:
log.warning(f"ONNX: provider{provider} {e}")
else:
available_execution_providers = []
modules/ui_control_elements.py
+2 -2
View File
@@ -300,13 +300,13 @@ def create_ui_elements(units, result_txt, output_gallery):
settings.append(gr.Checkbox(label="Boost", value=False))
settings.append(gr.Slider(label="Near threshold", minimum=0.0, maximum=1.0, step=0.01, value=0.0))
settings.append(gr.Slider(label="Depth threshold", minimum=0.0, maximum=1.0, step=0.01, value=0.0))
with gr.Accordion('MediaPipe Face (Legacy)', open=True, elem_classes=['processor-settings']):
with gr.Accordion('MediaPipe Face', open=True, elem_classes=['processor-settings']):
settings.append(gr.Slider(label="Max faces", minimum=1, maximum=10, step=1, value=1))
settings.append(gr.Slider(label="Face confidence", minimum=0.0, maximum=1.0, step=0.01, value=0.5))
with gr.Accordion('Canny', open=True, elem_classes=['processor-settings']):
settings.append(gr.Slider(label="Low threshold", minimum=0, maximum=1000, step=1, value=100))
settings.append(gr.Slider(label="High threshold", minimum=0, maximum=1000, step=1, value=200))
with gr.Accordion('DWPose (Legacy)', open=True, elem_classes=['processor-settings']):
with gr.Accordion('DWPose', open=True, elem_classes=['processor-settings']):
settings.append(gr.Radio(label="Pose Model", choices=['Tiny', 'Medium', 'Large'], value='Tiny'))
settings.append(gr.Slider(label="Pose confidence", minimum=0.0, maximum=1.0, step=0.01, value=0.3))
with gr.Accordion('SegmentAnything 1.0', open=True, elem_classes=['processor-settings']):