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