Hugging Face's logo

tomasmcm
/
ibm-research-knowgl-large-onnx

Text2Text Generation
Transformers.js
ONNX
English
bart
406M
Model card Files Community
tomasmcm commited on
Commit
951c049
·
verified ·
1 Parent(s): 417dbef

Upload folder using huggingface_hub

Browse files
Files changed (41) hide show
  1. added_tokens.json +3 -0
  2. config.json +62 -0
  3. generation_config.json +12 -0
  4. merges.txt +0 -0
  5. onnx/decoder_model.onnx +3 -0
  6. onnx/decoder_model_bnb4.onnx +3 -0
  7. onnx/decoder_model_fp16.onnx +3 -0
  8. onnx/decoder_model_int8.onnx +3 -0
  9. onnx/decoder_model_merged.onnx +3 -0
  10. onnx/decoder_model_merged_bnb4.onnx +3 -0
  11. onnx/decoder_model_merged_fp16.onnx +3 -0
  12. onnx/decoder_model_merged_int8.onnx +3 -0
  13. onnx/decoder_model_merged_q4.onnx +3 -0
  14. onnx/decoder_model_merged_q4f16.onnx +3 -0
  15. onnx/decoder_model_merged_quantized.onnx +3 -0
  16. onnx/decoder_model_merged_uint8.onnx +3 -0
  17. onnx/decoder_model_q4.onnx +3 -0
  18. onnx/decoder_model_q4f16.onnx +3 -0
  19. onnx/decoder_model_quantized.onnx +3 -0
  20. onnx/decoder_model_uint8.onnx +3 -0
  21. onnx/decoder_with_past_model.onnx +3 -0
  22. onnx/decoder_with_past_model_bnb4.onnx +3 -0
  23. onnx/decoder_with_past_model_fp16.onnx +3 -0
  24. onnx/decoder_with_past_model_int8.onnx +3 -0
  25. onnx/decoder_with_past_model_q4.onnx +3 -0
  26. onnx/decoder_with_past_model_q4f16.onnx +3 -0
  27. onnx/decoder_with_past_model_quantized.onnx +3 -0
  28. onnx/decoder_with_past_model_uint8.onnx +3 -0
  29. onnx/encoder_model.onnx +3 -0
  30. onnx/encoder_model_bnb4.onnx +3 -0
  31. onnx/encoder_model_fp16.onnx +3 -0
  32. onnx/encoder_model_int8.onnx +3 -0
  33. onnx/encoder_model_q4.onnx +3 -0
  34. onnx/encoder_model_q4f16.onnx +3 -0
  35. onnx/encoder_model_quantized.onnx +3 -0
  36. onnx/encoder_model_uint8.onnx +3 -0
  37. quantize_config.json +18 -0
  38. special_tokens_map.json +54 -0
  39. tokenizer.json +0 -0
  40. tokenizer_config.json +73 -0
  41. vocab.json +0 -0
