feat: experimentally add TensorRT runtime

2023-04-29 15:55:58 -05:00 · 2023-04-29 15:55:58 -05:00 · f420a3a612
parent fbab26fe31
commit f420a3a612
2 changed files with 2 additions and 0 deletions
--- 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,
--- 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',