diff --git a/config.pbtxt b/config.pbtxt index 2984b50..f557638 100644 --- a/config.pbtxt +++ b/config.pbtxt @@ -1,6 +1,5 @@ # Triton backend to use -backend: "pytorch" -platform: "pytorch_libtorch" +backend: "python" max_batch_size: 0 input [ @@ -8,6 +7,11 @@ input [ name: "rm-new-regi" data_type: TYPE_STRING dims: [ -1 ] + }, + { + name: "be-to-py" + data_type: TYPE_STRING + dims: [ -1 ] } ] @@ -15,6 +19,12 @@ output [ ] +parameters: [ + { + key: "model_path", + value: {string_value: "/cheetah/input/model/groupuser/triton-python-model"} + } +] instance_group [