Upload MBartForConditionalGeneration
Browse files- config.json +1 -0
config.json
CHANGED
|
@@ -52,6 +52,7 @@
|
|
| 52 |
"scale_embedding": true,
|
| 53 |
"static_position_embeddings": false,
|
| 54 |
"tokenizer_class": "MBart50Tokenizer",
|
|
|
|
| 55 |
"transformers_version": "4.31.0",
|
| 56 |
"use_cache": true,
|
| 57 |
"vocab_size": 250057
|
|
|
|
| 52 |
"scale_embedding": true,
|
| 53 |
"static_position_embeddings": false,
|
| 54 |
"tokenizer_class": "MBart50Tokenizer",
|
| 55 |
+
"torch_dtype": "float32",
|
| 56 |
"transformers_version": "4.31.0",
|
| 57 |
"use_cache": true,
|
| 58 |
"vocab_size": 250057
|