apextoaster
/
onnx-web
1
0
Fork 0
Code Actions

feat: experimentally add TensorRT runtime

This commit is contained in:
Sean Sube 2023-04-29 15:55:58 -05:00
parent fbab26fe31
commit f420a3a612
Signed by: ssube
GPG Key ID: 3EED7B957D362AF1
2 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
api/onnx_web/server/load.py
View File

@ -67,6 +67,7 @@ platform_providers = {
"cuda": "CUDAExecutionProvider", "cuda": "CUDAExecutionProvider",
"directml": "DmlExecutionProvider", "directml": "DmlExecutionProvider",
"rocm": "ROCMExecutionProvider", "rocm": "ROCMExecutionProvider",
"tensorrt": "TensorRTExecutionProvider",
} }
source_filters = { source_filters = {
"canny": source_filter_canny, "canny": source_filter_canny,

1
gui/src/strings/en.ts
View File

@ -218,6 +218,7 @@ export const I18N_STRINGS_EN = {
directml: 'DirectML', directml: 'DirectML',
nvidia: 'Nvidia GPU', nvidia: 'Nvidia GPU',
rocm: 'ROCm', rocm: 'ROCm',
tensorrt: 'TensorRT',
}, },
setting: { setting: {
connectServer: 'Connect', connectServer: 'Connect',