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_ffn_dim": 5120,
"decoder_layerdrop": 0.0, "decoder_layerdrop": 0.0,
"decoder_layers": 32, "decoder_layers": 32,
"decoder_start_token_id": 50257, "decoder_start_token_id": 50258,
"dropout": 0.0, "dropout": 0.0,
"encoder_attention_heads": 20, "encoder_attention_heads": 20,
"encoder_ffn_dim": 5120, "encoder_ffn_dim": 5120,
@ -26,18 +26,14 @@
"forced_decoder_ids": [ "forced_decoder_ids": [
[ [
1, 1,
50258
],
[
2,
50259 50259
], ],
[ [
3, 2,
50359 50359
], ],
[ [
4, 3,
50363 50363
] ]
], ],