triton-python-model/config.pbtxt

31 lines
317 B
Plaintext

# Triton backend to use
backend: "pytorch"
platform: "pytorch_libtorch"
max_batch_size: 0
input [
{
name: "modi"
data_type: TYPE_STRING
dims: [ -1 ]
},
{
name: "ch-be"
data_type: TYPE_STRING
dims: [ -1 ]
}
]
output [
]
instance_group [
{
kind: KIND_AUTO
count: 1
}
]