MrMinister commited on
Commit
985facc
1 Parent(s): 4ff767f

Update generation_config.json

Browse files

Allow "return_timestamps": "word" by adding alignment_heads

Files changed (1) hide show
  1. 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,