Update generation config with word-level alignment heads

#21
by sanchit-gandhi HF staff - opened
Files changed (1) hide show
  1. generation_config.json +36 -2
generation_config.json CHANGED
@@ -1,4 +1,38 @@
1
  {
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2
  "begin_suppress_tokens": [
3
  220,
4
  50257
@@ -217,5 +251,5 @@
217
  "transcribe": 50359,
218
  "translate": 50358
219
  },
220
- "transformers_version": "4.27.0.dev0"
221
- }
 
1
  {
2
+ "alignment_heads": [
3
+ [
4
+ 3,
5
+ 1
6
+ ],
7
+ [
8
+ 4,
9
+ 2
10
+ ],
11
+ [
12
+ 4,
13
+ 3
14
+ ],
15
+ [
16
+ 4,
17
+ 7
18
+ ],
19
+ [
20
+ 5,
21
+ 1
22
+ ],
23
+ [
24
+ 5,
25
+ 2
26
+ ],
27
+ [
28
+ 5,
29
+ 4
30
+ ],
31
+ [
32
+ 5,
33
+ 6
34
+ ]
35
+ ],
36
  "begin_suppress_tokens": [
37
  220,
38
  50257
 
251
  "transcribe": 50359,
252
  "translate": 50358
253
  },
254
+ "transformers_version": "4.31.0.dev0"
255
+ }