add model
Browse files- config.json +5 -4
config.json
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
{
|
2 |
-
"_name_or_path": "/
|
3 |
"_num_labels": 3,
|
4 |
"activation_dropout": 0.0,
|
5 |
"activation_function": "gelu",
|
@@ -50,8 +50,9 @@
|
|
50 |
"pad_token_id": 1,
|
51 |
"scale_embedding": true,
|
52 |
"static_position_embeddings": false,
|
53 |
-
"
|
|
|
|
|
54 |
"use_cache": true,
|
55 |
-
"vocab_size": 250054
|
56 |
-
"tokenizer_class": "MBart50Tokenizer"
|
57 |
}
|
|
|
1 |
{
|
2 |
+
"_name_or_path": "facebook/mbart-large-50",
|
3 |
"_num_labels": 3,
|
4 |
"activation_dropout": 0.0,
|
5 |
"activation_function": "gelu",
|
|
|
50 |
"pad_token_id": 1,
|
51 |
"scale_embedding": true,
|
52 |
"static_position_embeddings": false,
|
53 |
+
"tokenizer_class": "MBart50Tokenizer",
|
54 |
+
"torch_dtype": "float16",
|
55 |
+
"transformers_version": "4.20.0",
|
56 |
"use_cache": true,
|
57 |
+
"vocab_size": 250054
|
|
|
58 |
}
|