diff --git a/config.pbtxt b/config.pbtxt index 71ef2ce..b7b6acc 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 [ @@ -13,6 +12,11 @@ input [ name: "new-regen-regi" data_type: TYPE_STRING dims: [ -1 ] + }, + { + name: "ch-to-python" + data_type: TYPE_STRING + dims: [ -1 ] } ] @@ -20,6 +24,12 @@ output [ ] +parameters: [ + { + key: "model_path", + value: {string_value: "/cheetah/input/model/groupuser/triton-python-model"} + } +] instance_group [