diff --git a/config.pbtxt b/config.pbtxt index f8b3301..c33e1e1 100644 --- a/config.pbtxt +++ b/config.pbtxt @@ -1,6 +1,6 @@ -# Triton Backend for TensorFlowGraphdef. -backend: "tensorflow" -platform: "tensorflow_graphdef" +# Triton backend to use +backend: "pytorch" +platform: "pytorch_libtorch" max_batch_size: 0 input [