Adding ONNX file of this model

#2
by chenyaoshun - opened
onnx/config.json ADDED
@@ -0,0 +1,79 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "HillZhang/real_learner_bart_CGEC_exam",
3
+ "activation_dropout": 0.1,
4
+ "activation_function": "gelu",
5
+ "add_bias_logits": false,
6
+ "add_final_layer_norm": false,
7
+ "architectures": [
8
+ "BartForConditionalGeneration"
9
+ ],
10
+ "attention_dropout": 0.1,
11
+ "bos_token_id": 0,
12
+ "classif_dropout": 0.1,
13
+ "classifier_dropout": 0.0,
14
+ "d_model": 1024,
15
+ "decoder_attention_heads": 16,
16
+ "decoder_ffn_dim": 4096,
17
+ "decoder_layerdrop": 0.0,
18
+ "decoder_layers": 12,
19
+ "decoder_start_token_id": 2,
20
+ "do_blenderbot_90_layernorm": false,
21
+ "dropout": 0.1,
22
+ "early_stopping": true,
23
+ "encoder_attention_heads": 16,
24
+ "encoder_ffn_dim": 4096,
25
+ "encoder_layerdrop": 0.0,
26
+ "encoder_layers": 12,
27
+ "eos_token_id": 2,
28
+ "extra_pos_embeddings": 2,
29
+ "force_bos_token_to_be_generated": true,
30
+ "forced_bos_token_id": 0,
31
+ "forced_eos_token_id": 2,
32
+ "gradient_checkpointing": false,
33
+ "id2label": {
34
+ "0": "LABEL_0",
35
+ "1": "LABEL_1",
36
+ "2": "LABEL_2"
37
+ },
38
+ "init_std": 0.02,
39
+ "is_encoder_decoder": true,
40
+ "label2id": {
41
+ "LABEL_0": 0,
42
+ "LABEL_1": 1,
43
+ "LABEL_2": 2
44
+ },
45
+ "max_position_embeddings": 512,
46
+ "model_type": "bart",
47
+ "no_repeat_ngram_size": 3,
48
+ "normalize_before": false,
49
+ "normalize_embedding": true,
50
+ "num_beams": 4,
51
+ "num_hidden_layers": 12,
52
+ "pad_token_id": 1,
53
+ "scale_embedding": false,
54
+ "static_position_embeddings": false,
55
+ "task_specific_params": {
56
+ "summarization": {
57
+ "length_penalty": 1.0,
58
+ "max_length": 128,
59
+ "min_length": 12,
60
+ "num_beams": 4
61
+ },
62
+ "summarization_cnn": {
63
+ "length_penalty": 2.0,
64
+ "max_length": 142,
65
+ "min_length": 56,
66
+ "num_beams": 4
67
+ },
68
+ "summarization_xsum": {
69
+ "length_penalty": 1.0,
70
+ "max_length": 62,
71
+ "min_length": 11,
72
+ "num_beams": 6
73
+ }
74
+ },
75
+ "transformers_version": "4.30.2",
76
+ "unk_token_id": 3,
77
+ "use_cache": true,
78
+ "vocab_size": 21132
79
+ }
onnx/decoder_model.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:821df80eef973cc7a949a772692e0a3492e308b0ac46dfceb19430431a4bd707
3
+ size 982110039
onnx/decoder_model_merged.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b9a9451f97beae33aef3c6432b0b960408278ca26d3045593edda65e7a507923
3
+ size 982592991
onnx/decoder_with_past_model.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dc2e892ea0e8e92a164296f62b0cd4869d02e143eb51f483d041a477b1a0c642
3
+ size 881268682
onnx/encoder_model.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:42f3be907e0ee1b8d99e28b5b226cdf154f9b674281cc339ba837bf0fa882bd9
3
+ size 693520433
onnx/generation_config.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 0,
4
+ "decoder_start_token_id": 2,
5
+ "early_stopping": true,
6
+ "eos_token_id": 2,
7
+ "forced_bos_token_id": 0,
8
+ "forced_eos_token_id": 2,
9
+ "no_repeat_ngram_size": 3,
10
+ "num_beams": 4,
11
+ "pad_token_id": 1,
12
+ "transformers_version": "4.30.2"
13
+ }