diff --git a/api/onnx_web/server/load.py b/api/onnx_web/server/load.py index 352d7239..1d33aabb 100644 --- a/api/onnx_web/server/load.py +++ b/api/onnx_web/server/load.py @@ -67,6 +67,7 @@ platform_providers = { "cuda": "CUDAExecutionProvider", "directml": "DmlExecutionProvider", "rocm": "ROCMExecutionProvider", + "tensorrt": "TensorRTExecutionProvider", } source_filters = { "canny": source_filter_canny, diff --git a/gui/src/strings/en.ts b/gui/src/strings/en.ts index 3cedea48..170cb1cd 100644 --- a/gui/src/strings/en.ts +++ b/gui/src/strings/en.ts @@ -218,6 +218,7 @@ export const I18N_STRINGS_EN = { directml: 'DirectML', nvidia: 'Nvidia GPU', rocm: 'ROCm', + tensorrt: 'TensorRT', }, setting: { connectServer: 'Connect',