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_layerdrop": 0.0,
|
||||||
"encoder_layers": 32,
|
"encoder_layers": 32,
|
||||||
"eos_token_id": 50256,
|
"eos_token_id": 50256,
|
||||||
|
"forced_decoder_ids": [
|
||||||
|
[
|
||||||
|
1,
|
||||||
|
50258
|
||||||
|
],
|
||||||
|
[
|
||||||
|
2,
|
||||||
|
50259
|
||||||
|
],
|
||||||
|
[
|
||||||
|
3,
|
||||||
|
50359
|
||||||
|
],
|
||||||
|
[
|
||||||
|
4,
|
||||||
|
50363
|
||||||
|
]
|
||||||
|
],
|
||||||
"init_std": 0.02,
|
"init_std": 0.02,
|
||||||
"is_encoder_decoder": true,
|
"is_encoder_decoder": true,
|
||||||
"max_source_positions": 1500,
|
"max_source_positions": 1500,
|
||||||
|
|
Loading…
Reference in New Issue