model_repository/sam_torchscript_fp32/config.pbtxt

name: "sam_torchscript_fp32"
platform: "pytorch_libtorch"
max_batch_size: 1
dynamic_batching {}
instance_group [
  {
    count: 1
    kind: KIND_GPU
  }
]
input [
  {
    name: "INPUT__0"
    data_type: TYPE_FP32
    dims: [ 3, 1024, 1024 ]
  }
]
output [
  {
    name: "OUTPUT__0"
    data_type: TYPE_FP32
    dims: [ 256, 64, 64 ]
  }
]