Commit
·
be22169
1
Parent(s):
ddb562e
edited 3
Browse files- generation_config.json +2 -0
generation_config.json
CHANGED
|
@@ -38,6 +38,8 @@
|
|
| 38 |
]
|
| 39 |
],
|
| 40 |
"bos_token_id": 50257,
|
|
|
|
|
|
|
| 41 |
"decoder_start_token_id": 50258,
|
| 42 |
"eos_token_id": 50257,
|
| 43 |
"forced_decoder_ids": [
|
|
|
|
| 38 |
]
|
| 39 |
],
|
| 40 |
"bos_token_id": 50257,
|
| 41 |
+
"decoder_layers": 6,
|
| 42 |
+
"decoder_attention_heads": 8,
|
| 43 |
"decoder_start_token_id": 50258,
|
| 44 |
"eos_token_id": 50257,
|
| 45 |
"forced_decoder_ids": [
|