diff --git a/config.json b/config.json index 9e9121d..2150571 100644 --- a/config.json +++ b/config.json @@ -8,11 +8,7 @@ "decoder_ffn_dim": 1536, "decoder_layerdrop": 0.0, "decoder_layers": 4, - "decoder_start_token_id": [ - 50258, - 50259, - 50359 - ], + "decoder_start_token_id": 50257, "dropout": 0.0, "encoder_attention_heads": 6, "encoder_ffn_dim": 1536,