From a4a47bc663c8b30e014f6d3bd884ea2da78b5e39 Mon Sep 17 00:00:00 2001 From: Arthur Zucker Date: Wed, 7 Dec 2022 12:07:36 +0000 Subject: [PATCH] Add forced decoder ids (#2) - Add forced decoder ids (463445a5528ffa20c59e7f4edf6d34faa6657a67) Co-authored-by: Sanchit Gandhi --- config.json | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) diff --git a/config.json b/config.json index d8008f9..0318c90 100644 --- a/config.json +++ b/config.json @@ -23,6 +23,24 @@ "encoder_layerdrop": 0.0, "encoder_layers": 32, "eos_token_id": 50256, + "forced_decoder_ids": [ + [ + 1, + 50258 + ], + [ + 2, + 50259 + ], + [ + 3, + 50359 + ], + [ + 4, + 50363 + ] + ], "init_std": 0.02, "is_encoder_decoder": true, "max_source_positions": 1500,