triton-python-model/config.pbtxt

50 lines
617 B
Plaintext

# Triton backend to use
backend: "python"
max_batch_size: 0
input [
{
name: "input0"
data_type: TYPE_BOOL
dims: [ 1 ]
reshape {
shape: [ -2 ]
}
},
{
name: "input1"
data_type: TYPE_UINT64
dims: [ 1, 100 ]
reshape {
shape: [ 150 ]
}
}
]
output [
{
name: "output0"
data_type: TYPE_UINT8
dims: [ 120 ]
reshape {
shape: [ 120, 130 ]
}
}
]
parameters: [
{
key: "model_path",
value: {string_value: "/cheetah/input/model/groupuser/triton-python-model"}
}
]
instance_group [
{
kind: KIND_AUTO
count: 1
}
]