diff --git a/config.pbtxt b/config.pbtxt index a738e22..c33e1e1 100644 --- a/config.pbtxt +++ b/config.pbtxt @@ -1,5 +1,6 @@ # Triton backend to use -backend: "python" +backend: "pytorch" +platform: "pytorch_libtorch" max_batch_size: 0 input [ @@ -9,12 +10,6 @@ output [ ] -parameters: [ - { - key: "model_path", - value: {string_value: "/cheetah/input/model/groupuser/triton-python-model"} - } -] instance_group [