saburq's picture
add model and config
507f4c6 verified
{
"model_name": "grounded",
"input_names": [
"img",
"input_ids",
"attention_mask",
"position_ids",
"token_type_ids",
"text_token_mask"
],
"output_names": [
"logits",
"boxes"
],
"input_shapes": {
"img": [
"dynamic",
3,
"dynamic",
"dynamic"
],
"input_ids": [
"dynamic",
"dynamic"
],
"attention_mask": [
"dynamic",
"dynamic"
],
"position_ids": [
"dynamic",
"dynamic"
],
"token_type_ids": [
"dynamic",
"dynamic"
],
"text_token_mask": [
"dynamic",
"dynamic",
"dynamic"
]
},
"output_shapes": {
"logits": [
"dynamic",
"dynamic",
"dynamic"
],
"boxes": [
"dynamic",
"dynamic",
4
]
}
}