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

Update onnx_config.pbtext

Browse files
Files changed (1) hide show
  1. onnx_config.pbtext +3 -3
onnx_config.pbtext CHANGED
@@ -1,16 +1,16 @@
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: [ -1, 512 ]
9
  },
10
  {
11
  name: "attention_mask"
12
  data_type: TYPE_INT64
13
- dims: [ -1, 512 ]
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 [