diff --git a/modules/onnx_impl/execution_providers.py b/modules/onnx_impl/execution_providers.py index 10d0bc53f..e493dffca 100644 --- a/modules/onnx_impl/execution_providers.py +++ b/modules/onnx_impl/execution_providers.py @@ -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 = [] diff --git a/modules/ui_control_elements.py b/modules/ui_control_elements.py index fc84ea07f..461476a5d 100644 --- a/modules/ui_control_elements.py +++ b/modules/ui_control_elements.py @@ -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']):