nbroad commited on
Commit
50b9337
·
verified ·
1 Parent(s): 8342bc7

Update onnx_config.pbtext

Browse files
Files changed (1) hide show
  1. onnx_config.pbtext +4 -4
onnx_config.pbtext CHANGED
@@ -1,23 +1,23 @@
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 [
17
  {
18
  name: "output"
19
  data_type: TYPE_FP32
20
- dims: [ -1, 2 ]
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 {