added_tokens.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ {
2
+ "<>": 50265
3
+ }
config.json ADDED
@@ -0,0 +1,62 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_attn_implementation_autoset": true,
3
+ "_name_or_path": "ibm-research/knowgl-large",
4
+ "activation_dropout": 0.1,
5
+ "activation_function": "gelu",
6
+ "add_bias_logits": false,
7
+ "add_final_layer_norm": false,
8
+ "architectures": [
9
+ "BartForConditionalGeneration"
10
+ ],
11
+ "attention_dropout": 0.1,
12
+ "bos_token_id": 0,
13
+ "classif_dropout": 0.1,
14
+ "classifier_dropout": 0.0,
15
+ "d_model": 1024,
16
+ "decoder_attention_heads": 16,
17
+ "decoder_ffn_dim": 4096,
18
+ "decoder_layerdrop": 0.0,
19
+ "decoder_layers": 12,
20
+ "decoder_start_token_id": 0,
21
+ "dropout": 0.1,
22
+ "encoder_attention_heads": 16,
23
+ "encoder_ffn_dim": 4096,
24
+ "encoder_layerdrop": 0.0,
25
+ "encoder_layers": 12,
26
+ "eos_token_id": 2,
27
+ "forced_eos_token_id": 2,
28
+ "gradient_checkpointing": false,
29
+ "id2label": {
30
+ "0": "LABEL_0",
31
+ "1": "LABEL_1",
32
+ "2": "LABEL_2"
33
+ },
34
+ "init_std": 0.02,
35
+ "is_encoder_decoder": true,
36
+ "label2id": {
37
+ "LABEL_0": 0,
38
+ "LABEL_1": 1,
39
+ "LABEL_2": 2
40
+ },
41
+ "max_length": null,
42
+ "max_position_embeddings": 1024,
43
+ "model_type": "bart",
44
+ "normalize_before": false,
45
+ "num_beams": null,
46
+ "num_hidden_layers": 12,
47
+ "pad_token_id": 1,
48
+ "scale_embedding": false,
49
+ "task_specific_params": {
50
+ "triple_generation": {
51
+ "length_penalty": 0.0,
52
+ "max_length": 1024,
53
+ "min_length": 12,
54
+ "no_repeat_ngram_size": 0,
55
+ "num_beams": 2
56
+ }
57
+ },
58
+ "torch_dtype": "float32",
59
+ "transformers_version": "4.49.0",
60
+ "use_cache": true,
61
+ "vocab_size": 50265
62
+ }
generation_config.json ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "bos_token_id": 0,
4
+ "decoder_start_token_id": 0,
5
+ "eos_token_id": 2,
6
+ "forced_bos_token_id": 0,
7
+ "forced_eos_token_id": 2,
8
+ "max_length": 1024,
9
+ "num_beams": 4,
10
+ "pad_token_id": 1,
11
+ "transformers_version": "4.49.0"
12
+ }
merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
onnx/decoder_model.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3ad0db051925807ab1af6e4cd9d41024566b09ddcba2b6ef20d4d529af98cc4a
3
+ size 1222755882
onnx/decoder_model_bnb4.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:af7a5700d2c9b65bc21effff6429cad1d7fb5c99717ab7366adc2c92b18b3d51
3
+ size 353781895
onnx/decoder_model_fp16.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d22646297476105e2b4d43eed3186450317a495873f222b18d41fb18920e3045
3
+ size 611647661
onnx/decoder_model_int8.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:11184d040b0e699a3c4b327dea72e48526b79e4762ecc251afe79b67d55aed6f
3
+ size 307994003
onnx/decoder_model_merged.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4e2f0b8ca08890e881ee91d342763842396e6fab8120f394ab3bcb0303ebdca9
3
+ size 1223609351
onnx/decoder_model_merged_bnb4.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c74549cd46c900c240819ca9830f3dc7fe410fbbc708e704672a3e3aeb2f14fa
3
+ size 354649570
onnx/decoder_model_merged_fp16.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:032ef603e6e0650a8c9f89ff2aa9aa72ad822f59a587fcb1b32d5a8c594cdcb3
3
+ size 612416183
onnx/decoder_model_merged_int8.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:84b88e27f229b0eedc821a7c14dfaa4f7773ba4fe0644bc2ce8dbe509ab0c2c1
3
+ size 308969126
onnx/decoder_model_merged_q4.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:73a92fc29c06832888585effdda3dc72a253c31115d276330118a2ac1f32f4d5
3
+ size 370447589
onnx/decoder_model_merged_q4f16.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ac6fd07bde4a9a741d1ebfa2175ea3c2ee87656b3300f711a71d21964291ad32
3
+ size 249050452
onnx/decoder_model_merged_quantized.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:84b88e27f229b0eedc821a7c14dfaa4f7773ba4fe0644bc2ce8dbe509ab0c2c1
3
+ size 308969126
onnx/decoder_model_merged_uint8.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a219d372e99b2d431b314cccb517b373a1e970fb9d68915e8addbc8969d9e669
3
+ size 308969126
onnx/decoder_model_q4.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3686a31d179419775a4856ffccf10a8d580bfde622021d0ef449d57d50cd5f6a
3
+ size 369580787
onnx/decoder_model_q4f16.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e95f3eef5cf37cff76e4082a9a8864494de8c66abdd385936b0a241e972ff362
3
+ size 248268597
onnx/decoder_model_quantized.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:11184d040b0e699a3c4b327dea72e48526b79e4762ecc251afe79b67d55aed6f
3
+ size 307994003
onnx/decoder_model_uint8.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:65d553cfe706a0ad47b40bd8a13540b4c12fbb30ea3d836636824019e54fd5a8
3
+ size 307994003
onnx/decoder_with_past_model.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b79bc981628a73847cd71d5df16dc235ea0fee7a6460f83c39d228d383d36fb7
3
+ size 1121862890
onnx/decoder_with_past_model_bnb4.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a520673cf317ae44dcfa82ef62b7c303b5440234188ab3e7eba4bad44c76ff9f
3
+ size 339392679
onnx/decoder_with_past_model_fp16.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fc1d0f23ee4ba221d0c2beb818d9b91318539be6c4e816d7497e93eb81a57cb8
3
+ size 561144965
onnx/decoder_with_past_model_int8.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9736f6e256fd85f128e3c81d88c38d7272300009783c909763bf7aa6638be202
3
+ size 282451888
onnx/decoder_with_past_model_q4.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4aaecb90b09790cc34fe834c561504dcce2b2e516e35c4114333ab5d9c2584e6
3
+ size 353618899
onnx/decoder_with_past_model_q4f16.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fff681b2f64fa83830b6e27d3eea329325c8bdac0e5a131bf86c712c57e6f9d0
3
+ size 233938221
onnx/decoder_with_past_model_quantized.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9736f6e256fd85f128e3c81d88c38d7272300009783c909763bf7aa6638be202
3
+ size 282451888
onnx/decoder_with_past_model_uint8.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:00ce519555309bcc47df2caca0eec30735be0a37e469ee909060d0a2cbbe85b2
3
+ size 282451888
onnx/encoder_model.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a25666e6805c1caf5b7b862ab81cac48579837de6ea03931279c557fdb63217e
3
+ size 814954023
onnx/encoder_model_bnb4.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:17abc7341cbff5434f9e521684030e145cc01b688b327ffadd3e3cf333165227
3
+ size 295920248
onnx/encoder_model_fp16.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:23bcab584eff56a989d4cf973cf62bd3566d4e55f50dd2bbc66e3efd2d34a1b0
3
+ size 407596951
onnx/encoder_model_int8.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f55300a14917b74b48a711d601b8a8eab34aef2dadf4a36a073756a334f4f218
3
+ size 205030270
onnx/encoder_model_q4.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bc625ae0064da2a4077eb62a3f9ebc24ccdc1f70ef7d310823c7c3b81344ba30
3
+ size 305356844
onnx/encoder_model_q4f16.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b544d87b32326d7ac0cc18acfa446387d3b26c7e52efa7b91a6e338839552a2a
3
+ size 190552475
onnx/encoder_model_quantized.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f55300a14917b74b48a711d601b8a8eab34aef2dadf4a36a073756a334f4f218
3
+ size 205030270
onnx/encoder_model_uint8.onnx ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d4d2356adcd3a3300dd85e24ab4b43a3d3e51a229240c442c536e93702ec3766
3
+ size 205030270
quantize_config.json ADDED
@@ -0,0 +1,18 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "modes": [
3
+ "fp16",
4
+ "q8",
5
+ "int8",
6
+ "uint8",
7
+ "q4",
8
+ "q4f16",
9
+ "bnb4"
10
+ ],
11
+ "per_channel": true,
12
+ "reduce_range": true,
13
+ "block_size": null,
14
+ "is_symmetric": true,
15
+ "accuracy_level": null,
16
+ "quant_type": 1,
17
+ "op_block_list": null
18
+ }
special_tokens_map.json ADDED
@@ -0,0 +1,54 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "additional_special_tokens": [
3
+ "<>"
4
+ ],
5
+ "bos_token": {
6
+ "content": "<s>",
7
+ "lstrip": false,
8
+ "normalized": false,
9
+ "rstrip": false,
10
+ "single_word": false
11
+ },
12
+ "cls_token": {
13
+ "content": "<s>",
14
+ "lstrip": false,
15
+ "normalized": false,
16
+ "rstrip": false,
17
+ "single_word": false
18
+ },
19
+ "eos_token": {
20
+ "content": "</s>",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false
25
+ },
26
+ "mask_token": {
27
+ "content": "<mask>",
28
+ "lstrip": true,
29
+ "normalized": false,
30
+ "rstrip": false,
31
+ "single_word": false
32
+ },
33
+ "pad_token": {
34
+ "content": "<pad>",
35
+ "lstrip": false,
36
+ "normalized": false,
37
+ "rstrip": false,
38
+ "single_word": false
39
+ },
40
+ "sep_token": {
41
+ "content": "</s>",
42
+ "lstrip": false,
43
+ "normalized": false,
44
+ "rstrip": false,
45
+ "single_word": false
46
+ },
47
+ "unk_token": {
48
+ "content": "<unk>",
49
+ "lstrip": false,
50
+ "normalized": false,
51
+ "rstrip": false,
52
+ "single_word": false
53
+ }
54
+ }
tokenizer.json ADDED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json ADDED
@@ -0,0 +1,73 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_prefix_space": false,
3
+ "added_tokens_decoder": {
4
+ "0": {
5
+ "content": "<s>",
6
+ "lstrip": false,
7
+ "normalized": false,
8
+ "rstrip": false,
9
+ "single_word": false,
10
+ "special": true
11
+ },
12
+ "1": {
13
+ "content": "<pad>",
14
+ "lstrip": false,
15
+ "normalized": false,
16
+ "rstrip": false,
17
+ "single_word": false,
18
+ "special": true
19
+ },
20
+ "2": {
21
+ "content": "</s>",
22
+ "lstrip": false,
23
+ "normalized": false,
24
+ "rstrip": false,
25
+ "single_word": false,
26
+ "special": true
27
+ },
28
+ "3": {
29
+ "content": "<unk>",
30
+ "lstrip": false,
31
+ "normalized": false,
32
+ "rstrip": false,
33
+ "single_word": false,
34
+ "special": true
35
+ },
36
+ "50264": {
37
+ "content": "<mask>",
38
+ "lstrip": true,
39
+ "normalized": false,
40
+ "rstrip": false,
41
+ "single_word": false,
42
+ "special": true
43
+ },
44
+ "50265": {
45
+ "content": "<>",
46
+ "lstrip": false,
47
+ "normalized": false,
48
+ "rstrip": false,
49
+ "single_word": false,
50
+ "special": true
51
+ }
52
+ },
53
+ "additional_special_tokens": [
54
+ "<>"
55
+ ],
56
+ "bos_token": "<s>",
57
+ "clean_up_tokenization_spaces": false,
58
+ "cls_token": "<s>",
59
+ "eos_token": "</s>",
60
+ "errors": "replace",
61
+ "extra_special_tokens": {},
62
+ "mask_token": "<mask>",
63
+ "max_length": 1024,
64
+ "model_max_length": 1024,
65
+ "pad_token": "<pad>",
66
+ "sep_token": "</s>",
67
+ "stride": 0,
68
+ "tokenizer_class": "BartTokenizer",
69
+ "trim_offsets": true,
70
+ "truncation_side": "right",
71
+ "truncation_strategy": "longest_first",
72
+ "unk_token": "<unk>"
73
+ }
vocab.json ADDED
The diff for this file is too large to render. See raw diff