Update onnx_config.pbtext
Browse files- onnx_config.pbtext +2 -2
onnx_config.pbtext
CHANGED
|
@@ -4,12 +4,12 @@ max_batch_size: 8
|
|
| 4 |
input [
|
| 5 |
{
|
| 6 |
name: "input_ids"
|
| 7 |
-
data_type:
|
| 8 |
dims: [ -1, 512 ]
|
| 9 |
},
|
| 10 |
{
|
| 11 |
name: "attention_mask"
|
| 12 |
-
data_type:
|
| 13 |
dims: [ -1, 512 ]
|
| 14 |
}
|
| 15 |
]
|
|
|
|
| 4 |
input [
|
| 5 |
{
|
| 6 |
name: "input_ids"
|
| 7 |
+
data_type: TYPE_INT64
|
| 8 |
dims: [ -1, 512 ]
|
| 9 |
},
|
| 10 |
{
|
| 11 |
name: "attention_mask"
|
| 12 |
+
data_type: TYPE_INT64
|
| 13 |
dims: [ -1, 512 ]
|
| 14 |
}
|
| 15 |
]
|