nbroad HF staff commited on
Commit
34b5041
1 Parent(s): a6e1756

Update onnx_config.pbtxt

Browse files
Files changed (1) hide show
  1. onnx_config.pbtxt +3 -3
onnx_config.pbtxt CHANGED
@@ -5,19 +5,19 @@ 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: "logits"
19
  data_type: TYPE_FP32
20
- dims: [ 2 ]
21
  }
22
  ]
23
  instance_group [
 
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 [
17
  {
18
  name: "logits"
19
  data_type: TYPE_FP32
20
+ dims: [ -1, 2 ]
21
  }
22
  ]
23
  instance_group [