whisper-medium.en / generation_config.json
patrickvonplaten's picture
Correct long-form generation config parameters 'max_initial_timestamp_index' and 'prev_sot_token_id'. (#16)
2e98eb6 verified
{
"alignment_heads": [
[
11,
4
],
[
14,
1
],
[
14,
12
],
[
14,
14
],
[
15,
4
],
[
16,
0
],
[
16,
4
],
[
16,
9
],
[
17,
12
],
[
17,
14
],
[
18,
7
],
[
18,
10
],
[
18,
15
],
[
20,
0
],
[
20,
3
],
[
20,
9
],
[
20,
14
],
[
21,
12
]
],
"begin_suppress_tokens": [
220,
50256
],
"bos_token_id": 50257,
"decoder_start_token_id": 50257,
"eos_token_id": 50256,
"forced_decoder_ids": [
[
1,
50362
]
],
"is_multilingual": false,
"max_initial_timestamp_index": 50,
"max_length": 448,
"no_timestamps_token_id": 50362,
"pad_token_id": 50256,
"prev_sot_token_id": 50360,
"return_timestamps": false,
"suppress_tokens": [
1,
2,
7,
8,
9,
10,
14,
25,
26,
27,
28,
29,
31,
58,
59,
60,
61,
62,
63,
90,
91,
92,
93,
357,
366,
438,
532,
685,
705,
796,
930,
1058,
1220,
1267,
1279,
1303,
1343,
1377,
1391,
1635,
1782,
1875,
2162,
2361,
2488,
3467,
4008,
4211,
4600,
4808,
5299,
5855,
6329,
7203,
9609,
9959,
10563,
10786,
11420,
11709,
11907,
13163,
13697,
13700,
14808,
15306,
16410,
16791,
17992,
19203,
19510,
20724,
22305,
22935,
27007,
30109,
30420,
33409,
34949,
40283,
40493,
40549,
47282,
49146,
50257,
50357,
50358,
50359,
50360,
50361
],
"transformers_version": "4.31.0.dev0"
}