Add forced decoder ids (#2)
- Add forced decoder ids (463445a5528ffa20c59e7f4edf6d34faa6657a67) Co-authored-by: Sanchit Gandhi <sanchit-gandhi@users.noreply.huggingface.co>
This commit is contained in:
parent
468057c79f
commit
a4a47bc663
18
config.json
18
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,
|
||||
|
|
Loading…
Reference in New Issue