feat: experimentally add TensorRT runtime

ssube · Apr 29, 2023 · f420a3a · f420a3a
1 parent fbab26f
commit f420a3a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/api/onnx_web/server/load.py b/api/onnx_web/server/load.py
@@ -67,6 +67,7 @@
     "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
@@ -218,6 +218,7 @@ export const I18N_STRINGS_EN = {
         directml: 'DirectML',
         nvidia: 'Nvidia GPU',
         rocm: 'ROCm',
+        tensorrt: 'TensorRT',
       },
       setting: {
         connectServer: 'Connect',