Add forced decoder ids (#2)

- Add forced decoder ids (463445a5528ffa20c59e7f4edf6d34faa6657a67) Co-authored-by: Sanchit Gandhi <sanchit-gandhi@users.noreply.huggingface.co>
2022-12-07 12:07:36 +00:00 · 2022-12-07 12:07:36 +00:00 · a4a47bc663
parent 468057c79f
commit a4a47bc663
1 changed files with 18 additions and 0 deletions
--- 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,