Instructions to use akreal/tiny-random-mbart with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use akreal/tiny-random-mbart with Transformers:
# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("akreal/tiny-random-mbart", dtype="auto") - Notebooks
- Google Colab
- Kaggle
Add decoder_start_token_id
Browse files- config.json +1 -0
config.json
CHANGED
|
@@ -9,6 +9,7 @@
|
|
| 9 |
"decoder_ffn_dim": 4,
|
| 10 |
"decoder_layerdrop": 0.0,
|
| 11 |
"decoder_layers": 2,
|
|
|
|
| 12 |
"dropout": 0.1,
|
| 13 |
"encoder_attention_heads": 4,
|
| 14 |
"encoder_ffn_dim": 4,
|
|
|
|
| 9 |
"decoder_ffn_dim": 4,
|
| 10 |
"decoder_layerdrop": 0.0,
|
| 11 |
"decoder_layers": 2,
|
| 12 |
+
"decoder_start_token_id": 2,
|
| 13 |
"dropout": 0.1,
|
| 14 |
"encoder_attention_heads": 4,
|
| 15 |
"encoder_ffn_dim": 4,
|