Hugging Face's logo

google
/
flan-ul2

Text2Text Generation
Transformers
PyTorch
t5
flan-ul2
text-generation-inference
Inference Endpoints
Model card Files Community
31

Upload FlaxT5ForConditionalGeneration

#11
by ArthurZ HF staff - opened
base: refs/heads/main
←
from: refs/pr/11
Discussion Files changed
+782
-3
Files changed (11) hide show
  1. config.json +3 -3
  2. flax_model-00001-of-00008.msgpack +3 -0
  3. flax_model-00002-of-00008.msgpack +3 -0
  4. flax_model-00003-of-00008.msgpack +3 -0
  5. flax_model-00004-of-00008.msgpack +3 -0
  6. flax_model-00005-of-00008.msgpack +3 -0
  7. flax_model-00006-of-00008.msgpack +3 -0
  8. flax_model-00007-of-00008.msgpack +3 -0
  9. flax_model-00008-of-00008.msgpack +3 -0
  10. flax_model.msgpack.index.json +748 -0
  11. generation_config.json +7 -0
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "/home/younes_huggingface_co/code/ul2/flan-ul2",
3
  "architectures": [
4
  "T5ForConditionalGeneration"
5
  ],
@@ -24,8 +24,8 @@
24
  "pad_token_id": 0,
25
  "relative_attention_max_distance": 128,
26
  "relative_attention_num_buckets": 32,
27
- "torch_dtype": "bfloat16",
28
- "transformers_version": "4.25.0.dev0",
29
  "use_cache": true,
30
  "vocab_size": 32128
31
  }
 
1
  {
2
+ "_name_or_path": "./google/flan-ul2",
3
  "architectures": [
4
  "T5ForConditionalGeneration"
5
  ],
 
24
  "pad_token_id": 0,
25
  "relative_attention_max_distance": 128,
26
  "relative_attention_num_buckets": 32,
27
+ "torch_dtype": "float32",
28
+ "transformers_version": "4.27.0.dev0",
29
  "use_cache": true,
30
  "vocab_size": 32128
31
  }
