From ab05bf3ff32d03d85e42d2796036215a263fdede Mon Sep 17 00:00:00 2001 From: Arthur Zucker Date: Wed, 7 Dec 2022 12:09:13 +0000 Subject: [PATCH] Update config Fill in the correct start token ids --- config.json | 10 +++------- 1 file changed, 3 insertions(+), 7 deletions(-) diff --git a/config.json b/config.json index 0318c90..cafb9ee 100644 --- a/config.json +++ b/config.json @@ -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 ] ],