diff --git a/config.pbtxt b/config.pbtxt index f0e9ff0..08b7114 100644 --- a/config.pbtxt +++ b/config.pbtxt @@ -1,5 +1,6 @@ # Triton backend to use -backend: "python" +backend: "onnxruntime" +platform: "onnxruntime_onnx" max_batch_size: 0 input [ @@ -7,6 +8,11 @@ input [ name: "modi" data_type: TYPE_STRING dims: [ -1 ] + }, + { + name: "onnx" + data_type: TYPE_UINT16 + dims: [ 12 ] } ] @@ -14,12 +20,6 @@ output [ ] -parameters: [ - { - key: "model_path", - value: {string_value: "/cheetah/input/model/groupuser/triton-python-model"} - } -] instance_group [