config.pbtxt 354 Bytes
Newer Older
Nikhilesh Bhatnagar's avatar
Nikhilesh Bhatnagar committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31
name: "model_name"
backend: "python"
max_batch_size: 512

input [
  {
    name: "INPUT_TEXT_TOKENIZED"
    data_type: TYPE_STRING
    dims: [ 1 ]
  }
]
output [
  {
    name: "OUTPUT_TEXT"
    data_type: TYPE_STRING
    dims: [ 1 ]
  }
]

dynamic_batching {}

instance_group [
  {
    count: 1
    kind: KIND_GPU
  }
]

response_cache {
  enable: true
}