Upload 11 files
Browse files- .gitattributes +1 -0
- README.md +25 -3
- config.json +33 -0
- onnx/decoder_model_merged_quantized.onnx +3 -0
- onnx/decoder_model_quantized.onnx +3 -0
- onnx/decoder_with_past_model_quantized.onnx +3 -0
- onnx/encoder_model_quantized.onnx +3 -0
- ort_config.json +33 -0
- special_tokens_map.json +23 -0
- spiece.model +3 -0
- tokenizer.json +3 -0
- tokenizer_config.json +40 -0
.gitattributes
CHANGED
@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
|
|
33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
|
|
|
33 |
*.zip filter=lfs diff=lfs merge=lfs -text
|
34 |
*.zst filter=lfs diff=lfs merge=lfs -text
|
35 |
*tfevents* filter=lfs diff=lfs merge=lfs -text
|
36 |
+
tokenizer.json filter=lfs diff=lfs merge=lfs -text
|
README.md
CHANGED
@@ -1,3 +1,25 @@
|
|
1 |
-
---
|
2 |
-
license: apache-2.0
|
3 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
license: apache-2.0
|
3 |
+
base_model:
|
4 |
+
- google/madlad400-3b-mt
|
5 |
+
pipeline_tag: translation
|
6 |
+
tags:
|
7 |
+
- text2text-generation
|
8 |
+
- text-generation-inference
|
9 |
+
---
|
10 |
+
|
11 |
+
This is same model as https://huggingface.co/ISoloist1/madlad400-3b-mt-onnx but with folder structure optimized for Transformers.js
|
12 |
+
|
13 |
+
## ONNX version of [google/madlad400-3b-mt](https://huggingface.co/google/madlad400-3b-mt)
|
14 |
+
|
15 |
+
## Converted and quantized with [optimum-cli](https://github.com/huggingface/optimum)
|
16 |
+
|
17 |
+
- Convert to ONNX:
|
18 |
+
```sh
|
19 |
+
optimum-cli onnxruntime export --model google/madlad400-3b-mt <output_path> --legacy
|
20 |
+
```
|
21 |
+
|
22 |
+
- Quantization:
|
23 |
+
```sh
|
24 |
+
optimum-cli onnxruntime quantize --onnx_model <input_model_path> -o <output_model_path> --avx512_vnni
|
25 |
+
```
|
config.json
ADDED
@@ -0,0 +1,33 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"_attn_implementation_autoset": true,
|
3 |
+
"architectures": [
|
4 |
+
"T5ForConditionalGeneration"
|
5 |
+
],
|
6 |
+
"classifier_dropout": 0.0,
|
7 |
+
"d_ff": 8192,
|
8 |
+
"d_kv": 128,
|
9 |
+
"d_model": 1024,
|
10 |
+
"decoder_start_token_id": 0,
|
11 |
+
"dense_act_fn": "gelu_new",
|
12 |
+
"dropout_rate": 0.1,
|
13 |
+
"eos_token_id": 2,
|
14 |
+
"feed_forward_proj": "gated-gelu",
|
15 |
+
"initializer_factor": 1.0,
|
16 |
+
"is_encoder_decoder": true,
|
17 |
+
"is_gated_act": true,
|
18 |
+
"layer_norm_epsilon": 1e-06,
|
19 |
+
"model_type": "t5",
|
20 |
+
"n_positions": 512,
|
21 |
+
"num_decoder_layers": 32,
|
22 |
+
"num_heads": 16,
|
23 |
+
"num_layers": 32,
|
24 |
+
"output_past": true,
|
25 |
+
"pad_token_id": 1,
|
26 |
+
"relative_attention_max_distance": 128,
|
27 |
+
"relative_attention_num_buckets": 32,
|
28 |
+
"task_specific_params": {},
|
29 |
+
"tie_word_embeddings": false,
|
30 |
+
"transformers_version": "4.51.3",
|
31 |
+
"use_cache": true,
|
32 |
+
"vocab_size": 256000
|
33 |
+
}
|
onnx/decoder_model_merged_quantized.onnx
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:6bd545f6b67ae96ff21d207c7b630b3df7a97a2379b433e060d2a5ea2a4a9a1d
|
3 |
+
size 1870804464
|
onnx/decoder_model_quantized.onnx
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:1894294807c530df6a17e773fe8753833866838fcd41084bdd21cad17e0b50c1
|
3 |
+
size 1868927672
|
onnx/decoder_with_past_model_quantized.onnx
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:2661e101fc310d6eefed50ca272e094b464cb886eaa000f2fafe6ad64ca941ec
|
3 |
+
size 1734505050
|
onnx/encoder_model_quantized.onnx
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:3ab4690aab2174ca29a9ce30cdb8152d9b639606cf2cf48e77112314e064e393
|
3 |
+
size 1337248123
|
ort_config.json
ADDED
@@ -0,0 +1,33 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"one_external_file": true,
|
3 |
+
"opset": null,
|
4 |
+
"optimization": {},
|
5 |
+
"quantization": {
|
6 |
+
"activations_dtype": "QUInt8",
|
7 |
+
"activations_symmetric": false,
|
8 |
+
"format": "QOperator",
|
9 |
+
"is_static": false,
|
10 |
+
"mode": "IntegerOps",
|
11 |
+
"nodes_to_exclude": [],
|
12 |
+
"nodes_to_quantize": [],
|
13 |
+
"operators_to_quantize": [
|
14 |
+
"Conv",
|
15 |
+
"MatMul",
|
16 |
+
"Attention",
|
17 |
+
"LSTM",
|
18 |
+
"Gather",
|
19 |
+
"Transpose",
|
20 |
+
"EmbedLayerNormalization"
|
21 |
+
],
|
22 |
+
"per_channel": false,
|
23 |
+
"qdq_add_pair_to_weight": false,
|
24 |
+
"qdq_dedicated_pair": false,
|
25 |
+
"qdq_op_type_per_channel_support_to_axis": {
|
26 |
+
"MatMul": 1
|
27 |
+
},
|
28 |
+
"reduce_range": false,
|
29 |
+
"weights_dtype": "QInt8",
|
30 |
+
"weights_symmetric": true
|
31 |
+
},
|
32 |
+
"use_external_data_format": false
|
33 |
+
}
|
special_tokens_map.json
ADDED
@@ -0,0 +1,23 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"eos_token": {
|
3 |
+
"content": "</s>",
|
4 |
+
"lstrip": false,
|
5 |
+
"normalized": false,
|
6 |
+
"rstrip": false,
|
7 |
+
"single_word": false
|
8 |
+
},
|
9 |
+
"pad_token": {
|
10 |
+
"content": "<s>",
|
11 |
+
"lstrip": false,
|
12 |
+
"normalized": false,
|
13 |
+
"rstrip": false,
|
14 |
+
"single_word": false
|
15 |
+
},
|
16 |
+
"unk_token": {
|
17 |
+
"content": "<unk>",
|
18 |
+
"lstrip": false,
|
19 |
+
"normalized": false,
|
20 |
+
"rstrip": false,
|
21 |
+
"single_word": false
|
22 |
+
}
|
23 |
+
}
|
spiece.model
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ef11ac9a22c7503492f56d48dce53be20e339b63605983e9f27d2cd0e0f3922c
|
3 |
+
size 4427844
|
tokenizer.json
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:03f5d7dc88da0cb4bb6b7a1d9d66ee62f5bd339ef0aaaf6e89d74829df5830c0
|
3 |
+
size 16613995
|
tokenizer_config.json
ADDED
@@ -0,0 +1,40 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
{
|
2 |
+
"add_prefix_space": null,
|
3 |
+
"added_tokens_decoder": {
|
4 |
+
"0": {
|
5 |
+
"content": "<unk>",
|
6 |
+
"lstrip": false,
|
7 |
+
"normalized": false,
|
8 |
+
"rstrip": false,
|
9 |
+
"single_word": false,
|
10 |
+
"special": true
|
11 |
+
},
|
12 |
+
"1": {
|
13 |
+
"content": "<s>",
|
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 |
+
},
|
29 |
+
"additional_special_tokens": [],
|
30 |
+
"clean_up_tokenization_spaces": true,
|
31 |
+
"eos_token": "</s>",
|
32 |
+
"extra_ids": 0,
|
33 |
+
"extra_special_tokens": {},
|
34 |
+
"legacy": false,
|
35 |
+
"model_max_length": 1000000000000000019884624838656,
|
36 |
+
"pad_token": "<s>",
|
37 |
+
"sp_model_kwargs": {},
|
38 |
+
"tokenizer_class": "T5Tokenizer",
|
39 |
+
"unk_token": "<unk>"
|
40 |
+
}
|