Update config

Fill in the correct start token ids
This commit is contained in:
Arthur Zucker 2022-12-07 12:09:13 +00:00 committed by huggingface-web
parent a4a47bc663
commit ab05bf3ff3
1 changed files with 3 additions and 7 deletions

View File

@ -16,7 +16,7 @@
"decoder_ffn_dim": 5120,
"decoder_layerdrop": 0.0,
"decoder_layers": 32,
"decoder_start_token_id": 50257,
"decoder_start_token_id": 50258,
"dropout": 0.0,
"encoder_attention_heads": 20,
"encoder_ffn_dim": 5120,
@ -26,18 +26,14 @@
"forced_decoder_ids": [
[
1,
50258
],
[
2,
50259
],
[
3,
2,
50359
],
[
4,
3,
50363
]
],