Update onnx_config.pbtext
Browse files- onnx_config.pbtext +4 -4
onnx_config.pbtext
CHANGED
@@ -1,23 +1,23 @@
|
|
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: [
|
9 |
},
|
10 |
{
|
11 |
name: "attention_mask"
|
12 |
data_type: TYPE_INT64
|
13 |
-
dims: [
|
14 |
}
|
15 |
]
|
16 |
output [
|
17 |
{
|
18 |
name: "output"
|
19 |
data_type: TYPE_FP32
|
20 |
-
dims: [
|
21 |
}
|
22 |
]
|
23 |
dynamic_batching {
|
|
|
1 |
name: "deberta"
|
2 |
platform: "onnxruntime_onnx"
|
3 |
+
max_batch_size: 8
|
4 |
input [
|
5 |
{
|
6 |
name: "input_ids"
|
7 |
data_type: TYPE_INT64
|
8 |
+
dims: [ 512 ]
|
9 |
},
|
10 |
{
|
11 |
name: "attention_mask"
|
12 |
data_type: TYPE_INT64
|
13 |
+
dims: [ 512 ]
|
14 |
}
|
15 |
]
|
16 |
output [
|
17 |
{
|
18 |
name: "output"
|
19 |
data_type: TYPE_FP32
|
20 |
+
dims: [ 2 ]
|
21 |
}
|
22 |
]
|
23 |
dynamic_batching {
|