MrMinister
commited on
Commit
•
985facc
1
Parent(s):
4ff767f
Update generation_config.json
Browse filesAllow "return_timestamps": "word" by adding alignment_heads
- generation_config.json +1 -0
generation_config.json
CHANGED
@@ -3,6 +3,7 @@
|
|
3 |
220,
|
4 |
50257
|
5 |
],
|
|
|
6 |
"bos_token_id": 50257,
|
7 |
"decoder_start_token_id": 50258,
|
8 |
"eos_token_id": 50257,
|
|
|
3 |
220,
|
4 |
50257
|
5 |
],
|
6 |
+
"alignment_heads": [[2, 2], [3, 0], [3, 2], [3, 3], [3, 4], [3, 5]],
|
7 |
"bos_token_id": 50257,
|
8 |
"decoder_start_token_id": 50258,
|
9 |
"eos_token_id": 50257,
|