Adding ONNX file of this model
#3
by
Gulugulu0810
- opened
- README.md +27 -18
- onnx/config.json +32 -0
- onnx/decoder_model.onnx +3 -0
- onnx/decoder_model_merged.onnx +3 -0
- onnx/decoder_with_past_model.onnx +3 -0
- onnx/encoder_model.onnx +3 -0
- onnx/generation_config.json +9 -0
README.md
CHANGED
|
@@ -5,30 +5,39 @@ language:
|
|
| 5 |
- en
|
| 6 |
tags:
|
| 7 |
- translation
|
|
|
|
| 8 |
license: apache-2.0
|
| 9 |
datasets:
|
| 10 |
- ccmatrix
|
| 11 |
metrics:
|
| 12 |
- sacrebleu
|
| 13 |
widget:
|
| 14 |
-
|
| 15 |
-
|
| 16 |
-
|
| 17 |
-
|
| 18 |
-
|
| 19 |
-
|
| 20 |
-
|
| 21 |
-
|
| 22 |
-
|
| 23 |
-
|
| 24 |
-
|
| 25 |
-
|
| 26 |
-
|
| 27 |
-
|
| 28 |
-
|
| 29 |
-
|
| 30 |
-
|
| 31 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 32 |
---
|
| 33 |
|
| 34 |
# T5 English, Russian and Chinese multilingual machine translation
|
|
|
|
| 5 |
- en
|
| 6 |
tags:
|
| 7 |
- translation
|
| 8 |
+
- onnx
|
| 9 |
license: apache-2.0
|
| 10 |
datasets:
|
| 11 |
- ccmatrix
|
| 12 |
metrics:
|
| 13 |
- sacrebleu
|
| 14 |
widget:
|
| 15 |
+
- example_title: translate zh-ru
|
| 16 |
+
text: 'translate to ru: 开发的目的是为用户提供个人同步翻译。
|
| 17 |
+
|
| 18 |
+
'
|
| 19 |
+
- example_title: translate ru-en
|
| 20 |
+
text: 'translate to en: Цель разработки — предоставить пользователям личного синхронного
|
| 21 |
+
переводчика.
|
| 22 |
+
|
| 23 |
+
'
|
| 24 |
+
- example_title: translate en-ru
|
| 25 |
+
text: 'translate to ru: The purpose of the development is to provide users with
|
| 26 |
+
a personal synchronized interpreter.
|
| 27 |
+
|
| 28 |
+
'
|
| 29 |
+
- example_title: translate en-zh
|
| 30 |
+
text: 'translate to zh: The purpose of the development is to provide users with
|
| 31 |
+
a personal synchronized interpreter.
|
| 32 |
+
|
| 33 |
+
'
|
| 34 |
+
- example_title: translate zh-en
|
| 35 |
+
text: 'translate to en: 开发的目的是为用户提供个人同步解释器。
|
| 36 |
+
|
| 37 |
+
'
|
| 38 |
+
- example_title: translate ru-zh
|
| 39 |
+
text: 'translate to zh: Цель разработки — предоставить пользователям личного синхронного
|
| 40 |
+
переводчика.'
|
| 41 |
---
|
| 42 |
|
| 43 |
# T5 English, Russian and Chinese multilingual machine translation
|
onnx/config.json
ADDED
|
@@ -0,0 +1,32 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"_attn_implementation_autoset": true,
|
| 3 |
+
"architectures": [
|
| 4 |
+
"T5ForConditionalGeneration"
|
| 5 |
+
],
|
| 6 |
+
"classifier_dropout": 0.0,
|
| 7 |
+
"d_ff": 1024,
|
| 8 |
+
"d_kv": 64,
|
| 9 |
+
"d_model": 512,
|
| 10 |
+
"decoder_start_token_id": 0,
|
| 11 |
+
"dense_act_fn": "gelu_new",
|
| 12 |
+
"dropout_rate": 0.1,
|
| 13 |
+
"eos_token_id": 1,
|
| 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 |
+
"num_decoder_layers": 8,
|
| 21 |
+
"num_heads": 6,
|
| 22 |
+
"num_layers": 8,
|
| 23 |
+
"output_past": true,
|
| 24 |
+
"pad_token_id": 0,
|
| 25 |
+
"relative_attention_max_distance": 128,
|
| 26 |
+
"relative_attention_num_buckets": 32,
|
| 27 |
+
"tie_word_embeddings": false,
|
| 28 |
+
"torch_dtype": "float32",
|
| 29 |
+
"transformers_version": "4.51.3",
|
| 30 |
+
"use_cache": true,
|
| 31 |
+
"vocab_size": 65100
|
| 32 |
+
}
|
onnx/decoder_model.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:af5dd30524f82f13bb1b0cd9bd23d63db6e279caca0669955bbe1a0b547132bb
|
| 3 |
+
size 367626170
|
onnx/decoder_model_merged.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:b5bf8b5093207169ec83ca9359c8130459b8a1435b50da08daf70e306c036de2
|
| 3 |
+
size 367869537
|
onnx/decoder_with_past_model.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:391453e32e95b803fd4aea7263c8e465dc6b77df9ce88997b613ce6f3dc84da3
|
| 3 |
+
size 355019871
|
onnx/encoder_model.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a15a0af111190342695419ca23ac50ffb8bd124b4b2b0ea7ca45478934faeb70
|
| 3 |
+
size 208987123
|
onnx/generation_config.json
ADDED
|
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"decoder_start_token_id": 0,
|
| 3 |
+
"eos_token_id": 1,
|
| 4 |
+
"max_new_tokens": 1024,
|
| 5 |
+
"num_beams": 5,
|
| 6 |
+
"pad_token_id": 0,
|
| 7 |
+
"repetition_penalty": 5.0,
|
| 8 |
+
"transformers_version": "4.51.3"
|
| 9 |
+
}
|