flax_model-00001-of-00008.msgpack ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:42d5688b908fa1d6ca9a622ba8707fa4ef273a9083ce10eb5c01d003b858e85d
3
+ size 9921910246
flax_model-00002-of-00008.msgpack ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:81f4fae8cb9ecf5666749344c123327d5620c3babdc19789c82968d7fcb890e9
3
+ size 9999535748
flax_model-00003-of-00008.msgpack ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d2f1a161ddd25e0a36ddb170036b3f4ebb97aa01c8329fb1becc9045a01a232c
3
+ size 9865318028
flax_model-00004-of-00008.msgpack ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:031b1c9957082ae678eefaac9e77a8b3c21d8d8054685ba5da45fc32783873d5
3
+ size 9932462146
flax_model-00005-of-00008.msgpack ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:519599ceb211b0d71e9ba38107b9508691d89190a21f01670b32dd5a2b2bc4ef
3
+ size 9999585600
flax_model-00006-of-00008.msgpack ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b382b4545a4d94915f88515076c871be470702b0525c0f47632e24050f10c976
3
+ size 9865384436
flax_model-00007-of-00008.msgpack ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4e91a2f53f0be6f82d3799f61ce273dcd5e17ce9809898072e4c601c5adbe9ae
3
+ size 9932460256
flax_model-00008-of-00008.msgpack ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e97e55f44ce3c9b0e77f25f858f1c057e6287df29119cb17929e7f723b627abc
3
+ size 8321830579
flax_model.msgpack.index.json ADDED
@@ -0,0 +1,748 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "metadata": {
3
+ "total_size": 77838454784
4
+ },
5
+ "weight_map": {
6
+ "decoder/block/0/layer/0/SelfAttention/k/kernel": "flax_model-00004-of-00008.msgpack",
7
+ "decoder/block/0/layer/0/SelfAttention/o/kernel": "flax_model-00004-of-00008.msgpack",
8
+ "decoder/block/0/layer/0/SelfAttention/q/kernel": "flax_model-00004-of-00008.msgpack",
9
+ "decoder/block/0/layer/0/SelfAttention/relative_attention_bias/embedding": "flax_model-00004-of-00008.msgpack",
10
+ "decoder/block/0/layer/0/SelfAttention/v/kernel": "flax_model-00004-of-00008.msgpack",
11
+ "decoder/block/0/layer/0/layer_norm/weight": "flax_model-00004-of-00008.msgpack",
12
+ "decoder/block/0/layer/1/EncDecAttention/k/kernel": "flax_model-00004-of-00008.msgpack",
13
+ "decoder/block/0/layer/1/EncDecAttention/o/kernel": "flax_model-00004-of-00008.msgpack",
14
+ "decoder/block/0/layer/1/EncDecAttention/q/kernel": "flax_model-00004-of-00008.msgpack",
15
+ "decoder/block/0/layer/1/EncDecAttention/v/kernel": "flax_model-00004-of-00008.msgpack",
16
+ "decoder/block/0/layer/1/layer_norm/weight": "flax_model-00004-of-00008.msgpack",
17
+ "decoder/block/0/layer/2/DenseReluDense/wi_0/kernel": "flax_model-00004-of-00008.msgpack",
18
+ "decoder/block/0/layer/2/DenseReluDense/wi_1/kernel": "flax_model-00004-of-00008.msgpack",
19
+ "decoder/block/0/layer/2/DenseReluDense/wo/kernel": "flax_model-00004-of-00008.msgpack",
20
+ "decoder/block/0/layer/2/layer_norm/weight": "flax_model-00004-of-00008.msgpack",
21
+ "decoder/block/1/layer/0/SelfAttention/k/kernel": "flax_model-00004-of-00008.msgpack",
22
+ "decoder/block/1/layer/0/SelfAttention/o/kernel": "flax_model-00004-of-00008.msgpack",
23
+ "decoder/block/1/layer/0/SelfAttention/q/kernel": "flax_model-00004-of-00008.msgpack",
24
+ "decoder/block/1/layer/0/SelfAttention/v/kernel": "flax_model-00004-of-00008.msgpack",
25
+ "decoder/block/1/layer/0/layer_norm/weight": "flax_model-00004-of-00008.msgpack",
26
+ "decoder/block/1/layer/1/EncDecAttention/k/kernel": "flax_model-00004-of-00008.msgpack",
27
+ "decoder/block/1/layer/1/EncDecAttention/o/kernel": "flax_model-00004-of-00008.msgpack",
28
+ "decoder/block/1/layer/1/EncDecAttention/q/kernel": "flax_model-00004-of-00008.msgpack",
29
+ "decoder/block/1/layer/1/EncDecAttention/v/kernel": "flax_model-00004-of-00008.msgpack",
30
+ "decoder/block/1/layer/1/layer_norm/weight": "flax_model-00004-of-00008.msgpack",
31
+ "decoder/block/1/layer/2/DenseReluDense/wi_0/kernel": "flax_model-00004-of-00008.msgpack",
32
+ "decoder/block/1/layer/2/DenseReluDense/wi_1/kernel": "flax_model-00004-of-00008.msgpack",
33
+ "decoder/block/1/layer/2/DenseReluDense/wo/kernel": "flax_model-00004-of-00008.msgpack",
34
+ "decoder/block/1/layer/2/layer_norm/weight": "flax_model-00004-of-00008.msgpack",
35
+ "decoder/block/10/layer/0/SelfAttention/k/kernel": "flax_model-00005-of-00008.msgpack",
36
+ "decoder/block/10/layer/0/SelfAttention/o/kernel": "flax_model-00005-of-00008.msgpack",
37
+ "decoder/block/10/layer/0/SelfAttention/q/kernel": "flax_model-00005-of-00008.msgpack",
38
+ "decoder/block/10/layer/0/SelfAttention/v/kernel": "flax_model-00005-of-00008.msgpack",
39
+ "decoder/block/10/layer/0/layer_norm/weight": "flax_model-00005-of-00008.msgpack",
40
+ "decoder/block/10/layer/1/EncDecAttention/k/kernel": "flax_model-00005-of-00008.msgpack",
41
+ "decoder/block/10/layer/1/EncDecAttention/o/kernel": "flax_model-00005-of-00008.msgpack",
42
+ "decoder/block/10/layer/1/EncDecAttention/q/kernel": "flax_model-00005-of-00008.msgpack",
43
+ "decoder/block/10/layer/1/EncDecAttention/v/kernel": "flax_model-00005-of-00008.msgpack",
44
+ "decoder/block/10/layer/1/layer_norm/weight": "flax_model-00005-of-00008.msgpack",
45
+ "decoder/block/10/layer/2/DenseReluDense/wi_0/kernel": "flax_model-00005-of-00008.msgpack",
46
+ "decoder/block/10/layer/2/DenseReluDense/wi_1/kernel": "flax_model-00005-of-00008.msgpack",
47
+ "decoder/block/10/layer/2/DenseReluDense/wo/kernel": "flax_model-00005-of-00008.msgpack",
48
+ "decoder/block/10/layer/2/layer_norm/weight": "flax_model-00005-of-00008.msgpack",
49
+ "decoder/block/11/layer/0/SelfAttention/k/kernel": "flax_model-00006-of-00008.msgpack",
50
+ "decoder/block/11/layer/0/SelfAttention/o/kernel": "flax_model-00006-of-00008.msgpack",
51
+ "decoder/block/11/layer/0/SelfAttention/q/kernel": "flax_model-00005-of-00008.msgpack",
52
+ "decoder/block/11/layer/0/SelfAttention/v/kernel": "flax_model-00006-of-00008.msgpack",
53
+ "decoder/block/11/layer/0/layer_norm/weight": "flax_model-00006-of-00008.msgpack",
54
+ "decoder/block/11/layer/1/EncDecAttention/k/kernel": "flax_model-00006-of-00008.msgpack",
55
+ "decoder/block/11/layer/1/EncDecAttention/o/kernel": "flax_model-00006-of-00008.msgpack",
56
+ "decoder/block/11/layer/1/EncDecAttention/q/kernel": "flax_model-00006-of-00008.msgpack",
57
+ "decoder/block/11/layer/1/EncDecAttention/v/kernel": "flax_model-00006-of-00008.msgpack",
58
+ "decoder/block/11/layer/1/layer_norm/weight": "flax_model-00006-of-00008.msgpack",
59
+ "decoder/block/11/layer/2/DenseReluDense/wi_0/kernel": "flax_model-00006-of-00008.msgpack",
60
+ "decoder/block/11/layer/2/DenseReluDense/wi_1/kernel": "flax_model-00006-of-00008.msgpack",
61
+ "decoder/block/11/layer/2/DenseReluDense/wo/kernel": "flax_model-00006-of-00008.msgpack",
62
+ "decoder/block/11/layer/2/layer_norm/weight": "flax_model-00006-of-00008.msgpack",
63
+ "decoder/block/12/layer/0/SelfAttention/k/kernel": "flax_model-00006-of-00008.msgpack",
64
+ "decoder/block/12/layer/0/SelfAttention/o/kernel": "flax_model-00006-of-00008.msgpack",
65
+ "decoder/block/12/layer/0/SelfAttention/q/kernel": "flax_model-00006-of-00008.msgpack",
66
+ "decoder/block/12/layer/0/SelfAttention/v/kernel": "flax_model-00006-of-00008.msgpack",
67
+ "decoder/block/12/layer/0/layer_norm/weight": "flax_model-00006-of-00008.msgpack",
68
+ "decoder/block/12/layer/1/EncDecAttention/k/kernel": "flax_model-00006-of-00008.msgpack",
69
+ "decoder/block/12/layer/1/EncDecAttention/o/kernel": "flax_model-00006-of-00008.msgpack",
70
+ "decoder/block/12/layer/1/EncDecAttention/q/kernel": "flax_model-00006-of-00008.msgpack",
71
+ "decoder/block/12/layer/1/EncDecAttention/v/kernel": "flax_model-00006-of-00008.msgpack",
72
+ "decoder/block/12/layer/1/layer_norm/weight": "flax_model-00006-of-00008.msgpack",
73
+ "decoder/block/12/layer/2/DenseReluDense/wi_0/kernel": "flax_model-00006-of-00008.msgpack",
74
+ "decoder/block/12/layer/2/DenseReluDense/wi_1/kernel": "flax_model-00006-of-00008.msgpack",
75
+ "decoder/block/12/layer/2/DenseReluDense/wo/kernel": "flax_model-00006-of-00008.msgpack",
76
+ "decoder/block/12/layer/2/layer_norm/weight": "flax_model-00006-of-00008.msgpack",
77
+ "decoder/block/13/layer/0/SelfAttention/k/kernel": "flax_model-00006-of-00008.msgpack",
78
+ "decoder/block/13/layer/0/SelfAttention/o/kernel": "flax_model-00006-of-00008.msgpack",
79
+ "decoder/block/13/layer/0/SelfAttention/q/kernel": "flax_model-00006-of-00008.msgpack",
80
+ "decoder/block/13/layer/0/SelfAttention/v/kernel": "flax_model-00006-of-00008.msgpack",
81
+ "decoder/block/13/layer/0/layer_norm/weight": "flax_model-00006-of-00008.msgpack",
82
+ "decoder/block/13/layer/1/EncDecAttention/k/kernel": "flax_model-00006-of-00008.msgpack",
83
+ "decoder/block/13/layer/1/EncDecAttention/o/kernel": "flax_model-00006-of-00008.msgpack",
84
+ "decoder/block/13/layer/1/EncDecAttention/q/kernel": "flax_model-00006-of-00008.msgpack",
85
+ "decoder/block/13/layer/1/EncDecAttention/v/kernel": "flax_model-00006-of-00008.msgpack",
86
+ "decoder/block/13/layer/1/layer_norm/weight": "flax_model-00006-of-00008.msgpack",
87
+ "decoder/block/13/layer/2/DenseReluDense/wi_0/kernel": "flax_model-00006-of-00008.msgpack",
88
+ "decoder/block/13/layer/2/DenseReluDense/wi_1/kernel": "flax_model-00006-of-00008.msgpack",
89
+ "decoder/block/13/layer/2/DenseReluDense/wo/kernel": "flax_model-00006-of-00008.msgpack",
90
+ "decoder/block/13/layer/2/layer_norm/weight": "flax_model-00006-of-00008.msgpack",
91
+ "decoder/block/14/layer/0/SelfAttention/k/kernel": "flax_model-00006-of-00008.msgpack",
92
+ "decoder/block/14/layer/0/SelfAttention/o/kernel": "flax_model-00006-of-00008.msgpack",
93
+ "decoder/block/14/layer/0/SelfAttention/q/kernel": "flax_model-00006-of-00008.msgpack",
94
+ "decoder/block/14/layer/0/SelfAttention/v/kernel": "flax_model-00006-of-00008.msgpack",
95
+ "decoder/block/14/layer/0/layer_norm/weight": "flax_model-00006-of-00008.msgpack",
96
+ "decoder/block/14/layer/1/EncDecAttention/k/kernel": "flax_model-00006-of-00008.msgpack",
97
+ "decoder/block/14/layer/1/EncDecAttention/o/kernel": "flax_model-00006-of-00008.msgpack",
98
+ "decoder/block/14/layer/1/EncDecAttention/q/kernel": "flax_model-00006-of-00008.msgpack",
99
+ "decoder/block/14/layer/1/EncDecAttention/v/kernel": "flax_model-00006-of-00008.msgpack",
100
+ "decoder/block/14/layer/1/layer_norm/weight": "flax_model-00006-of-00008.msgpack",
101
+ "decoder/block/14/layer/2/DenseReluDense/wi_0/kernel": "flax_model-00006-of-00008.msgpack",
102
+ "decoder/block/14/layer/2/DenseReluDense/wi_1/kernel": "flax_model-00006-of-00008.msgpack",
103
+ "decoder/block/14/layer/2/DenseReluDense/wo/kernel": "flax_model-00006-of-00008.msgpack",
104
+ "decoder/block/14/layer/2/layer_norm/weight": "flax_model-00006-of-00008.msgpack",
105
+ "decoder/block/15/layer/0/SelfAttention/k/kernel": "flax_model-00006-of-00008.msgpack",
106
+ "decoder/block/15/layer/0/SelfAttention/o/kernel": "flax_model-00006-of-00008.msgpack",
107
+ "decoder/block/15/layer/0/SelfAttention/q/kernel": "flax_model-00006-of-00008.msgpack",
108
+ "decoder/block/15/layer/0/SelfAttention/v/kernel": "flax_model-00006-of-00008.msgpack",
109
+ "decoder/block/15/layer/0/layer_norm/weight": "flax_model-00006-of-00008.msgpack",
110
+ "decoder/block/15/layer/1/EncDecAttention/k/kernel": "flax_model-00006-of-00008.msgpack",
111
+ "decoder/block/15/layer/1/EncDecAttention/o/kernel": "flax_model-00006-of-00008.msgpack",
112
+ "decoder/block/15/layer/1/EncDecAttention/q/kernel": "flax_model-00006-of-00008.msgpack",
113
+ "decoder/block/15/layer/1/EncDecAttention/v/kernel": "flax_model-00006-of-00008.msgpack",
114
+ "decoder/block/15/layer/1/layer_norm/weight": "flax_model-00006-of-00008.msgpack",
115
+ "decoder/block/15/layer/2/DenseReluDense/wi_0/kernel": "flax_model-00006-of-00008.msgpack",
116
+ "decoder/block/15/layer/2/DenseReluDense/wi_1/kernel": "flax_model-00006-of-00008.msgpack",
117
+ "decoder/block/15/layer/2/DenseReluDense/wo/kernel": "flax_model-00006-of-00008.msgpack",
118
+ "decoder/block/15/layer/2/layer_norm/weight": "flax_model-00006-of-00008.msgpack",
119
+ "decoder/block/16/layer/0/SelfAttention/k/kernel": "flax_model-00006-of-00008.msgpack",
120
+ "decoder/block/16/layer/0/SelfAttention/o/kernel": "flax_model-00006-of-00008.msgpack",
121
+ "decoder/block/16/layer/0/SelfAttention/q/kernel": "flax_model-00006-of-00008.msgpack",
122
+ "decoder/block/16/layer/0/SelfAttention/v/kernel": "flax_model-00006-of-00008.msgpack",
123
+ "decoder/block/16/layer/0/layer_norm/weight": "flax_model-00006-of-00008.msgpack",
124
+ "decoder/block/16/layer/1/EncDecAttention/k/kernel": "flax_model-00006-of-00008.msgpack",
125
+ "decoder/block/16/layer/1/EncDecAttention/o/kernel": "flax_model-00006-of-00008.msgpack",
126
+ "decoder/block/16/layer/1/EncDecAttention/q/kernel": "flax_model-00006-of-00008.msgpack",
127
+ "decoder/block/16/layer/1/EncDecAttention/v/kernel": "flax_model-00006-of-00008.msgpack",
128
+ "decoder/block/16/layer/1/layer_norm/weight": "flax_model-00006-of-00008.msgpack",
129
+ "decoder/block/16/layer/2/DenseReluDense/wi_0/kernel": "flax_model-00006-of-00008.msgpack",
130
+ "decoder/block/16/layer/2/DenseReluDense/wi_1/kernel": "flax_model-00006-of-00008.msgpack",
131
+ "decoder/block/16/layer/2/DenseReluDense/wo/kernel": "flax_model-00006-of-00008.msgpack",
132
+ "decoder/block/16/layer/2/layer_norm/weight": "flax_model-00006-of-00008.msgpack",
133
+ "decoder/block/17/layer/0/SelfAttention/k/kernel": "flax_model-00006-of-00008.msgpack",
134
+ "decoder/block/17/layer/0/SelfAttention/o/kernel": "flax_model-00006-of-00008.msgpack",
135
+ "decoder/block/17/layer/0/SelfAttention/q/kernel": "flax_model-00006-of-00008.msgpack",
136
+ "decoder/block/17/layer/0/SelfAttention/v/kernel": "flax_model-00006-of-00008.msgpack",
137
+ "decoder/block/17/layer/0/layer_norm/weight": "flax_model-00006-of-00008.msgpack",
138
+ "decoder/block/17/layer/1/EncDecAttention/k/kernel": "flax_model-00006-of-00008.msgpack",
139
+ "decoder/block/17/layer/1/EncDecAttention/o/kernel": "flax_model-00006-of-00008.msgpack",
140
+ "decoder/block/17/layer/1/EncDecAttention/q/kernel": "flax_model-00006-of-00008.msgpack",
141
+ "decoder/block/17/layer/1/EncDecAttention/v/kernel": "flax_model-00006-of-00008.msgpack",
142
+ "decoder/block/17/layer/1/layer_norm/weight": "flax_model-00006-of-00008.msgpack",
143
+ "decoder/block/17/layer/2/DenseReluDense/wi_0/kernel": "flax_model-00006-of-00008.msgpack",
144
+ "decoder/block/17/layer/2/DenseReluDense/wi_1/kernel": "flax_model-00006-of-00008.msgpack",
145
+ "decoder/block/17/layer/2/DenseReluDense/wo/kernel": "flax_model-00006-of-00008.msgpack",
146
+ "decoder/block/17/layer/2/layer_norm/weight": "flax_model-00006-of-00008.msgpack",
147
+ "decoder/block/18/layer/0/SelfAttention/k/kernel": "flax_model-00006-of-00008.msgpack",
148
+ "decoder/block/18/layer/0/SelfAttention/o/kernel": "flax_model-00006-of-00008.msgpack",
149
+ "decoder/block/18/layer/0/SelfAttention/q/kernel": "flax_model-00006-of-00008.msgpack",
150
+ "decoder/block/18/layer/0/SelfAttention/v/kernel": "flax_model-00006-of-00008.msgpack",
151
+ "decoder/block/18/layer/0/layer_norm/weight": "flax_model-00006-of-00008.msgpack",
152
+ "decoder/block/18/layer/1/EncDecAttention/k/kernel": "flax_model-00006-of-00008.msgpack",
153
+ "decoder/block/18/layer/1/EncDecAttention/o/kernel": "flax_model-00006-of-00008.msgpack",
154
+ "decoder/block/18/layer/1/EncDecAttention/q/kernel": "flax_model-00006-of-00008.msgpack",
155
+ "decoder/block/18/layer/1/EncDecAttention/v/kernel": "flax_model-00006-of-00008.msgpack",
156
+ "decoder/block/18/layer/1/layer_norm/weight": "flax_model-00006-of-00008.msgpack",
157
+ "decoder/block/18/layer/2/DenseReluDense/wi_0/kernel": "flax_model-00007-of-00008.msgpack",
158
+ "decoder/block/18/layer/2/DenseReluDense/wi_1/kernel": "flax_model-00007-of-00008.msgpack",
159
+ "decoder/block/18/layer/2/DenseReluDense/wo/kernel": "flax_model-00007-of-00008.msgpack",
160
+ "decoder/block/18/layer/2/layer_norm/weight": "flax_model-00007-of-00008.msgpack",
161
+ "decoder/block/19/layer/0/SelfAttention/k/kernel": "flax_model-00007-of-00008.msgpack",
162
+ "decoder/block/19/layer/0/SelfAttention/o/kernel": "flax_model-00007-of-00008.msgpack",
163
+ "decoder/block/19/layer/0/SelfAttention/q/kernel": "flax_model-00007-of-00008.msgpack",
164
+ "decoder/block/19/layer/0/SelfAttention/v/kernel": "flax_model-00007-of-00008.msgpack",
165
+ "decoder/block/19/layer/0/layer_norm/weight": "flax_model-00007-of-00008.msgpack",
166
+ "decoder/block/19/layer/1/EncDecAttention/k/kernel": "flax_model-00007-of-00008.msgpack",
167
+ "decoder/block/19/layer/1/EncDecAttention/o/kernel": "flax_model-00007-of-00008.msgpack",
168
+ "decoder/block/19/layer/1/EncDecAttention/q/kernel": "flax_model-00007-of-00008.msgpack",
169
+ "decoder/block/19/layer/1/EncDecAttention/v/kernel": "flax_model-00007-of-00008.msgpack",
170
+ "decoder/block/19/layer/1/layer_norm/weight": "flax_model-00007-of-00008.msgpack",
171
+ "decoder/block/19/layer/2/DenseReluDense/wi_0/kernel": "flax_model-00007-of-00008.msgpack",
172
+ "decoder/block/19/layer/2/DenseReluDense/wi_1/kernel": "flax_model-00007-of-00008.msgpack",
173
+ "decoder/block/19/layer/2/DenseReluDense/wo/kernel": "flax_model-00007-of-00008.msgpack",
174
+ "decoder/block/19/layer/2/layer_norm/weight": "flax_model-00007-of-00008.msgpack",
175
+ "decoder/block/2/layer/0/SelfAttention/k/kernel": "flax_model-00004-of-00008.msgpack",
176
+ "decoder/block/2/layer/0/SelfAttention/o/kernel": "flax_model-00004-of-00008.msgpack",
177
+ "decoder/block/2/layer/0/SelfAttention/q/kernel": "flax_model-00004-of-00008.msgpack",
178
+ "decoder/block/2/layer/0/SelfAttention/v/kernel": "flax_model-00004-of-00008.msgpack",
179
+ "decoder/block/2/layer/0/layer_norm/weight": "flax_model-00004-of-00008.msgpack",
180
+ "decoder/block/2/layer/1/EncDecAttention/k/kernel": "flax_model-00004-of-00008.msgpack",
181
+ "decoder/block/2/layer/1/EncDecAttention/o/kernel": "flax_model-00004-of-00008.msgpack",
182
+ "decoder/block/2/layer/1/EncDecAttention/q/kernel": "flax_model-00004-of-00008.msgpack",
183
+ "decoder/block/2/layer/1/EncDecAttention/v/kernel": "flax_model-00004-of-00008.msgpack",
184
+ "decoder/block/2/layer/1/layer_norm/weight": "flax_model-00004-of-00008.msgpack",
185
+ "decoder/block/2/layer/2/DenseReluDense/wi_0/kernel": "flax_model-00004-of-00008.msgpack",
186
+ "decoder/block/2/layer/2/DenseReluDense/wi_1/kernel": "flax_model-00004-of-00008.msgpack",
187
+ "decoder/block/2/layer/2/DenseReluDense/wo/kernel": "flax_model-00004-of-00008.msgpack",
188
+ "decoder/block/2/layer/2/layer_norm/weight": "flax_model-00004-of-00008.msgpack",
189
+ "decoder/block/20/layer/0/SelfAttention/k/kernel": "flax_model-00007-of-00008.msgpack",
190
+ "decoder/block/20/layer/0/SelfAttention/o/kernel": "flax_model-00007-of-00008.msgpack",
191
+ "decoder/block/20/layer/0/SelfAttention/q/kernel": "flax_model-00007-of-00008.msgpack",
192
+ "decoder/block/20/layer/0/SelfAttention/v/kernel": "flax_model-00007-of-00008.msgpack",
193
+ "decoder/block/20/layer/0/layer_norm/weight": "flax_model-00007-of-00008.msgpack",
194
+ "decoder/block/20/layer/1/EncDecAttention/k/kernel": "flax_model-00007-of-00008.msgpack",
195
+ "decoder/block/20/layer/1/EncDecAttention/o/kernel": "flax_model-00007-of-00008.msgpack",
196
+ "decoder/block/20/layer/1/EncDecAttention/q/kernel": "flax_model-00007-of-00008.msgpack",
197
+ "decoder/block/20/layer/1/EncDecAttention/v/kernel": "flax_model-00007-of-00008.msgpack",
198
+ "decoder/block/20/layer/1/layer_norm/weight": "flax_model-00007-of-00008.msgpack",
199
+ "decoder/block/20/layer/2/DenseReluDense/wi_0/kernel": "flax_model-00007-of-00008.msgpack",
200
+ "decoder/block/20/layer/2/DenseReluDense/wi_1/kernel": "flax_model-00007-of-00008.msgpack",
201
+ "decoder/block/20/layer/2/DenseReluDense/wo/kernel": "flax_model-00007-of-00008.msgpack",
202
+ "decoder/block/20/layer/2/layer_norm/weight": "flax_model-00007-of-00008.msgpack",
203
+ "decoder/block/21/layer/0/SelfAttention/k/kernel": "flax_model-00007-of-00008.msgpack",
204
+ "decoder/block/21/layer/0/SelfAttention/o/kernel": "flax_model-00007-of-00008.msgpack",
205
+ "decoder/block/21/layer/0/SelfAttention/q/kernel": "flax_model-00007-of-00008.msgpack",
206
+ "decoder/block/21/layer/0/SelfAttention/v/kernel": "flax_model-00007-of-00008.msgpack",
207
+ "decoder/block/21/layer/0/layer_norm/weight": "flax_model-00007-of-00008.msgpack",
208
+ "decoder/block/21/layer/1/EncDecAttention/k/kernel": "flax_model-00007-of-00008.msgpack",
209
+ "decoder/block/21/layer/1/EncDecAttention/o/kernel": "flax_model-00007-of-00008.msgpack",
210
+ "decoder/block/21/layer/1/EncDecAttention/q/kernel": "flax_model-00007-of-00008.msgpack",
211
+ "decoder/block/21/layer/1/EncDecAttention/v/kernel": "flax_model-00007-of-00008.msgpack",
212
+ "decoder/block/21/layer/1/layer_norm/weight": "flax_model-00007-of-00008.msgpack",
213
+ "decoder/block/21/layer/2/DenseReluDense/wi_0/kernel": "flax_model-00007-of-00008.msgpack",
214
+ "decoder/block/21/layer/2/DenseReluDense/wi_1/kernel": "flax_model-00007-of-00008.msgpack",
215
+ "decoder/block/21/layer/2/DenseReluDense/wo/kernel": "flax_model-00007-of-00008.msgpack",
216
+ "decoder/block/21/layer/2/layer_norm/weight": "flax_model-00007-of-00008.msgpack",
217
+ "decoder/block/22/layer/0/SelfAttention/k/kernel": "flax_model-00007-of-00008.msgpack",
218
+ "decoder/block/22/layer/0/SelfAttention/o/kernel": "flax_model-00007-of-00008.msgpack",
219
+ "decoder/block/22/layer/0/SelfAttention/q/kernel": "flax_model-00007-of-00008.msgpack",
220
+ "decoder/block/22/layer/0/SelfAttention/v/kernel": "flax_model-00007-of-00008.msgpack",
221
+ "decoder/block/22/layer/0/layer_norm/weight": "flax_model-00007-of-00008.msgpack",
222
+ "decoder/block/22/layer/1/EncDecAttention/k/kernel": "flax_model-00007-of-00008.msgpack",
223
+ "decoder/block/22/layer/1/EncDecAttention/o/kernel": "flax_model-00007-of-00008.msgpack",
224
+ "decoder/block/22/layer/1/EncDecAttention/q/kernel": "flax_model-00007-of-00008.msgpack",
225
+ "decoder/block/22/layer/1/EncDecAttention/v/kernel": "flax_model-00007-of-00008.msgpack",
226
+ "decoder/block/22/layer/1/layer_norm/weight": "flax_model-00007-of-00008.msgpack",
227
+ "decoder/block/22/layer/2/DenseReluDense/wi_0/kernel": "flax_model-00007-of-00008.msgpack",
228
+ "decoder/block/22/layer/2/DenseReluDense/wi_1/kernel": "flax_model-00007-of-00008.msgpack",
229
+ "decoder/block/22/layer/2/DenseReluDense/wo/kernel": "flax_model-00007-of-00008.msgpack",
230
+ "decoder/block/22/layer/2/layer_norm/weight": "flax_model-00007-of-00008.msgpack",
231
+ "decoder/block/23/layer/0/SelfAttention/k/kernel": "flax_model-00007-of-00008.msgpack",
232
+ "decoder/block/23/layer/0/SelfAttention/o/kernel": "flax_model-00007-of-00008.msgpack",
233
+ "decoder/block/23/layer/0/SelfAttention/q/kernel": "flax_model-00007-of-00008.msgpack",
234
+ "decoder/block/23/layer/0/SelfAttention/v/kernel": "flax_model-00007-of-00008.msgpack",
235
+ "decoder/block/23/layer/0/layer_norm/weight": "flax_model-00007-of-00008.msgpack",
236
+ "decoder/block/23/layer/1/EncDecAttention/k/kernel": "flax_model-00007-of-00008.msgpack",
237
+ "decoder/block/23/layer/1/EncDecAttention/o/kernel": "flax_model-00007-of-00008.msgpack",
238
+ "decoder/block/23/layer/1/EncDecAttention/q/kernel": "flax_model-00007-of-00008.msgpack",
239
+ "decoder/block/23/layer/1/EncDecAttention/v/kernel": "flax_model-00007-of-00008.msgpack",
240
+ "decoder/block/23/layer/1/layer_norm/weight": "flax_model-00007-of-00008.msgpack",
241
+ "decoder/block/23/layer/2/DenseReluDense/wi_0/kernel": "flax_model-00007-of-00008.msgpack",
242
+ "decoder/block/23/layer/2/DenseReluDense/wi_1/kernel": "flax_model-00007-of-00008.msgpack",
243
+ "decoder/block/23/layer/2/DenseReluDense/wo/kernel": "flax_model-00007-of-00008.msgpack",
244
+ "decoder/block/23/layer/2/layer_norm/weight": "flax_model-00007-of-00008.msgpack",
245
+ "decoder/block/24/layer/0/SelfAttention/k/kernel": "flax_model-00007-of-00008.msgpack",
246
+ "decoder/block/24/layer/0/SelfAttention/o/kernel": "flax_model-00007-of-00008.msgpack",
247
+ "decoder/block/24/layer/0/SelfAttention/q/kernel": "flax_model-00007-of-00008.msgpack",
248
+ "decoder/block/24/layer/0/SelfAttention/v/kernel": "flax_model-00007-of-00008.msgpack",
249
+ "decoder/block/24/layer/0/layer_norm/weight": "flax_model-00007-of-00008.msgpack",
250
+ "decoder/block/24/layer/1/EncDecAttention/k/kernel": "flax_model-00007-of-00008.msgpack",
251
+ "decoder/block/24/layer/1/EncDecAttention/o/kernel": "flax_model-00007-of-00008.msgpack",
252
+ "decoder/block/24/layer/1/EncDecAttention/q/kernel": "flax_model-00007-of-00008.msgpack",
253
+ "decoder/block/24/layer/1/EncDecAttention/v/kernel": "flax_model-00007-of-00008.msgpack",
254
+ "decoder/block/24/layer/1/layer_norm/weight": "flax_model-00007-of-00008.msgpack",
255
+ "decoder/block/24/layer/2/DenseReluDense/wi_0/kernel": "flax_model-00007-of-00008.msgpack",
256
+ "decoder/block/24/layer/2/DenseReluDense/wi_1/kernel": "flax_model-00007-of-00008.msgpack",
257
+ "decoder/block/24/layer/2/DenseReluDense/wo/kernel": "flax_model-00007-of-00008.msgpack",
258
+ "decoder/block/24/layer/2/layer_norm/weight": "flax_model-00007-of-00008.msgpack",
259
+ "decoder/block/25/layer/0/SelfAttention/k/kernel": "flax_model-00007-of-00008.msgpack",
260
+ "decoder/block/25/layer/0/SelfAttention/o/kernel": "flax_model-00007-of-00008.msgpack",
261
+ "decoder/block/25/layer/0/SelfAttention/q/kernel": "flax_model-00007-of-00008.msgpack",
262
+ "decoder/block/25/layer/0/SelfAttention/v/kernel": "flax_model-00007-of-00008.msgpack",
263
+ "decoder/block/25/layer/0/layer_norm/weight": "flax_model-00007-of-00008.msgpack",
264
+ "decoder/block/25/layer/1/EncDecAttention/k/kernel": "flax_model-00007-of-00008.msgpack",
265
+ "decoder/block/25/layer/1/EncDecAttention/o/kernel": "flax_model-00007-of-00008.msgpack",
266
+ "decoder/block/25/layer/1/EncDecAttention/q/kernel": "flax_model-00007-of-00008.msgpack",
267
+ "decoder/block/25/layer/1/EncDecAttention/v/kernel": "flax_model-00007-of-00008.msgpack",
268
+ "decoder/block/25/layer/1/layer_norm/weight": "flax_model-00007-of-00008.msgpack",
269
+ "decoder/block/25/layer/2/DenseReluDense/wi_0/kernel": "flax_model-00007-of-00008.msgpack",
270
+ "decoder/block/25/layer/2/DenseReluDense/wi_1/kernel": "flax_model-00007-of-00008.msgpack",
271
+ "decoder/block/25/layer/2/DenseReluDense/wo/kernel": "flax_model-00008-of-00008.msgpack",
272
+ "decoder/block/25/layer/2/layer_norm/weight": "flax_model-00008-of-00008.msgpack",
273
+ "decoder/block/26/layer/0/SelfAttention/k/kernel": "flax_model-00008-of-00008.msgpack",
274
+ "decoder/block/26/layer/0/SelfAttention/o/kernel": "flax_model-00008-of-00008.msgpack",
275
+ "decoder/block/26/layer/0/SelfAttention/q/kernel": "flax_model-00008-of-00008.msgpack",
276
+ "decoder/block/26/layer/0/SelfAttention/v/kernel": "flax_model-00008-of-00008.msgpack",
277
+ "decoder/block/26/layer/0/layer_norm/weight": "flax_model-00008-of-00008.msgpack",
278
+ "decoder/block/26/layer/1/EncDecAttention/k/kernel": "flax_model-00008-of-00008.msgpack",
279
+ "decoder/block/26/layer/1/EncDecAttention/o/kernel": "flax_model-00008-of-00008.msgpack",
280
+ "decoder/block/26/layer/1/EncDecAttention/q/kernel": "flax_model-00008-of-00008.msgpack",
281
+ "decoder/block/26/layer/1/EncDecAttention/v/kernel": "flax_model-00008-of-00008.msgpack",
282
+ "decoder/block/26/layer/1/layer_norm/weight": "flax_model-00008-of-00008.msgpack",
283
+ "decoder/block/26/layer/2/DenseReluDense/wi_0/kernel": "flax_model-00008-of-00008.msgpack",
284
+ "decoder/block/26/layer/2/DenseReluDense/wi_1/kernel": "flax_model-00008-of-00008.msgpack",
285
+ "decoder/block/26/layer/2/DenseReluDense/wo/kernel": "flax_model-00008-of-00008.msgpack",
286
+ "decoder/block/26/layer/2/layer_norm/weight": "flax_model-00008-of-00008.msgpack",
287
+ "decoder/block/27/layer/0/SelfAttention/k/kernel": "flax_model-00008-of-00008.msgpack",
288
+ "decoder/block/27/layer/0/SelfAttention/o/kernel": "flax_model-00008-of-00008.msgpack",
289
+ "decoder/block/27/layer/0/SelfAttention/q/kernel": "flax_model-00008-of-00008.msgpack",
290
+ "decoder/block/27/layer/0/SelfAttention/v/kernel": "flax_model-00008-of-00008.msgpack",
291
+ "decoder/block/27/layer/0/layer_norm/weight": "flax_model-00008-of-00008.msgpack",
292
+ "decoder/block/27/layer/1/EncDecAttention/k/kernel": "flax_model-00008-of-00008.msgpack",
293
+ "decoder/block/27/layer/1/EncDecAttention/o/kernel": "flax_model-00008-of-00008.msgpack",
294
+ "decoder/block/27/layer/1/EncDecAttention/q/kernel": "flax_model-00008-of-00008.msgpack",
295
+ "decoder/block/27/layer/1/EncDecAttention/v/kernel": "flax_model-00008-of-00008.msgpack",
296
+ "decoder/block/27/layer/1/layer_norm/weight": "flax_model-00008-of-00008.msgpack",
297
+ "decoder/block/27/layer/2/DenseReluDense/wi_0/kernel": "flax_model-00008-of-00008.msgpack",
298
+ "decoder/block/27/layer/2/DenseReluDense/wi_1/kernel": "flax_model-00008-of-00008.msgpack",
299
+ "decoder/block/27/layer/2/DenseReluDense/wo/kernel": "flax_model-00008-of-00008.msgpack",
300
+ "decoder/block/27/layer/2/layer_norm/weight": "flax_model-00008-of-00008.msgpack",
301
+ "decoder/block/28/layer/0/SelfAttention/k/kernel": "flax_model-00008-of-00008.msgpack",
302
+ "decoder/block/28/layer/0/SelfAttention/o/kernel": "flax_model-00008-of-00008.msgpack",
303
+ "decoder/block/28/layer/0/SelfAttention/q/kernel": "flax_model-00008-of-00008.msgpack",
304
+ "decoder/block/28/layer/0/SelfAttention/v/kernel": "flax_model-00008-of-00008.msgpack",
305
+ "decoder/block/28/layer/0/layer_norm/weight": "flax_model-00008-of-00008.msgpack",
306
+ "decoder/block/28/layer/1/EncDecAttention/k/kernel": "flax_model-00008-of-00008.msgpack",
307
+ "decoder/block/28/layer/1/EncDecAttention/o/kernel": "flax_model-00008-of-00008.msgpack",
308
+ "decoder/block/28/layer/1/EncDecAttention/q/kernel": "flax_model-00008-of-00008.msgpack",
309
+ "decoder/block/28/layer/1/EncDecAttention/v/kernel": "flax_model-00008-of-00008.msgpack",
310
+ "decoder/block/28/layer/1/layer_norm/weight": "flax_model-00008-of-00008.msgpack",
311
+ "decoder/block/28/layer/2/DenseReluDense/wi_0/kernel": "flax_model-00008-of-00008.msgpack",
312
+ "decoder/block/28/layer/2/DenseReluDense/wi_1/kernel": "flax_model-00008-of-00008.msgpack",
313
+ "decoder/block/28/layer/2/DenseReluDense/wo/kernel": "flax_model-00008-of-00008.msgpack",
314
+ "decoder/block/28/layer/2/layer_norm/weight": "flax_model-00008-of-00008.msgpack",
315
+ "decoder/block/29/layer/0/SelfAttention/k/kernel": "flax_model-00008-of-00008.msgpack",
316
+ "decoder/block/29/layer/0/SelfAttention/o/kernel": "flax_model-00008-of-00008.msgpack",
317
+ "decoder/block/29/layer/0/SelfAttention/q/kernel": "flax_model-00008-of-00008.msgpack",
318
+ "decoder/block/29/layer/0/SelfAttention/v/kernel": "flax_model-00008-of-00008.msgpack",
319
+ "decoder/block/29/layer/0/layer_norm/weight": "flax_model-00008-of-00008.msgpack",
320
+ "decoder/block/29/layer/1/EncDecAttention/k/kernel": "flax_model-00008-of-00008.msgpack",
321
+ "decoder/block/29/layer/1/EncDecAttention/o/kernel": "flax_model-00008-of-00008.msgpack",
322
+ "decoder/block/29/layer/1/EncDecAttention/q/kernel": "flax_model-00008-of-00008.msgpack",
323
+ "decoder/block/29/layer/1/EncDecAttention/v/kernel": "flax_model-00008-of-00008.msgpack",
324
+ "decoder/block/29/layer/1/layer_norm/weight": "flax_model-00008-of-00008.msgpack",
325
+ "decoder/block/29/layer/2/DenseReluDense/wi_0/kernel": "flax_model-00008-of-00008.msgpack",
326
+ "decoder/block/29/layer/2/DenseReluDense/wi_1/kernel": "flax_model-00008-of-00008.msgpack",
327
+ "decoder/block/29/layer/2/DenseReluDense/wo/kernel": "flax_model-00008-of-00008.msgpack",
328
+ "decoder/block/29/layer/2/layer_norm/weight": "flax_model-00008-of-00008.msgpack",
329
+ "decoder/block/3/layer/0/SelfAttention/k/kernel": "flax_model-00004-of-00008.msgpack",
330
+ "decoder/block/3/layer/0/SelfAttention/o/kernel": "flax_model-00004-of-00008.msgpack",
331
+ "decoder/block/3/layer/0/SelfAttention/q/kernel": "flax_model-00004-of-00008.msgpack",
332
+ "decoder/block/3/layer/0/SelfAttention/v/kernel": "flax_model-00004-of-00008.msgpack",
333
+ "decoder/block/3/layer/0/layer_norm/weight": "flax_model-00004-of-00008.msgpack",
334
+ "decoder/block/3/layer/1/EncDecAttention/k/kernel": "flax_model-00004-of-00008.msgpack",
335
+ "decoder/block/3/layer/1/EncDecAttention/o/kernel": "flax_model-00004-of-00008.msgpack",
336
+ "decoder/block/3/layer/1/EncDecAttention/q/kernel": "flax_model-00004-of-00008.msgpack",
337
+ "decoder/block/3/layer/1/EncDecAttention/v/kernel": "flax_model-00004-of-00008.msgpack",
338
+ "decoder/block/3/layer/1/layer_norm/weight": "flax_model-00004-of-00008.msgpack",
339
+ "decoder/block/3/layer/2/DenseReluDense/wi_0/kernel": "flax_model-00004-of-00008.msgpack",
340
+ "decoder/block/3/layer/2/DenseReluDense/wi_1/kernel": "flax_model-00005-of-00008.msgpack",
341
+ "decoder/block/3/layer/2/DenseReluDense/wo/kernel": "flax_model-00005-of-00008.msgpack",
342
+ "decoder/block/3/layer/2/layer_norm/weight": "flax_model-00005-of-00008.msgpack",
343
+ "decoder/block/30/layer/0/SelfAttention/k/kernel": "flax_model-00008-of-00008.msgpack",
344
+ "decoder/block/30/layer/0/SelfAttention/o/kernel": "flax_model-00008-of-00008.msgpack",
345
+ "decoder/block/30/layer/0/SelfAttention/q/kernel": "flax_model-00008-of-00008.msgpack",
346
+ "decoder/block/30/layer/0/SelfAttention/v/kernel": "flax_model-00008-of-00008.msgpack",
347
+ "decoder/block/30/layer/0/layer_norm/weight": "flax_model-00008-of-00008.msgpack",
348
+ "decoder/block/30/layer/1/EncDecAttention/k/kernel": "flax_model-00008-of-00008.msgpack",
349
+ "decoder/block/30/layer/1/EncDecAttention/o/kernel": "flax_model-00008-of-00008.msgpack",
350
+ "decoder/block/30/layer/1/EncDecAttention/q/kernel": "flax_model-00008-of-00008.msgpack",
351
+ "decoder/block/30/layer/1/EncDecAttention/v/kernel": "flax_model-00008-of-00008.msgpack",
352
+ "decoder/block/30/layer/1/layer_norm/weight": "flax_model-00008-of-00008.msgpack",
353
+ "decoder/block/30/layer/2/DenseReluDense/wi_0/kernel": "flax_model-00008-of-00008.msgpack",
354
+ "decoder/block/30/layer/2/DenseReluDense/wi_1/kernel": "flax_model-00008-of-00008.msgpack",
355
+ "decoder/block/30/layer/2/DenseReluDense/wo/kernel": "flax_model-00008-of-00008.msgpack",
356
+ "decoder/block/30/layer/2/layer_norm/weight": "flax_model-00008-of-00008.msgpack",
357
+ "decoder/block/31/layer/0/SelfAttention/k/kernel": "flax_model-00008-of-00008.msgpack",
358
+ "decoder/block/31/layer/0/SelfAttention/o/kernel": "flax_model-00008-of-00008.msgpack",
359
+ "decoder/block/31/layer/0/SelfAttention/q/kernel": "flax_model-00008-of-00008.msgpack",
360
+ "decoder/block/31/layer/0/SelfAttention/v/kernel": "flax_model-00008-of-00008.msgpack",
361
+ "decoder/block/31/layer/0/layer_norm/weight": "flax_model-00008-of-00008.msgpack",
362
+ "decoder/block/31/layer/1/EncDecAttention/k/kernel": "flax_model-00008-of-00008.msgpack",
363
+ "decoder/block/31/layer/1/EncDecAttention/o/kernel": "flax_model-00008-of-00008.msgpack",
364
+ "decoder/block/31/layer/1/EncDecAttention/q/kernel": "flax_model-00008-of-00008.msgpack",
365
+ "decoder/block/31/layer/1/EncDecAttention/v/kernel": "flax_model-00008-of-00008.msgpack",
366
+ "decoder/block/31/layer/1/layer_norm/weight": "flax_model-00008-of-00008.msgpack",
367
+ "decoder/block/31/layer/2/DenseReluDense/wi_0/kernel": "flax_model-00008-of-00008.msgpack",
368
+ "decoder/block/31/layer/2/DenseReluDense/wi_1/kernel": "flax_model-00008-of-00008.msgpack",
369
+ "decoder/block/31/layer/2/DenseReluDense/wo/kernel": "flax_model-00008-of-00008.msgpack",
370
+ "decoder/block/31/layer/2/layer_norm/weight": "flax_model-00008-of-00008.msgpack",
371
+ "decoder/block/4/layer/0/SelfAttention/k/kernel": "flax_model-00005-of-00008.msgpack",
372
+ "decoder/block/4/layer/0/SelfAttention/o/kernel": "flax_model-00005-of-00008.msgpack",
373
+ "decoder/block/4/layer/0/SelfAttention/q/kernel": "flax_model-00005-of-00008.msgpack",
374
+ "decoder/block/4/layer/0/SelfAttention/v/kernel": "flax_model-00005-of-00008.msgpack",
375
+ "decoder/block/4/layer/0/layer_norm/weight": "flax_model-00005-of-00008.msgpack",
376
+ "decoder/block/4/layer/1/EncDecAttention/k/kernel": "flax_model-00005-of-00008.msgpack",
377
+ "decoder/block/4/layer/1/EncDecAttention/o/kernel": "flax_model-00005-of-00008.msgpack",
378
+ "decoder/block/4/layer/1/EncDecAttention/q/kernel": "flax_model-00005-of-00008.msgpack",
379
+ "decoder/block/4/layer/1/EncDecAttention/v/kernel": "flax_model-00005-of-00008.msgpack",
380
+ "decoder/block/4/layer/1/layer_norm/weight": "flax_model-00005-of-00008.msgpack",
381
+ "decoder/block/4/layer/2/DenseReluDense/wi_0/kernel": "flax_model-00005-of-00008.msgpack",
382
+ "decoder/block/4/layer/2/DenseReluDense/wi_1/kernel": "flax_model-00005-of-00008.msgpack",
383
+ "decoder/block/4/layer/2/DenseReluDense/wo/kernel": "flax_model-00005-of-00008.msgpack",
384
+ "decoder/block/4/layer/2/layer_norm/weight": "flax_model-00005-of-00008.msgpack",
385
+ "decoder/block/5/layer/0/SelfAttention/k/kernel": "flax_model-00005-of-00008.msgpack",
386
+ "decoder/block/5/layer/0/SelfAttention/o/kernel": "flax_model-00005-of-00008.msgpack",
387
+ "decoder/block/5/layer/0/SelfAttention/q/kernel": "flax_model-00005-of-00008.msgpack",
388
+ "decoder/block/5/layer/0/SelfAttention/v/kernel": "flax_model-00005-of-00008.msgpack",
389
+ "decoder/block/5/layer/0/layer_norm/weight": "flax_model-00005-of-00008.msgpack",
390
+ "decoder/block/5/layer/1/EncDecAttention/k/kernel": "flax_model-00005-of-00008.msgpack",
391
+ "decoder/block/5/layer/1/EncDecAttention/o/kernel": "flax_model-00005-of-00008.msgpack",
392
+ "decoder/block/5/layer/1/EncDecAttention/q/kernel": "flax_model-00005-of-00008.msgpack",
393
+ "decoder/block/5/layer/1/EncDecAttention/v/kernel": "flax_model-00005-of-00008.msgpack",
394
+ "decoder/block/5/layer/1/layer_norm/weight": "flax_model-00005-of-00008.msgpack",
395
+ "decoder/block/5/layer/2/DenseReluDense/wi_0/kernel": "flax_model-00005-of-00008.msgpack",
396
+ "decoder/block/5/layer/2/DenseReluDense/wi_1/kernel": "flax_model-00005-of-00008.msgpack",
397
+ "decoder/block/5/layer/2/DenseReluDense/wo/kernel": "flax_model-00005-of-00008.msgpack",
398
+ "decoder/block/5/layer/2/layer_norm/weight": "flax_model-00005-of-00008.msgpack",
399
+ "decoder/block/6/layer/0/SelfAttention/k/kernel": "flax_model-00005-of-00008.msgpack",
400
+ "decoder/block/6/layer/0/SelfAttention/o/kernel": "flax_model-00005-of-00008.msgpack",
401
+ "decoder/block/6/layer/0/SelfAttention/q/kernel": "flax_model-00005-of-00008.msgpack",
402
+ "decoder/block/6/layer/0/SelfAttention/v/kernel": "flax_model-00005-of-00008.msgpack",
403
+ "decoder/block/6/layer/0/layer_norm/weight": "flax_model-00005-of-00008.msgpack",
404
+ "decoder/block/6/layer/1/EncDecAttention/k/kernel": "flax_model-00005-of-00008.msgpack",
405
+ "decoder/block/6/layer/1/EncDecAttention/o/kernel": "flax_model-00005-of-00008.msgpack",
406
+ "decoder/block/6/layer/1/EncDecAttention/q/kernel": "flax_model-00005-of-00008.msgpack",
407
+ "decoder/block/6/layer/1/EncDecAttention/v/kernel": "flax_model-00005-of-00008.msgpack",
408
+ "decoder/block/6/layer/1/layer_norm/weight": "flax_model-00005-of-00008.msgpack",
409
+ "decoder/block/6/layer/2/DenseReluDense/wi_0/kernel": "flax_model-00005-of-00008.msgpack",
410
+ "decoder/block/6/layer/2/DenseReluDense/wi_1/kernel": "flax_model-00005-of-00008.msgpack",
411
+ "decoder/block/6/layer/2/DenseReluDense/wo/kernel": "flax_model-00005-of-00008.msgpack",
412
+ "decoder/block/6/layer/2/layer_norm/weight": "flax_model-00005-of-00008.msgpack",
413
+ "decoder/block/7/layer/0/SelfAttention/k/kernel": "flax_model-00005-of-00008.msgpack",
414
+ "decoder/block/7/layer/0/SelfAttention/o/kernel": "flax_model-00005-of-00008.msgpack",
415
+ "decoder/block/7/layer/0/SelfAttention/q/kernel": "flax_model-00005-of-00008.msgpack",
416
+ "decoder/block/7/layer/0/SelfAttention/v/kernel": "flax_model-00005-of-00008.msgpack",
417
+ "decoder/block/7/layer/0/layer_norm/weight": "flax_model-00005-of-00008.msgpack",
418
+ "decoder/block/7/layer/1/EncDecAttention/k/kernel": "flax_model-00005-of-00008.msgpack",
419
+ "decoder/block/7/layer/1/EncDecAttention/o/kernel": "flax_model-00005-of-00008.msgpack",
420
+ "decoder/block/7/layer/1/EncDecAttention/q/kernel": "flax_model-00005-of-00008.msgpack",
421
+ "decoder/block/7/layer/1/EncDecAttention/v/kernel": "flax_model-00005-of-00008.msgpack",
422
+ "decoder/block/7/layer/1/layer_norm/weight": "flax_model-00005-of-00008.msgpack",
423
+ "decoder/block/7/layer/2/DenseReluDense/wi_0/kernel": "flax_model-00005-of-00008.msgpack",
424
+ "decoder/block/7/layer/2/DenseReluDense/wi_1/kernel": "flax_model-00005-of-00008.msgpack",
425
+ "decoder/block/7/layer/2/DenseReluDense/wo/kernel": "flax_model-00005-of-00008.msgpack",
426
+ "decoder/block/7/layer/2/layer_norm/weight": "flax_model-00005-of-00008.msgpack",
427
+ "decoder/block/8/layer/0/SelfAttention/k/kernel": "flax_model-00005-of-00008.msgpack",
428
+ "decoder/block/8/layer/0/SelfAttention/o/kernel": "flax_model-00005-of-00008.msgpack",
429
+ "decoder/block/8/layer/0/SelfAttention/q/kernel": "flax_model-00005-of-00008.msgpack",
430
+ "decoder/block/8/layer/0/SelfAttention/v/kernel": "flax_model-00005-of-00008.msgpack",
431
+ "decoder/block/8/layer/0/layer_norm/weight": "flax_model-00005-of-00008.msgpack",
432
+ "decoder/block/8/layer/1/EncDecAttention/k/kernel": "flax_model-00005-of-00008.msgpack",
433
+ "decoder/block/8/layer/1/EncDecAttention/o/kernel": "flax_model-00005-of-00008.msgpack",
434
+ "decoder/block/8/layer/1/EncDecAttention/q/kernel": "flax_model-00005-of-00008.msgpack",
435
+ "decoder/block/8/layer/1/EncDecAttention/v/kernel": "flax_model-00005-of-00008.msgpack",
436
+ "decoder/block/8/layer/1/layer_norm/weight": "flax_model-00005-of-00008.msgpack",
437
+ "decoder/block/8/layer/2/DenseReluDense/wi_0/kernel": "flax_model-00005-of-00008.msgpack",
438
+ "decoder/block/8/layer/2/DenseReluDense/wi_1/kernel": "flax_model-00005-of-00008.msgpack",
439
+ "decoder/block/8/layer/2/DenseReluDense/wo/kernel": "flax_model-00005-of-00008.msgpack",
440
+ "decoder/block/8/layer/2/layer_norm/weight": "flax_model-00005-of-00008.msgpack",
441
+ "decoder/block/9/layer/0/SelfAttention/k/kernel": "flax_model-00005-of-00008.msgpack",
442
+ "decoder/block/9/layer/0/SelfAttention/o/kernel": "flax_model-00005-of-00008.msgpack",
443
+ "decoder/block/9/layer/0/SelfAttention/q/kernel": "flax_model-00005-of-00008.msgpack",
444
+ "decoder/block/9/layer/0/SelfAttention/v/kernel": "flax_model-00005-of-00008.msgpack",
445
+ "decoder/block/9/layer/0/layer_norm/weight": "flax_model-00005-of-00008.msgpack",
446
+ "decoder/block/9/layer/1/EncDecAttention/k/kernel": "flax_model-00005-of-00008.msgpack",
447
+ "decoder/block/9/layer/1/EncDecAttention/o/kernel": "flax_model-00005-of-00008.msgpack",
448
+ "decoder/block/9/layer/1/EncDecAttention/q/kernel": "flax_model-00005-of-00008.msgpack",
449
+ "decoder/block/9/layer/1/EncDecAttention/v/kernel": "flax_model-00005-of-00008.msgpack",
450
+ "decoder/block/9/layer/1/layer_norm/weight": "flax_model-00005-of-00008.msgpack",
451
+ "decoder/block/9/layer/2/DenseReluDense/wi_0/kernel": "flax_model-00005-of-00008.msgpack",
452
+ "decoder/block/9/layer/2/DenseReluDense/wi_1/kernel": "flax_model-00005-of-00008.msgpack",
453
+ "decoder/block/9/layer/2/DenseReluDense/wo/kernel": "flax_model-00005-of-00008.msgpack",
454
+ "decoder/block/9/layer/2/layer_norm/weight": "flax_model-00005-of-00008.msgpack",
455
+ "decoder/final_layer_norm/weight": "flax_model-00008-of-00008.msgpack",
456
+ "encoder/block/0/layer/0/SelfAttention/k/kernel": "flax_model-00001-of-00008.msgpack",
457
+ "encoder/block/0/layer/0/SelfAttention/o/kernel": "flax_model-00001-of-00008.msgpack",
458
+ "encoder/block/0/layer/0/SelfAttention/q/kernel": "flax_model-00001-of-00008.msgpack",
459
+ "encoder/block/0/layer/0/SelfAttention/relative_attention_bias/embedding": "flax_model-00001-of-00008.msgpack",
460
+ "encoder/block/0/layer/0/SelfAttention/v/kernel": "flax_model-00001-of-00008.msgpack",
461
+ "encoder/block/0/layer/0/layer_norm/weight": "flax_model-00001-of-00008.msgpack",
462
+ "encoder/block/0/layer/1/DenseReluDense/wi_0/kernel": "flax_model-00001-of-00008.msgpack",
463
+ "encoder/block/0/layer/1/DenseReluDense/wi_1/kernel": "flax_model-00001-of-00008.msgpack",
464
+ "encoder/block/0/layer/1/DenseReluDense/wo/kernel": "flax_model-00001-of-00008.msgpack",
465
+ "encoder/block/0/layer/1/layer_norm/weight": "flax_model-00001-of-00008.msgpack",
466
+ "encoder/block/1/layer/0/SelfAttention/k/kernel": "flax_model-00001-of-00008.msgpack",
467
+ "encoder/block/1/layer/0/SelfAttention/o/kernel": "flax_model-00001-of-00008.msgpack",
468
+ "encoder/block/1/layer/0/SelfAttention/q/kernel": "flax_model-00001-of-00008.msgpack",
469
+ "encoder/block/1/layer/0/SelfAttention/v/kernel": "flax_model-00001-of-00008.msgpack",
470
+ "encoder/block/1/layer/0/layer_norm/weight": "flax_model-00001-of-00008.msgpack",
471
+ "encoder/block/1/layer/1/DenseReluDense/wi_0/kernel": "flax_model-00001-of-00008.msgpack",
472
+ "encoder/block/1/layer/1/DenseReluDense/wi_1/kernel": "flax_model-00001-of-00008.msgpack",
473
+ "encoder/block/1/layer/1/DenseReluDense/wo/kernel": "flax_model-00001-of-00008.msgpack",
474
+ "encoder/block/1/layer/1/layer_norm/weight": "flax_model-00001-of-00008.msgpack",
475
+ "encoder/block/10/layer/0/SelfAttention/k/kernel": "flax_model-00002-of-00008.msgpack",
476
+ "encoder/block/10/layer/0/SelfAttention/o/kernel": "flax_model-00002-of-00008.msgpack",
477
+ "encoder/block/10/layer/0/SelfAttention/q/kernel": "flax_model-00002-of-00008.msgpack",
478
+ "encoder/block/10/layer/0/SelfAttention/v/kernel": "flax_model-00002-of-00008.msgpack",
479
+ "encoder/block/10/layer/0/layer_norm/weight": "flax_model-00002-of-00008.msgpack",
480
+ "encoder/block/10/layer/1/DenseReluDense/wi_0/kernel": "flax_model-00002-of-00008.msgpack",
481
+ "encoder/block/10/layer/1/DenseReluDense/wi_1/kernel": "flax_model-00002-of-00008.msgpack",
482
+ "encoder/block/10/layer/1/DenseReluDense/wo/kernel": "flax_model-00002-of-00008.msgpack",
483
+ "encoder/block/10/layer/1/layer_norm/weight": "flax_model-00002-of-00008.msgpack",
484
+ "encoder/block/11/layer/0/SelfAttention/k/kernel": "flax_model-00002-of-00008.msgpack",
485
+ "encoder/block/11/layer/0/SelfAttention/o/kernel": "flax_model-00002-of-00008.msgpack",
486
+ "encoder/block/11/layer/0/SelfAttention/q/kernel": "flax_model-00002-of-00008.msgpack",
487
+ "encoder/block/11/layer/0/SelfAttention/v/kernel": "flax_model-00002-of-00008.msgpack",
488
+ "encoder/block/11/layer/0/layer_norm/weight": "flax_model-00002-of-00008.msgpack",
489
+ "encoder/block/11/layer/1/DenseReluDense/wi_0/kernel": "flax_model-00002-of-00008.msgpack",
490
+ "encoder/block/11/layer/1/DenseReluDense/wi_1/kernel": "flax_model-00002-of-00008.msgpack",
491
+ "encoder/block/11/layer/1/DenseReluDense/wo/kernel": "flax_model-00002-of-00008.msgpack",
492
+ "encoder/block/11/layer/1/layer_norm/weight": "flax_model-00002-of-00008.msgpack",
493
+ "encoder/block/12/layer/0/SelfAttention/k/kernel": "flax_model-00002-of-00008.msgpack",
494
+ "encoder/block/12/layer/0/SelfAttention/o/kernel": "flax_model-00002-of-00008.msgpack",
495
+ "encoder/block/12/layer/0/SelfAttention/q/kernel": "flax_model-00002-of-00008.msgpack",
496
+ "encoder/block/12/layer/0/SelfAttention/v/kernel": "flax_model-00002-of-00008.msgpack",
497
+ "encoder/block/12/layer/0/layer_norm/weight": "flax_model-00002-of-00008.msgpack",
498
+ "encoder/block/12/layer/1/DenseReluDense/wi_0/kernel": "flax_model-00002-of-00008.msgpack",
499
+ "encoder/block/12/layer/1/DenseReluDense/wi_1/kernel": "flax_model-00002-of-00008.msgpack",
500
+ "encoder/block/12/layer/1/DenseReluDense/wo/kernel": "flax_model-00002-of-00008.msgpack",
501
+ "encoder/block/12/layer/1/layer_norm/weight": "flax_model-00002-of-00008.msgpack",
502
+ "encoder/block/13/layer/0/SelfAttention/k/kernel": "flax_model-00002-of-00008.msgpack",
503
+ "encoder/block/13/layer/0/SelfAttention/o/kernel": "flax_model-00002-of-00008.msgpack",
504
+ "encoder/block/13/layer/0/SelfAttention/q/kernel": "flax_model-00002-of-00008.msgpack",
505
+ "encoder/block/13/layer/0/SelfAttention/v/kernel": "flax_model-00002-of-00008.msgpack",
506
+ "encoder/block/13/layer/0/layer_norm/weight": "flax_model-00002-of-00008.msgpack",
507
+ "encoder/block/13/layer/1/DenseReluDense/wi_0/kernel": "flax_model-00002-of-00008.msgpack",
508
+ "encoder/block/13/layer/1/DenseReluDense/wi_1/kernel": "flax_model-00002-of-00008.msgpack",
509
+ "encoder/block/13/layer/1/DenseReluDense/wo/kernel": "flax_model-00002-of-00008.msgpack",
510
+ "encoder/block/13/layer/1/layer_norm/weight": "flax_model-00002-of-00008.msgpack",
511
+ "encoder/block/14/layer/0/SelfAttention/k/kernel": "flax_model-00002-of-00008.msgpack",
512
+ "encoder/block/14/layer/0/SelfAttention/o/kernel": "flax_model-00002-of-00008.msgpack",
513
+ "encoder/block/14/layer/0/SelfAttention/q/kernel": "flax_model-00002-of-00008.msgpack",
514
+ "encoder/block/14/layer/0/SelfAttention/v/kernel": "flax_model-00002-of-00008.msgpack",
515
+ "encoder/block/14/layer/0/layer_norm/weight": "flax_model-00002-of-00008.msgpack",
516
+ "encoder/block/14/layer/1/DenseReluDense/wi_0/kernel": "flax_model-00002-of-00008.msgpack",
517
+ "encoder/block/14/layer/1/DenseReluDense/wi_1/kernel": "flax_model-00002-of-00008.msgpack",
518
+ "encoder/block/14/layer/1/DenseReluDense/wo/kernel": "flax_model-00002-of-00008.msgpack",
519
+ "encoder/block/14/layer/1/layer_norm/weight": "flax_model-00002-of-00008.msgpack",
520
+ "encoder/block/15/layer/0/SelfAttention/k/kernel": "flax_model-00002-of-00008.msgpack",
521
+ "encoder/block/15/layer/0/SelfAttention/o/kernel": "flax_model-00002-of-00008.msgpack",
522
+ "encoder/block/15/layer/0/SelfAttention/q/kernel": "flax_model-00002-of-00008.msgpack",
523
+ "encoder/block/15/layer/0/SelfAttention/v/kernel": "flax_model-00002-of-00008.msgpack",
524
+ "encoder/block/15/layer/0/layer_norm/weight": "flax_model-00002-of-00008.msgpack",
525
+ "encoder/block/15/layer/1/DenseReluDense/wi_0/kernel": "flax_model-00002-of-00008.msgpack",
526
+ "encoder/block/15/layer/1/DenseReluDense/wi_1/kernel": "flax_model-00002-of-00008.msgpack",
527
+ "encoder/block/15/layer/1/DenseReluDense/wo/kernel": "flax_model-00002-of-00008.msgpack",
528
+ "encoder/block/15/layer/1/layer_norm/weight": "flax_model-00002-of-00008.msgpack",
529
+ "encoder/block/16/layer/0/SelfAttention/k/kernel": "flax_model-00002-of-00008.msgpack",
530
+ "encoder/block/16/layer/0/SelfAttention/o/kernel": "flax_model-00002-of-00008.msgpack",
531
+ "encoder/block/16/layer/0/SelfAttention/q/kernel": "flax_model-00002-of-00008.msgpack",
532
+ "encoder/block/16/layer/0/SelfAttention/v/kernel": "flax_model-00002-of-00008.msgpack",
533
+ "encoder/block/16/layer/0/layer_norm/weight": "flax_model-00002-of-00008.msgpack",
534
+ "encoder/block/16/layer/1/DenseReluDense/wi_0/kernel": "flax_model-00002-of-00008.msgpack",
535
+ "encoder/block/16/layer/1/DenseReluDense/wi_1/kernel": "flax_model-00002-of-00008.msgpack",
536
+ "encoder/block/16/layer/1/DenseReluDense/wo/kernel": "flax_model-00002-of-00008.msgpack",
537
+ "encoder/block/16/layer/1/layer_norm/weight": "flax_model-00002-of-00008.msgpack",
538
+ "encoder/block/17/layer/0/SelfAttention/k/kernel": "flax_model-00002-of-00008.msgpack",
539
+ "encoder/block/17/layer/0/SelfAttention/o/kernel": "flax_model-00002-of-00008.msgpack",
540
+ "encoder/block/17/layer/0/SelfAttention/q/kernel": "flax_model-00002-of-00008.msgpack",
541
+ "encoder/block/17/layer/0/SelfAttention/v/kernel": "flax_model-00002-of-00008.msgpack",
542
+ "encoder/block/17/layer/0/layer_norm/weight": "flax_model-00002-of-00008.msgpack",
543
+ "encoder/block/17/layer/1/DenseReluDense/wi_0/kernel": "flax_model-00002-of-00008.msgpack",
544
+ "encoder/block/17/layer/1/DenseReluDense/wi_1/kernel": "flax_model-00002-of-00008.msgpack",
545
+ "encoder/block/17/layer/1/DenseReluDense/wo/kernel": "flax_model-00002-of-00008.msgpack",
546
+ "encoder/block/17/layer/1/layer_norm/weight": "flax_model-00002-of-00008.msgpack",
547
+ "encoder/block/18/layer/0/SelfAttention/k/kernel": "flax_model-00003-of-00008.msgpack",
548
+ "encoder/block/18/layer/0/SelfAttention/o/kernel": "flax_model-00003-of-00008.msgpack",
549
+ "encoder/block/18/layer/0/SelfAttention/q/kernel": "flax_model-00002-of-00008.msgpack",
550
+ "encoder/block/18/layer/0/SelfAttention/v/kernel": "flax_model-00003-of-00008.msgpack",
551
+ "encoder/block/18/layer/0/layer_norm/weight": "flax_model-00003-of-00008.msgpack",
552
+ "encoder/block/18/layer/1/DenseReluDense/wi_0/kernel": "flax_model-00003-of-00008.msgpack",
553
+ "encoder/block/18/layer/1/DenseReluDense/wi_1/kernel": "flax_model-00003-of-00008.msgpack",
554
+ "encoder/block/18/layer/1/DenseReluDense/wo/kernel": "flax_model-00003-of-00008.msgpack",
555
+ "encoder/block/18/layer/1/layer_norm/weight": "flax_model-00003-of-00008.msgpack",
556
+ "encoder/block/19/layer/0/SelfAttention/k/kernel": "flax_model-00003-of-00008.msgpack",
557
+ "encoder/block/19/layer/0/SelfAttention/o/kernel": "flax_model-00003-of-00008.msgpack",
558
+ "encoder/block/19/layer/0/SelfAttention/q/kernel": "flax_model-00003-of-00008.msgpack",
559
+ "encoder/block/19/layer/0/SelfAttention/v/kernel": "flax_model-00003-of-00008.msgpack",
560
+ "encoder/block/19/layer/0/layer_norm/weight": "flax_model-00003-of-00008.msgpack",
561
+ "encoder/block/19/layer/1/DenseReluDense/wi_0/kernel": "flax_model-00003-of-00008.msgpack",
562
+ "encoder/block/19/layer/1/DenseReluDense/wi_1/kernel": "flax_model-00003-of-00008.msgpack",
563
+ "encoder/block/19/layer/1/DenseReluDense/wo/kernel": "flax_model-00003-of-00008.msgpack",
564
+ "encoder/block/19/layer/1/layer_norm/weight": "flax_model-00003-of-00008.msgpack",
565
+ "encoder/block/2/layer/0/SelfAttention/k/kernel": "flax_model-00001-of-00008.msgpack",
566
+ "encoder/block/2/layer/0/SelfAttention/o/kernel": "flax_model-00001-of-00008.msgpack",
567
+ "encoder/block/2/layer/0/SelfAttention/q/kernel": "flax_model-00001-of-00008.msgpack",
568
+ "encoder/block/2/layer/0/SelfAttention/v/kernel": "flax_model-00001-of-00008.msgpack",
569
+ "encoder/block/2/layer/0/layer_norm/weight": "flax_model-00001-of-00008.msgpack",
570
+ "encoder/block/2/layer/1/DenseReluDense/wi_0/kernel": "flax_model-00001-of-00008.msgpack",
571
+ "encoder/block/2/layer/1/DenseReluDense/wi_1/kernel": "flax_model-00001-of-00008.msgpack",
572
+ "encoder/block/2/layer/1/DenseReluDense/wo/kernel": "flax_model-00001-of-00008.msgpack",
573
+ "encoder/block/2/layer/1/layer_norm/weight": "flax_model-00001-of-00008.msgpack",
574
+ "encoder/block/20/layer/0/SelfAttention/k/kernel": "flax_model-00003-of-00008.msgpack",
575
+ "encoder/block/20/layer/0/SelfAttention/o/kernel": "flax_model-00003-of-00008.msgpack",
576
+ "encoder/block/20/layer/0/SelfAttention/q/kernel": "flax_model-00003-of-00008.msgpack",
577
+ "encoder/block/20/layer/0/SelfAttention/v/kernel": "flax_model-00003-of-00008.msgpack",
578
+ "encoder/block/20/layer/0/layer_norm/weight": "flax_model-00003-of-00008.msgpack",
579
+ "encoder/block/20/layer/1/DenseReluDense/wi_0/kernel": "flax_model-00003-of-00008.msgpack",
580
+ "encoder/block/20/layer/1/DenseReluDense/wi_1/kernel": "flax_model-00003-of-00008.msgpack",
581
+ "encoder/block/20/layer/1/DenseReluDense/wo/kernel": "flax_model-00003-of-00008.msgpack",
582
+ "encoder/block/20/layer/1/layer_norm/weight": "flax_model-00003-of-00008.msgpack",
583
+ "encoder/block/21/layer/0/SelfAttention/k/kernel": "flax_model-00003-of-00008.msgpack",
584
+ "encoder/block/21/layer/0/SelfAttention/o/kernel": "flax_model-00003-of-00008.msgpack",
585
+ "encoder/block/21/layer/0/SelfAttention/q/kernel": "flax_model-00003-of-00008.msgpack",
586
+ "encoder/block/21/layer/0/SelfAttention/v/kernel": "flax_model-00003-of-00008.msgpack",
587
+ "encoder/block/21/layer/0/layer_norm/weight": "flax_model-00003-of-00008.msgpack",
588
+ "encoder/block/21/layer/1/DenseReluDense/wi_0/kernel": "flax_model-00003-of-00008.msgpack",
589
+ "encoder/block/21/layer/1/DenseReluDense/wi_1/kernel": "flax_model-00003-of-00008.msgpack",
590
+ "encoder/block/21/layer/1/DenseReluDense/wo/kernel": "flax_model-00003-of-00008.msgpack",
591
+ "encoder/block/21/layer/1/layer_norm/weight": "flax_model-00003-of-00008.msgpack",
592
+ "encoder/block/22/layer/0/SelfAttention/k/kernel": "flax_model-00003-of-00008.msgpack",
593
+ "encoder/block/22/layer/0/SelfAttention/o/kernel": "flax_model-00003-of-00008.msgpack",
594
+ "encoder/block/22/layer/0/SelfAttention/q/kernel": "flax_model-00003-of-00008.msgpack",
595
+ "encoder/block/22/layer/0/SelfAttention/v/kernel": "flax_model-00003-of-00008.msgpack",
596
+ "encoder/block/22/layer/0/layer_norm/weight": "flax_model-00003-of-00008.msgpack",
597
+ "encoder/block/22/layer/1/DenseReluDense/wi_0/kernel": "flax_model-00003-of-00008.msgpack",
598
+ "encoder/block/22/layer/1/DenseReluDense/wi_1/kernel": "flax_model-00003-of-00008.msgpack",
599
+ "encoder/block/22/layer/1/DenseReluDense/wo/kernel": "flax_model-00003-of-00008.msgpack",
600
+ "encoder/block/22/layer/1/layer_norm/weight": "flax_model-00003-of-00008.msgpack",
601
+ "encoder/block/23/layer/0/SelfAttention/k/kernel": "flax_model-00003-of-00008.msgpack",
602
+ "encoder/block/23/layer/0/SelfAttention/o/kernel": "flax_model-00003-of-00008.msgpack",
603
+ "encoder/block/23/layer/0/SelfAttention/q/kernel": "flax_model-00003-of-00008.msgpack",
604
+ "encoder/block/23/layer/0/SelfAttention/v/kernel": "flax_model-00003-of-00008.msgpack",
605
+ "encoder/block/23/layer/0/layer_norm/weight": "flax_model-00003-of-00008.msgpack",
606
+ "encoder/block/23/layer/1/DenseReluDense/wi_0/kernel": "flax_model-00003-of-00008.msgpack",
607
+ "encoder/block/23/layer/1/DenseReluDense/wi_1/kernel": "flax_model-00003-of-00008.msgpack",
608
+ "encoder/block/23/layer/1/DenseReluDense/wo/kernel": "flax_model-00003-of-00008.msgpack",
609
+ "encoder/block/23/layer/1/layer_norm/weight": "flax_model-00003-of-00008.msgpack",
610
+ "encoder/block/24/layer/0/SelfAttention/k/kernel": "flax_model-00003-of-00008.msgpack",
611
+ "encoder/block/24/layer/0/SelfAttention/o/kernel": "flax_model-00003-of-00008.msgpack",
612
+ "encoder/block/24/layer/0/SelfAttention/q/kernel": "flax_model-00003-of-00008.msgpack",
613
+ "encoder/block/24/layer/0/SelfAttention/v/kernel": "flax_model-00003-of-00008.msgpack",
614
+ "encoder/block/24/layer/0/layer_norm/weight": "flax_model-00003-of-00008.msgpack",
615
+ "encoder/block/24/layer/1/DenseReluDense/wi_0/kernel": "flax_model-00003-of-00008.msgpack",
616
+ "encoder/block/24/layer/1/DenseReluDense/wi_1/kernel": "flax_model-00003-of-00008.msgpack",
617
+ "encoder/block/24/layer/1/DenseReluDense/wo/kernel": "flax_model-00003-of-00008.msgpack",
618
+ "encoder/block/24/layer/1/layer_norm/weight": "flax_model-00003-of-00008.msgpack",
619
+ "encoder/block/25/layer/0/SelfAttention/k/kernel": "flax_model-00003-of-00008.msgpack",
620
+ "encoder/block/25/layer/0/SelfAttention/o/kernel": "flax_model-00003-of-00008.msgpack",
621
+ "encoder/block/25/layer/0/SelfAttention/q/kernel": "flax_model-00003-of-00008.msgpack",
622
+ "encoder/block/25/layer/0/SelfAttention/v/kernel": "flax_model-00003-of-00008.msgpack",
623
+ "encoder/block/25/layer/0/layer_norm/weight": "flax_model-00003-of-00008.msgpack",
624
+ "encoder/block/25/layer/1/DenseReluDense/wi_0/kernel": "flax_model-00003-of-00008.msgpack",
625
+ "encoder/block/25/layer/1/DenseReluDense/wi_1/kernel": "flax_model-00003-of-00008.msgpack",
626
+ "encoder/block/25/layer/1/DenseReluDense/wo/kernel": "flax_model-00003-of-00008.msgpack",
627
+ "encoder/block/25/layer/1/layer_norm/weight": "flax_model-00003-of-00008.msgpack",
628
+ "encoder/block/26/layer/0/SelfAttention/k/kernel": "flax_model-00003-of-00008.msgpack",
629
+ "encoder/block/26/layer/0/SelfAttention/o/kernel": "flax_model-00003-of-00008.msgpack",
630
+ "encoder/block/26/layer/0/SelfAttention/q/kernel": "flax_model-00003-of-00008.msgpack",
631
+ "encoder/block/26/layer/0/SelfAttention/v/kernel": "flax_model-00003-of-00008.msgpack",
632
+ "encoder/block/26/layer/0/layer_norm/weight": "flax_model-00003-of-00008.msgpack",
633
+ "encoder/block/26/layer/1/DenseReluDense/wi_0/kernel": "flax_model-00003-of-00008.msgpack",
634
+ "encoder/block/26/layer/1/DenseReluDense/wi_1/kernel": "flax_model-00003-of-00008.msgpack",
635
+ "encoder/block/26/layer/1/DenseReluDense/wo/kernel": "flax_model-00003-of-00008.msgpack",
636
+ "encoder/block/26/layer/1/layer_norm/weight": "flax_model-00003-of-00008.msgpack",
637
+ "encoder/block/27/layer/0/SelfAttention/k/kernel": "flax_model-00003-of-00008.msgpack",
638
+ "encoder/block/27/layer/0/SelfAttention/o/kernel": "flax_model-00003-of-00008.msgpack",
639
+ "encoder/block/27/layer/0/SelfAttention/q/kernel": "flax_model-00003-of-00008.msgpack",
640
+ "encoder/block/27/layer/0/SelfAttention/v/kernel": "flax_model-00003-of-00008.msgpack",
641
+ "encoder/block/27/layer/0/layer_norm/weight": "flax_model-00003-of-00008.msgpack",
642
+ "encoder/block/27/layer/1/DenseReluDense/wi_0/kernel": "flax_model-00004-of-00008.msgpack",
643
+ "encoder/block/27/layer/1/DenseReluDense/wi_1/kernel": "flax_model-00004-of-00008.msgpack",
644
+ "encoder/block/27/layer/1/DenseReluDense/wo/kernel": "flax_model-00004-of-00008.msgpack",
645
+ "encoder/block/27/layer/1/layer_norm/weight": "flax_model-00004-of-00008.msgpack",
646
+ "encoder/block/28/layer/0/SelfAttention/k/kernel": "flax_model-00004-of-00008.msgpack",
647
+ "encoder/block/28/layer/0/SelfAttention/o/kernel": "flax_model-00004-of-00008.msgpack",
648
+ "encoder/block/28/layer/0/SelfAttention/q/kernel": "flax_model-00004-of-00008.msgpack",
649
+ "encoder/block/28/layer/0/SelfAttention/v/kernel": "flax_model-00004-of-00008.msgpack",
650
+ "encoder/block/28/layer/0/layer_norm/weight": "flax_model-00004-of-00008.msgpack",
651
+ "encoder/block/28/layer/1/DenseReluDense/wi_0/kernel": "flax_model-00004-of-00008.msgpack",
652
+ "encoder/block/28/layer/1/DenseReluDense/wi_1/kernel": "flax_model-00004-of-00008.msgpack",
653
+ "encoder/block/28/layer/1/DenseReluDense/wo/kernel": "flax_model-00004-of-00008.msgpack",
654
+ "encoder/block/28/layer/1/layer_norm/weight": "flax_model-00004-of-00008.msgpack",
655
+ "encoder/block/29/layer/0/SelfAttention/k/kernel": "flax_model-00004-of-00008.msgpack",
656
+ "encoder/block/29/layer/0/SelfAttention/o/kernel": "flax_model-00004-of-00008.msgpack",
657
+ "encoder/block/29/layer/0/SelfAttention/q/kernel": "flax_model-00004-of-00008.msgpack",
658
+ "encoder/block/29/layer/0/SelfAttention/v/kernel": "flax_model-00004-of-00008.msgpack",
659
+ "encoder/block/29/layer/0/layer_norm/weight": "flax_model-00004-of-00008.msgpack",
660
+ "encoder/block/29/layer/1/DenseReluDense/wi_0/kernel": "flax_model-00004-of-00008.msgpack",
661
+ "encoder/block/29/layer/1/DenseReluDense/wi_1/kernel": "flax_model-00004-of-00008.msgpack",
662
+ "encoder/block/29/layer/1/DenseReluDense/wo/kernel": "flax_model-00004-of-00008.msgpack",
663
+ "encoder/block/29/layer/1/layer_norm/weight": "flax_model-00004-of-00008.msgpack",
664
+ "encoder/block/3/layer/0/SelfAttention/k/kernel": "flax_model-00001-of-00008.msgpack",
665
+ "encoder/block/3/layer/0/SelfAttention/o/kernel": "flax_model-00001-of-00008.msgpack",
666
+ "encoder/block/3/layer/0/SelfAttention/q/kernel": "flax_model-00001-of-00008.msgpack",
667
+ "encoder/block/3/layer/0/SelfAttention/v/kernel": "flax_model-00001-of-00008.msgpack",
668
+ "encoder/block/3/layer/0/layer_norm/weight": "flax_model-00001-of-00008.msgpack",
669
+ "encoder/block/3/layer/1/DenseReluDense/wi_0/kernel": "flax_model-00001-of-00008.msgpack",
670
+ "encoder/block/3/layer/1/DenseReluDense/wi_1/kernel": "flax_model-00001-of-00008.msgpack",
671
+ "encoder/block/3/layer/1/DenseReluDense/wo/kernel": "flax_model-00001-of-00008.msgpack",
672
+ "encoder/block/3/layer/1/layer_norm/weight": "flax_model-00001-of-00008.msgpack",
673
+ "encoder/block/30/layer/0/SelfAttention/k/kernel": "flax_model-00004-of-00008.msgpack",
674
+ "encoder/block/30/layer/0/SelfAttention/o/kernel": "flax_model-00004-of-00008.msgpack",
675
+ "encoder/block/30/layer/0/SelfAttention/q/kernel": "flax_model-00004-of-00008.msgpack",
676
+ "encoder/block/30/layer/0/SelfAttention/v/kernel": "flax_model-00004-of-00008.msgpack",
677
+ "encoder/block/30/layer/0/layer_norm/weight": "flax_model-00004-of-00008.msgpack",
678
+ "encoder/block/30/layer/1/DenseReluDense/wi_0/kernel": "flax_model-00004-of-00008.msgpack",
679
+ "encoder/block/30/layer/1/DenseReluDense/wi_1/kernel": "flax_model-00004-of-00008.msgpack",
680
+ "encoder/block/30/layer/1/DenseReluDense/wo/kernel": "flax_model-00004-of-00008.msgpack",
681
+ "encoder/block/30/layer/1/layer_norm/weight": "flax_model-00004-of-00008.msgpack",
682
+ "encoder/block/31/layer/0/SelfAttention/k/kernel": "flax_model-00004-of-00008.msgpack",
683
+ "encoder/block/31/layer/0/SelfAttention/o/kernel": "flax_model-00004-of-00008.msgpack",
684
+ "encoder/block/31/layer/0/SelfAttention/q/kernel": "flax_model-00004-of-00008.msgpack",
685
+ "encoder/block/31/layer/0/SelfAttention/v/kernel": "flax_model-00004-of-00008.msgpack",
686
+ "encoder/block/31/layer/0/layer_norm/weight": "flax_model-00004-of-00008.msgpack",
687
+ "encoder/block/31/layer/1/DenseReluDense/wi_0/kernel": "flax_model-00004-of-00008.msgpack",
688
+ "encoder/block/31/layer/1/DenseReluDense/wi_1/kernel": "flax_model-00004-of-00008.msgpack",
689
+ "encoder/block/31/layer/1/DenseReluDense/wo/kernel": "flax_model-00004-of-00008.msgpack",
690
+ "encoder/block/31/layer/1/layer_norm/weight": "flax_model-00004-of-00008.msgpack",
691
+ "encoder/block/4/layer/0/SelfAttention/k/kernel": "flax_model-00001-of-00008.msgpack",
692
+ "encoder/block/4/layer/0/SelfAttention/o/kernel": "flax_model-00001-of-00008.msgpack",
693
+ "encoder/block/4/layer/0/SelfAttention/q/kernel": "flax_model-00001-of-00008.msgpack",
694
+ "encoder/block/4/layer/0/SelfAttention/v/kernel": "flax_model-00001-of-00008.msgpack",
695
+ "encoder/block/4/layer/0/layer_norm/weight": "flax_model-00001-of-00008.msgpack",
696
+ "encoder/block/4/layer/1/DenseReluDense/wi_0/kernel": "flax_model-00001-of-00008.msgpack",
697
+ "encoder/block/4/layer/1/DenseReluDense/wi_1/kernel": "flax_model-00001-of-00008.msgpack",
698
+ "encoder/block/4/layer/1/DenseReluDense/wo/kernel": "flax_model-00001-of-00008.msgpack",
699
+ "encoder/block/4/layer/1/layer_norm/weight": "flax_model-00001-of-00008.msgpack",
700
+ "encoder/block/5/layer/0/SelfAttention/k/kernel": "flax_model-00001-of-00008.msgpack",
701
+ "encoder/block/5/layer/0/SelfAttention/o/kernel": "flax_model-00001-of-00008.msgpack",
702
+ "encoder/block/5/layer/0/SelfAttention/q/kernel": "flax_model-00001-of-00008.msgpack",
703
+ "encoder/block/5/layer/0/SelfAttention/v/kernel": "flax_model-00001-of-00008.msgpack",
704
+ "encoder/block/5/layer/0/layer_norm/weight": "flax_model-00001-of-00008.msgpack",
705
+ "encoder/block/5/layer/1/DenseReluDense/wi_0/kernel": "flax_model-00001-of-00008.msgpack",
706
+ "encoder/block/5/layer/1/DenseReluDense/wi_1/kernel": "flax_model-00001-of-00008.msgpack",
707
+ "encoder/block/5/layer/1/DenseReluDense/wo/kernel": "flax_model-00001-of-00008.msgpack",
708
+ "encoder/block/5/layer/1/layer_norm/weight": "flax_model-00001-of-00008.msgpack",
709
+ "encoder/block/6/layer/0/SelfAttention/k/kernel": "flax_model-00001-of-00008.msgpack",
710
+ "encoder/block/6/layer/0/SelfAttention/o/kernel": "flax_model-00001-of-00008.msgpack",
711
+ "encoder/block/6/layer/0/SelfAttention/q/kernel": "flax_model-00001-of-00008.msgpack",
712
+ "encoder/block/6/layer/0/SelfAttention/v/kernel": "flax_model-00001-of-00008.msgpack",
713
+ "encoder/block/6/layer/0/layer_norm/weight": "flax_model-00001-of-00008.msgpack",
714
+ "encoder/block/6/layer/1/DenseReluDense/wi_0/kernel": "flax_model-00001-of-00008.msgpack",
715
+ "encoder/block/6/layer/1/DenseReluDense/wi_1/kernel": "flax_model-00001-of-00008.msgpack",
716
+ "encoder/block/6/layer/1/DenseReluDense/wo/kernel": "flax_model-00001-of-00008.msgpack",
717
+ "encoder/block/6/layer/1/layer_norm/weight": "flax_model-00001-of-00008.msgpack",
718
+ "encoder/block/7/layer/0/SelfAttention/k/kernel": "flax_model-00001-of-00008.msgpack",
719
+ "encoder/block/7/layer/0/SelfAttention/o/kernel": "flax_model-00001-of-00008.msgpack",
720
+ "encoder/block/7/layer/0/SelfAttention/q/kernel": "flax_model-00001-of-00008.msgpack",
721
+ "encoder/block/7/layer/0/SelfAttention/v/kernel": "flax_model-00001-of-00008.msgpack",
722
+ "encoder/block/7/layer/0/layer_norm/weight": "flax_model-00001-of-00008.msgpack",
723
+ "encoder/block/7/layer/1/DenseReluDense/wi_0/kernel": "flax_model-00001-of-00008.msgpack",
724
+ "encoder/block/7/layer/1/DenseReluDense/wi_1/kernel": "flax_model-00001-of-00008.msgpack",
725
+ "encoder/block/7/layer/1/DenseReluDense/wo/kernel": "flax_model-00001-of-00008.msgpack",
726
+ "encoder/block/7/layer/1/layer_norm/weight": "flax_model-00001-of-00008.msgpack",
727
+ "encoder/block/8/layer/0/SelfAttention/k/kernel": "flax_model-00001-of-00008.msgpack",
728
+ "encoder/block/8/layer/0/SelfAttention/o/kernel": "flax_model-00001-of-00008.msgpack",
729
+ "encoder/block/8/layer/0/SelfAttention/q/kernel": "flax_model-00001-of-00008.msgpack",
730
+ "encoder/block/8/layer/0/SelfAttention/v/kernel": "flax_model-00001-of-00008.msgpack",
731
+ "encoder/block/8/layer/0/layer_norm/weight": "flax_model-00001-of-00008.msgpack",
732
+ "encoder/block/8/layer/1/DenseReluDense/wi_0/kernel": "flax_model-00001-of-00008.msgpack",
733
+ "encoder/block/8/layer/1/DenseReluDense/wi_1/kernel": "flax_model-00001-of-00008.msgpack",
734
+ "encoder/block/8/layer/1/DenseReluDense/wo/kernel": "flax_model-00002-of-00008.msgpack",
735
+ "encoder/block/8/layer/1/layer_norm/weight": "flax_model-00002-of-00008.msgpack",
736
+ "encoder/block/9/layer/0/SelfAttention/k/kernel": "flax_model-00002-of-00008.msgpack",
737
+ "encoder/block/9/layer/0/SelfAttention/o/kernel": "flax_model-00002-of-00008.msgpack",
738
+ "encoder/block/9/layer/0/SelfAttention/q/kernel": "flax_model-00002-of-00008.msgpack",
739
+ "encoder/block/9/layer/0/SelfAttention/v/kernel": "flax_model-00002-of-00008.msgpack",
740
+ "encoder/block/9/layer/0/layer_norm/weight": "flax_model-00002-of-00008.msgpack",
741
+ "encoder/block/9/layer/1/DenseReluDense/wi_0/kernel": "flax_model-00002-of-00008.msgpack",
742
+ "encoder/block/9/layer/1/DenseReluDense/wi_1/kernel": "flax_model-00002-of-00008.msgpack",
743
+ "encoder/block/9/layer/1/DenseReluDense/wo/kernel": "flax_model-00002-of-00008.msgpack",
744
+ "encoder/block/9/layer/1/layer_norm/weight": "flax_model-00002-of-00008.msgpack",
745
+ "encoder/final_layer_norm/weight": "flax_model-00004-of-00008.msgpack",
746
+ "shared/embedding": "flax_model-00001-of-00008.msgpack"
747
+ }
748
+ }
generation_config.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "decoder_start_token_id": 0,
4
+ "eos_token_id": 1,
5
+ "pad_token_id": 0,
6
+ "transformers_version": "4.27.0.dev0"
7
+ }