Alexandre-Numind commited on
Commit
7399534
1 Parent(s): a2a9f62

Upload 8 files

Browse files
config.json ADDED
@@ -0,0 +1,34 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_name_or_path": "microsoft/phi-1_5",
3
+ "architectures": [
4
+ "PhiForCausalLM"
5
+ ],
6
+ "attention_dropout": 0.0,
7
+ "auto_map": {
8
+ "AutoConfig": "microsoft/phi-1_5--configuration_phi.PhiConfig",
9
+ "AutoModelForCausalLM": "microsoft/phi-1_5--modeling_phi.PhiForCausalLM"
10
+ },
11
+ "bos_token_id": null,
12
+ "embd_pdrop": 0.0,
13
+ "eos_token_id": null,
14
+ "hidden_act": "gelu_new",
15
+ "hidden_size": 2048,
16
+ "initializer_range": 0.02,
17
+ "intermediate_size": 8192,
18
+ "layer_norm_eps": 1e-05,
19
+ "max_position_embeddings": 2048,
20
+ "model_type": "phi",
21
+ "num_attention_heads": 32,
22
+ "num_hidden_layers": 24,
23
+ "num_key_value_heads": 32,
24
+ "partial_rotary_factor": 0.5,
25
+ "qk_layernorm": false,
26
+ "resid_pdrop": 0.0,
27
+ "rope_scaling": null,
28
+ "rope_theta": 10000.0,
29
+ "tie_word_embeddings": false,
30
+ "torch_dtype": "float32",
31
+ "transformers_version": "4.40.0.dev0",
32
+ "use_cache": true,
33
+ "vocab_size": 51200
34
+ }
generation_config.json ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "transformers_version": "4.40.0.dev0"
4
+ }
model-00001-of-00002.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:405ab540656989da80ba9df89253fc0108f988e4e2edcc99124742d906fc100f
3
+ size 4984916152
model-00002-of-00002.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4d4b997597d2971f9d04a65de50cc9ccd0beee6d27d605cdd5eeda28f298e5d7
3
+ size 688204064
model.safetensors.index.json ADDED
@@ -0,0 +1,348 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "metadata": {
3
+ "total_size": 5673082880
4
+ },
5
+ "weight_map": {
6
+ "lm_head.bias": "model-00002-of-00002.safetensors",
7
+ "lm_head.weight": "model-00002-of-00002.safetensors",
8
+ "model.embed_tokens.weight": "model-00001-of-00002.safetensors",
9
+ "model.final_layernorm.bias": "model-00002-of-00002.safetensors",
10
+ "model.final_layernorm.weight": "model-00002-of-00002.safetensors",
11
+ "model.layers.0.input_layernorm.bias": "model-00001-of-00002.safetensors",
12
+ "model.layers.0.input_layernorm.weight": "model-00001-of-00002.safetensors",
13
+ "model.layers.0.mlp.fc1.bias": "model-00001-of-00002.safetensors",
14
+ "model.layers.0.mlp.fc1.weight": "model-00001-of-00002.safetensors",
15
+ "model.layers.0.mlp.fc2.bias": "model-00001-of-00002.safetensors",
16
+ "model.layers.0.mlp.fc2.weight": "model-00001-of-00002.safetensors",
17
+ "model.layers.0.self_attn.dense.bias": "model-00001-of-00002.safetensors",
18
+ "model.layers.0.self_attn.dense.weight": "model-00001-of-00002.safetensors",
19
+ "model.layers.0.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
20
+ "model.layers.0.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
21
+ "model.layers.0.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
22
+ "model.layers.0.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
23
+ "model.layers.0.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
24
+ "model.layers.0.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
25
+ "model.layers.1.input_layernorm.bias": "model-00001-of-00002.safetensors",
26
+ "model.layers.1.input_layernorm.weight": "model-00001-of-00002.safetensors",
27
+ "model.layers.1.mlp.fc1.bias": "model-00001-of-00002.safetensors",
28
+ "model.layers.1.mlp.fc1.weight": "model-00001-of-00002.safetensors",
29
+ "model.layers.1.mlp.fc2.bias": "model-00001-of-00002.safetensors",
30
+ "model.layers.1.mlp.fc2.weight": "model-00001-of-00002.safetensors",
31
+ "model.layers.1.self_attn.dense.bias": "model-00001-of-00002.safetensors",
32
+ "model.layers.1.self_attn.dense.weight": "model-00001-of-00002.safetensors",
33
+ "model.layers.1.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
34
+ "model.layers.1.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
35
+ "model.layers.1.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
36
+ "model.layers.1.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
37
+ "model.layers.1.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
38
+ "model.layers.1.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
39
+ "model.layers.10.input_layernorm.bias": "model-00001-of-00002.safetensors",
40
+ "model.layers.10.input_layernorm.weight": "model-00001-of-00002.safetensors",
41
+ "model.layers.10.mlp.fc1.bias": "model-00001-of-00002.safetensors",
42
+ "model.layers.10.mlp.fc1.weight": "model-00001-of-00002.safetensors",
43
+ "model.layers.10.mlp.fc2.bias": "model-00001-of-00002.safetensors",
44
+ "model.layers.10.mlp.fc2.weight": "model-00001-of-00002.safetensors",
45
+ "model.layers.10.self_attn.dense.bias": "model-00001-of-00002.safetensors",
46
+ "model.layers.10.self_attn.dense.weight": "model-00001-of-00002.safetensors",
47
+ "model.layers.10.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
48
+ "model.layers.10.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
49
+ "model.layers.10.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
50
+ "model.layers.10.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
51
+ "model.layers.10.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
52
+ "model.layers.10.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
53
+ "model.layers.11.input_layernorm.bias": "model-00001-of-00002.safetensors",
54
+ "model.layers.11.input_layernorm.weight": "model-00001-of-00002.safetensors",
55
+ "model.layers.11.mlp.fc1.bias": "model-00001-of-00002.safetensors",
56
+ "model.layers.11.mlp.fc1.weight": "model-00001-of-00002.safetensors",
57
+ "model.layers.11.mlp.fc2.bias": "model-00001-of-00002.safetensors",
58
+ "model.layers.11.mlp.fc2.weight": "model-00001-of-00002.safetensors",
59
+ "model.layers.11.self_attn.dense.bias": "model-00001-of-00002.safetensors",
60
+ "model.layers.11.self_attn.dense.weight": "model-00001-of-00002.safetensors",
61
+ "model.layers.11.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
62
+ "model.layers.11.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
63
+ "model.layers.11.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
64
+ "model.layers.11.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
65
+ "model.layers.11.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
66
+ "model.layers.11.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
67
+ "model.layers.12.input_layernorm.bias": "model-00001-of-00002.safetensors",
68
+ "model.layers.12.input_layernorm.weight": "model-00001-of-00002.safetensors",
69
+ "model.layers.12.mlp.fc1.bias": "model-00001-of-00002.safetensors",
70
+ "model.layers.12.mlp.fc1.weight": "model-00001-of-00002.safetensors",
71
+ "model.layers.12.mlp.fc2.bias": "model-00001-of-00002.safetensors",
72
+ "model.layers.12.mlp.fc2.weight": "model-00001-of-00002.safetensors",
73
+ "model.layers.12.self_attn.dense.bias": "model-00001-of-00002.safetensors",
74
+ "model.layers.12.self_attn.dense.weight": "model-00001-of-00002.safetensors",
75
+ "model.layers.12.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
76
+ "model.layers.12.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
77
+ "model.layers.12.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
78
+ "model.layers.12.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
79
+ "model.layers.12.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
80
+ "model.layers.12.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
81
+ "model.layers.13.input_layernorm.bias": "model-00001-of-00002.safetensors",
82
+ "model.layers.13.input_layernorm.weight": "model-00001-of-00002.safetensors",
83
+ "model.layers.13.mlp.fc1.bias": "model-00001-of-00002.safetensors",
84
+ "model.layers.13.mlp.fc1.weight": "model-00001-of-00002.safetensors",
85
+ "model.layers.13.mlp.fc2.bias": "model-00001-of-00002.safetensors",
86
+ "model.layers.13.mlp.fc2.weight": "model-00001-of-00002.safetensors",
87
+ "model.layers.13.self_attn.dense.bias": "model-00001-of-00002.safetensors",
88
+ "model.layers.13.self_attn.dense.weight": "model-00001-of-00002.safetensors",
89
+ "model.layers.13.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
90
+ "model.layers.13.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
91
+ "model.layers.13.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
92
+ "model.layers.13.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
93
+ "model.layers.13.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
94
+ "model.layers.13.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
95
+ "model.layers.14.input_layernorm.bias": "model-00001-of-00002.safetensors",
96
+ "model.layers.14.input_layernorm.weight": "model-00001-of-00002.safetensors",
97
+ "model.layers.14.mlp.fc1.bias": "model-00001-of-00002.safetensors",
98
+ "model.layers.14.mlp.fc1.weight": "model-00001-of-00002.safetensors",
99
+ "model.layers.14.mlp.fc2.bias": "model-00001-of-00002.safetensors",
100
+ "model.layers.14.mlp.fc2.weight": "model-00001-of-00002.safetensors",
101
+ "model.layers.14.self_attn.dense.bias": "model-00001-of-00002.safetensors",
102
+ "model.layers.14.self_attn.dense.weight": "model-00001-of-00002.safetensors",
103
+ "model.layers.14.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
104
+ "model.layers.14.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
105
+ "model.layers.14.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
106
+ "model.layers.14.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
107
+ "model.layers.14.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
108
+ "model.layers.14.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
109
+ "model.layers.15.input_layernorm.bias": "model-00001-of-00002.safetensors",
110
+ "model.layers.15.input_layernorm.weight": "model-00001-of-00002.safetensors",
111
+ "model.layers.15.mlp.fc1.bias": "model-00001-of-00002.safetensors",
112
+ "model.layers.15.mlp.fc1.weight": "model-00001-of-00002.safetensors",
113
+ "model.layers.15.mlp.fc2.bias": "model-00001-of-00002.safetensors",
114
+ "model.layers.15.mlp.fc2.weight": "model-00001-of-00002.safetensors",
115
+ "model.layers.15.self_attn.dense.bias": "model-00001-of-00002.safetensors",
116
+ "model.layers.15.self_attn.dense.weight": "model-00001-of-00002.safetensors",
117
+ "model.layers.15.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
118
+ "model.layers.15.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
119
+ "model.layers.15.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
120
+ "model.layers.15.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
121
+ "model.layers.15.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
122
+ "model.layers.15.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
123
+ "model.layers.16.input_layernorm.bias": "model-00001-of-00002.safetensors",
124
+ "model.layers.16.input_layernorm.weight": "model-00001-of-00002.safetensors",
125
+ "model.layers.16.mlp.fc1.bias": "model-00001-of-00002.safetensors",
126
+ "model.layers.16.mlp.fc1.weight": "model-00001-of-00002.safetensors",
127
+ "model.layers.16.mlp.fc2.bias": "model-00001-of-00002.safetensors",
128
+ "model.layers.16.mlp.fc2.weight": "model-00001-of-00002.safetensors",
129
+ "model.layers.16.self_attn.dense.bias": "model-00001-of-00002.safetensors",
130
+ "model.layers.16.self_attn.dense.weight": "model-00001-of-00002.safetensors",
131
+ "model.layers.16.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
132
+ "model.layers.16.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
133
+ "model.layers.16.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
134
+ "model.layers.16.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
135
+ "model.layers.16.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
136
+ "model.layers.16.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
137
+ "model.layers.17.input_layernorm.bias": "model-00001-of-00002.safetensors",
138
+ "model.layers.17.input_layernorm.weight": "model-00001-of-00002.safetensors",
139
+ "model.layers.17.mlp.fc1.bias": "model-00001-of-00002.safetensors",
140
+ "model.layers.17.mlp.fc1.weight": "model-00001-of-00002.safetensors",
141
+ "model.layers.17.mlp.fc2.bias": "model-00001-of-00002.safetensors",
142
+ "model.layers.17.mlp.fc2.weight": "model-00001-of-00002.safetensors",
143
+ "model.layers.17.self_attn.dense.bias": "model-00001-of-00002.safetensors",
144
+ "model.layers.17.self_attn.dense.weight": "model-00001-of-00002.safetensors",
145
+ "model.layers.17.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
146
+ "model.layers.17.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
147
+ "model.layers.17.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
148
+ "model.layers.17.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
149
+ "model.layers.17.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
150
+ "model.layers.17.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
151
+ "model.layers.18.input_layernorm.bias": "model-00001-of-00002.safetensors",
152
+ "model.layers.18.input_layernorm.weight": "model-00001-of-00002.safetensors",
153
+ "model.layers.18.mlp.fc1.bias": "model-00001-of-00002.safetensors",
154
+ "model.layers.18.mlp.fc1.weight": "model-00001-of-00002.safetensors",
155
+ "model.layers.18.mlp.fc2.bias": "model-00001-of-00002.safetensors",
156
+ "model.layers.18.mlp.fc2.weight": "model-00001-of-00002.safetensors",
157
+ "model.layers.18.self_attn.dense.bias": "model-00001-of-00002.safetensors",
158
+ "model.layers.18.self_attn.dense.weight": "model-00001-of-00002.safetensors",
159
+ "model.layers.18.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
160
+ "model.layers.18.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
161
+ "model.layers.18.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
162
+ "model.layers.18.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
163
+ "model.layers.18.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
164
+ "model.layers.18.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
165
+ "model.layers.19.input_layernorm.bias": "model-00001-of-00002.safetensors",
166
+ "model.layers.19.input_layernorm.weight": "model-00001-of-00002.safetensors",
167
+ "model.layers.19.mlp.fc1.bias": "model-00001-of-00002.safetensors",
168
+ "model.layers.19.mlp.fc1.weight": "model-00001-of-00002.safetensors",
169
+ "model.layers.19.mlp.fc2.bias": "model-00001-of-00002.safetensors",
170
+ "model.layers.19.mlp.fc2.weight": "model-00001-of-00002.safetensors",
171
+ "model.layers.19.self_attn.dense.bias": "model-00001-of-00002.safetensors",
172
+ "model.layers.19.self_attn.dense.weight": "model-00001-of-00002.safetensors",
173
+ "model.layers.19.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
174
+ "model.layers.19.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
175
+ "model.layers.19.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
176
+ "model.layers.19.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
177
+ "model.layers.19.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
178
+ "model.layers.19.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
179
+ "model.layers.2.input_layernorm.bias": "model-00001-of-00002.safetensors",
180
+ "model.layers.2.input_layernorm.weight": "model-00001-of-00002.safetensors",
181
+ "model.layers.2.mlp.fc1.bias": "model-00001-of-00002.safetensors",
182
+ "model.layers.2.mlp.fc1.weight": "model-00001-of-00002.safetensors",
183
+ "model.layers.2.mlp.fc2.bias": "model-00001-of-00002.safetensors",
184
+ "model.layers.2.mlp.fc2.weight": "model-00001-of-00002.safetensors",
185
+ "model.layers.2.self_attn.dense.bias": "model-00001-of-00002.safetensors",
186
+ "model.layers.2.self_attn.dense.weight": "model-00001-of-00002.safetensors",
187
+ "model.layers.2.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
188
+ "model.layers.2.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
189
+ "model.layers.2.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
190
+ "model.layers.2.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
191
+ "model.layers.2.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
192
+ "model.layers.2.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
193
+ "model.layers.20.input_layernorm.bias": "model-00001-of-00002.safetensors",
194
+ "model.layers.20.input_layernorm.weight": "model-00001-of-00002.safetensors",
195
+ "model.layers.20.mlp.fc1.bias": "model-00001-of-00002.safetensors",
196
+ "model.layers.20.mlp.fc1.weight": "model-00001-of-00002.safetensors",
197
+ "model.layers.20.mlp.fc2.bias": "model-00001-of-00002.safetensors",
198
+ "model.layers.20.mlp.fc2.weight": "model-00001-of-00002.safetensors",
199
+ "model.layers.20.self_attn.dense.bias": "model-00001-of-00002.safetensors",
200
+ "model.layers.20.self_attn.dense.weight": "model-00001-of-00002.safetensors",
201
+ "model.layers.20.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
202
+ "model.layers.20.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
203
+ "model.layers.20.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
204
+ "model.layers.20.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
205
+ "model.layers.20.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
206
+ "model.layers.20.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
207
+ "model.layers.21.input_layernorm.bias": "model-00001-of-00002.safetensors",
208
+ "model.layers.21.input_layernorm.weight": "model-00001-of-00002.safetensors",
209
+ "model.layers.21.mlp.fc1.bias": "model-00001-of-00002.safetensors",
210
+ "model.layers.21.mlp.fc1.weight": "model-00001-of-00002.safetensors",
211
+ "model.layers.21.mlp.fc2.bias": "model-00001-of-00002.safetensors",
212
+ "model.layers.21.mlp.fc2.weight": "model-00001-of-00002.safetensors",
213
+ "model.layers.21.self_attn.dense.bias": "model-00001-of-00002.safetensors",
214
+ "model.layers.21.self_attn.dense.weight": "model-00001-of-00002.safetensors",
215
+ "model.layers.21.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
216
+ "model.layers.21.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
217
+ "model.layers.21.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
218
+ "model.layers.21.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
219
+ "model.layers.21.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
220
+ "model.layers.21.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
221
+ "model.layers.22.input_layernorm.bias": "model-00002-of-00002.safetensors",
222
+ "model.layers.22.input_layernorm.weight": "model-00002-of-00002.safetensors",
223
+ "model.layers.22.mlp.fc1.bias": "model-00001-of-00002.safetensors",
224
+ "model.layers.22.mlp.fc1.weight": "model-00001-of-00002.safetensors",
225
+ "model.layers.22.mlp.fc2.bias": "model-00002-of-00002.safetensors",
226
+ "model.layers.22.mlp.fc2.weight": "model-00002-of-00002.safetensors",
227
+ "model.layers.22.self_attn.dense.bias": "model-00001-of-00002.safetensors",
228
+ "model.layers.22.self_attn.dense.weight": "model-00001-of-00002.safetensors",
229
+ "model.layers.22.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
230
+ "model.layers.22.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
231
+ "model.layers.22.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
232
+ "model.layers.22.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
233
+ "model.layers.22.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
234
+ "model.layers.22.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
235
+ "model.layers.23.input_layernorm.bias": "model-00002-of-00002.safetensors",
236
+ "model.layers.23.input_layernorm.weight": "model-00002-of-00002.safetensors",
237
+ "model.layers.23.mlp.fc1.bias": "model-00002-of-00002.safetensors",
238
+ "model.layers.23.mlp.fc1.weight": "model-00002-of-00002.safetensors",
239
+ "model.layers.23.mlp.fc2.bias": "model-00002-of-00002.safetensors",
240
+ "model.layers.23.mlp.fc2.weight": "model-00002-of-00002.safetensors",
241
+ "model.layers.23.self_attn.dense.bias": "model-00002-of-00002.safetensors",
242
+ "model.layers.23.self_attn.dense.weight": "model-00002-of-00002.safetensors",
243
+ "model.layers.23.self_attn.k_proj.bias": "model-00002-of-00002.safetensors",
244
+ "model.layers.23.self_attn.k_proj.weight": "model-00002-of-00002.safetensors",
245
+ "model.layers.23.self_attn.q_proj.bias": "model-00002-of-00002.safetensors",
246
+ "model.layers.23.self_attn.q_proj.weight": "model-00002-of-00002.safetensors",
247
+ "model.layers.23.self_attn.v_proj.bias": "model-00002-of-00002.safetensors",
248
+ "model.layers.23.self_attn.v_proj.weight": "model-00002-of-00002.safetensors",
249
+ "model.layers.3.input_layernorm.bias": "model-00001-of-00002.safetensors",
250
+ "model.layers.3.input_layernorm.weight": "model-00001-of-00002.safetensors",
251
+ "model.layers.3.mlp.fc1.bias": "model-00001-of-00002.safetensors",
252
+ "model.layers.3.mlp.fc1.weight": "model-00001-of-00002.safetensors",
253
+ "model.layers.3.mlp.fc2.bias": "model-00001-of-00002.safetensors",
254
+ "model.layers.3.mlp.fc2.weight": "model-00001-of-00002.safetensors",
255
+ "model.layers.3.self_attn.dense.bias": "model-00001-of-00002.safetensors",
256
+ "model.layers.3.self_attn.dense.weight": "model-00001-of-00002.safetensors",
257
+ "model.layers.3.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
258
+ "model.layers.3.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
259
+ "model.layers.3.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
260
+ "model.layers.3.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
261
+ "model.layers.3.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
262
+ "model.layers.3.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
263
+ "model.layers.4.input_layernorm.bias": "model-00001-of-00002.safetensors",
264
+ "model.layers.4.input_layernorm.weight": "model-00001-of-00002.safetensors",
265
+ "model.layers.4.mlp.fc1.bias": "model-00001-of-00002.safetensors",
266
+ "model.layers.4.mlp.fc1.weight": "model-00001-of-00002.safetensors",
267
+ "model.layers.4.mlp.fc2.bias": "model-00001-of-00002.safetensors",
268
+ "model.layers.4.mlp.fc2.weight": "model-00001-of-00002.safetensors",
269
+ "model.layers.4.self_attn.dense.bias": "model-00001-of-00002.safetensors",
270
+ "model.layers.4.self_attn.dense.weight": "model-00001-of-00002.safetensors",
271
+ "model.layers.4.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
272
+ "model.layers.4.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
273
+ "model.layers.4.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
274
+ "model.layers.4.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
275
+ "model.layers.4.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
276
+ "model.layers.4.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
277
+ "model.layers.5.input_layernorm.bias": "model-00001-of-00002.safetensors",
278
+ "model.layers.5.input_layernorm.weight": "model-00001-of-00002.safetensors",
279
+ "model.layers.5.mlp.fc1.bias": "model-00001-of-00002.safetensors",
280
+ "model.layers.5.mlp.fc1.weight": "model-00001-of-00002.safetensors",
281
+ "model.layers.5.mlp.fc2.bias": "model-00001-of-00002.safetensors",
282
+ "model.layers.5.mlp.fc2.weight": "model-00001-of-00002.safetensors",
283
+ "model.layers.5.self_attn.dense.bias": "model-00001-of-00002.safetensors",
284
+ "model.layers.5.self_attn.dense.weight": "model-00001-of-00002.safetensors",
285
+ "model.layers.5.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
286
+ "model.layers.5.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
287
+ "model.layers.5.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
288
+ "model.layers.5.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
289
+ "model.layers.5.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
290
+ "model.layers.5.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
291
+ "model.layers.6.input_layernorm.bias": "model-00001-of-00002.safetensors",
292
+ "model.layers.6.input_layernorm.weight": "model-00001-of-00002.safetensors",
293
+ "model.layers.6.mlp.fc1.bias": "model-00001-of-00002.safetensors",
294
+ "model.layers.6.mlp.fc1.weight": "model-00001-of-00002.safetensors",
295
+ "model.layers.6.mlp.fc2.bias": "model-00001-of-00002.safetensors",
296
+ "model.layers.6.mlp.fc2.weight": "model-00001-of-00002.safetensors",
297
+ "model.layers.6.self_attn.dense.bias": "model-00001-of-00002.safetensors",
298
+ "model.layers.6.self_attn.dense.weight": "model-00001-of-00002.safetensors",
299
+ "model.layers.6.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
300
+ "model.layers.6.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
301
+ "model.layers.6.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
302
+ "model.layers.6.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
303
+ "model.layers.6.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
304
+ "model.layers.6.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
305
+ "model.layers.7.input_layernorm.bias": "model-00001-of-00002.safetensors",
306
+ "model.layers.7.input_layernorm.weight": "model-00001-of-00002.safetensors",
307
+ "model.layers.7.mlp.fc1.bias": "model-00001-of-00002.safetensors",
308
+ "model.layers.7.mlp.fc1.weight": "model-00001-of-00002.safetensors",
309
+ "model.layers.7.mlp.fc2.bias": "model-00001-of-00002.safetensors",
310
+ "model.layers.7.mlp.fc2.weight": "model-00001-of-00002.safetensors",
311
+ "model.layers.7.self_attn.dense.bias": "model-00001-of-00002.safetensors",
312
+ "model.layers.7.self_attn.dense.weight": "model-00001-of-00002.safetensors",
313
+ "model.layers.7.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
314
+ "model.layers.7.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
315
+ "model.layers.7.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
316
+ "model.layers.7.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
317
+ "model.layers.7.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
318
+ "model.layers.7.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
319
+ "model.layers.8.input_layernorm.bias": "model-00001-of-00002.safetensors",
320
+ "model.layers.8.input_layernorm.weight": "model-00001-of-00002.safetensors",
321
+ "model.layers.8.mlp.fc1.bias": "model-00001-of-00002.safetensors",
322
+ "model.layers.8.mlp.fc1.weight": "model-00001-of-00002.safetensors",
323
+ "model.layers.8.mlp.fc2.bias": "model-00001-of-00002.safetensors",
324
+ "model.layers.8.mlp.fc2.weight": "model-00001-of-00002.safetensors",
325
+ "model.layers.8.self_attn.dense.bias": "model-00001-of-00002.safetensors",
326
+ "model.layers.8.self_attn.dense.weight": "model-00001-of-00002.safetensors",
327
+ "model.layers.8.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
328
+ "model.layers.8.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
329
+ "model.layers.8.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
330
+ "model.layers.8.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
331
+ "model.layers.8.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
332
+ "model.layers.8.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
333
+ "model.layers.9.input_layernorm.bias": "model-00001-of-00002.safetensors",
334
+ "model.layers.9.input_layernorm.weight": "model-00001-of-00002.safetensors",
335
+ "model.layers.9.mlp.fc1.bias": "model-00001-of-00002.safetensors",
336
+ "model.layers.9.mlp.fc1.weight": "model-00001-of-00002.safetensors",
337
+ "model.layers.9.mlp.fc2.bias": "model-00001-of-00002.safetensors",
338
+ "model.layers.9.mlp.fc2.weight": "model-00001-of-00002.safetensors",
339
+ "model.layers.9.self_attn.dense.bias": "model-00001-of-00002.safetensors",
340
+ "model.layers.9.self_attn.dense.weight": "model-00001-of-00002.safetensors",
341
+ "model.layers.9.self_attn.k_proj.bias": "model-00001-of-00002.safetensors",
342
+ "model.layers.9.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
343
+ "model.layers.9.self_attn.q_proj.bias": "model-00001-of-00002.safetensors",
344
+ "model.layers.9.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
345
+ "model.layers.9.self_attn.v_proj.bias": "model-00001-of-00002.safetensors",
346
+ "model.layers.9.self_attn.v_proj.weight": "model-00001-of-00002.safetensors"
347
+ }
348
+ }
rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f8ff17ad8275aaab889de49950978c13c1e8be2f90923bebd009f6bb2324c58b
3
+ size 14244
trainer_state.json ADDED
@@ -0,0 +1,771 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 1.820250284414107,
5
+ "eval_steps": 200,
6
+ "global_step": 5000,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 0.04,
13
+ "grad_norm": 1.7517368793487549,
14
+ "learning_rate": 9.999e-07,
15
+ "loss": 0.2253,
16
+ "step": 100
17
+ },
18
+ {
19
+ "epoch": 0.07,
20
+ "grad_norm": 1.4760167598724365,
21
+ "learning_rate": 9.998e-07,
22
+ "loss": 0.1588,
23
+ "step": 200
24
+ },
25
+ {
26
+ "epoch": 0.07,
27
+ "eval_validation_loss": 0.1540723592042923,
28
+ "eval_validation_runtime": 386.0092,
29
+ "eval_validation_samples_per_second": 3.0,
30
+ "eval_validation_steps_per_second": 0.376,
31
+ "step": 200
32
+ },
33
+ {
34
+ "epoch": 0.07,
35
+ "eval_benchmark_loss": 0.3658876121044159,
36
+ "eval_benchmark_runtime": 9.4993,
37
+ "eval_benchmark_samples_per_second": 2.737,
38
+ "eval_benchmark_steps_per_second": 0.421,
39
+ "step": 200
40
+ },
41
+ {
42
+ "epoch": 0.11,
43
+ "grad_norm": 1.4102065563201904,
44
+ "learning_rate": 9.997e-07,
45
+ "loss": 0.1423,
46
+ "step": 300
47
+ },
48
+ {
49
+ "epoch": 0.15,
50
+ "grad_norm": 1.657212495803833,
51
+ "learning_rate": 9.996e-07,
52
+ "loss": 0.1345,
53
+ "step": 400
54
+ },
55
+ {
56
+ "epoch": 0.15,
57
+ "eval_validation_loss": 0.13775885105133057,
58
+ "eval_validation_runtime": 383.1365,
59
+ "eval_validation_samples_per_second": 3.022,
60
+ "eval_validation_steps_per_second": 0.378,
61
+ "step": 400
62
+ },
63
+ {
64
+ "epoch": 0.15,
65
+ "eval_benchmark_loss": 0.3596523106098175,
66
+ "eval_benchmark_runtime": 9.4551,
67
+ "eval_benchmark_samples_per_second": 2.75,
68
+ "eval_benchmark_steps_per_second": 0.423,
69
+ "step": 400
70
+ },
71
+ {
72
+ "epoch": 0.18,
73
+ "grad_norm": 1.1109451055526733,
74
+ "learning_rate": 9.995e-07,
75
+ "loss": 0.1278,
76
+ "step": 500
77
+ },
78
+ {
79
+ "epoch": 0.22,
80
+ "grad_norm": 1.219743013381958,
81
+ "learning_rate": 9.994e-07,
82
+ "loss": 0.128,
83
+ "step": 600
84
+ },
85
+ {
86
+ "epoch": 0.22,
87
+ "eval_validation_loss": 0.1310659945011139,
88
+ "eval_validation_runtime": 382.987,
89
+ "eval_validation_samples_per_second": 3.024,
90
+ "eval_validation_steps_per_second": 0.379,
91
+ "step": 600
92
+ },
93
+ {
94
+ "epoch": 0.22,
95
+ "eval_benchmark_loss": 0.35413601994514465,
96
+ "eval_benchmark_runtime": 9.452,
97
+ "eval_benchmark_samples_per_second": 2.751,
98
+ "eval_benchmark_steps_per_second": 0.423,
99
+ "step": 600
100
+ },
101
+ {
102
+ "epoch": 0.25,
103
+ "grad_norm": 1.3631513118743896,
104
+ "learning_rate": 9.993e-07,
105
+ "loss": 0.1251,
106
+ "step": 700
107
+ },
108
+ {
109
+ "epoch": 0.29,
110
+ "grad_norm": 1.6779521703720093,
111
+ "learning_rate": 9.992e-07,
112
+ "loss": 0.1198,
113
+ "step": 800
114
+ },
115
+ {
116
+ "epoch": 0.29,
117
+ "eval_validation_loss": 0.12629443407058716,
118
+ "eval_validation_runtime": 383.2491,
119
+ "eval_validation_samples_per_second": 3.022,
120
+ "eval_validation_steps_per_second": 0.378,
121
+ "step": 800
122
+ },
123
+ {
124
+ "epoch": 0.29,
125
+ "eval_benchmark_loss": 0.35233384370803833,
126
+ "eval_benchmark_runtime": 9.4696,
127
+ "eval_benchmark_samples_per_second": 2.746,
128
+ "eval_benchmark_steps_per_second": 0.422,
129
+ "step": 800
130
+ },
131
+ {
132
+ "epoch": 0.33,
133
+ "grad_norm": 2.0499491691589355,
134
+ "learning_rate": 9.990999999999999e-07,
135
+ "loss": 0.1127,
136
+ "step": 900
137
+ },
138
+ {
139
+ "epoch": 0.36,
140
+ "grad_norm": 2.0875236988067627,
141
+ "learning_rate": 9.989999999999999e-07,
142
+ "loss": 0.1227,
143
+ "step": 1000
144
+ },
145
+ {
146
+ "epoch": 0.36,
147
+ "eval_validation_loss": 0.12267088890075684,
148
+ "eval_validation_runtime": 383.4228,
149
+ "eval_validation_samples_per_second": 3.02,
150
+ "eval_validation_steps_per_second": 0.378,
151
+ "step": 1000
152
+ },
153
+ {
154
+ "epoch": 0.36,
155
+ "eval_benchmark_loss": 0.35131120681762695,
156
+ "eval_benchmark_runtime": 9.4614,
157
+ "eval_benchmark_samples_per_second": 2.748,
158
+ "eval_benchmark_steps_per_second": 0.423,
159
+ "step": 1000
160
+ },
161
+ {
162
+ "epoch": 0.4,
163
+ "grad_norm": 2.061170816421509,
164
+ "learning_rate": 9.988999999999999e-07,
165
+ "loss": 0.1178,
166
+ "step": 1100
167
+ },
168
+ {
169
+ "epoch": 0.44,
170
+ "grad_norm": 1.5514030456542969,
171
+ "learning_rate": 9.988e-07,
172
+ "loss": 0.1157,
173
+ "step": 1200
174
+ },
175
+ {
176
+ "epoch": 0.44,
177
+ "eval_validation_loss": 0.12000161409378052,
178
+ "eval_validation_runtime": 383.1507,
179
+ "eval_validation_samples_per_second": 3.022,
180
+ "eval_validation_steps_per_second": 0.378,
181
+ "step": 1200
182
+ },
183
+ {
184
+ "epoch": 0.44,
185
+ "eval_benchmark_loss": 0.3518659174442291,
186
+ "eval_benchmark_runtime": 9.4808,
187
+ "eval_benchmark_samples_per_second": 2.742,
188
+ "eval_benchmark_steps_per_second": 0.422,
189
+ "step": 1200
190
+ },
191
+ {
192
+ "epoch": 0.47,
193
+ "grad_norm": 1.9566336870193481,
194
+ "learning_rate": 9.987e-07,
195
+ "loss": 0.1114,
196
+ "step": 1300
197
+ },
198
+ {
199
+ "epoch": 0.51,
200
+ "grad_norm": 1.365792155265808,
201
+ "learning_rate": 9.986e-07,
202
+ "loss": 0.115,
203
+ "step": 1400
204
+ },
205
+ {
206
+ "epoch": 0.51,
207
+ "eval_validation_loss": 0.117983378469944,
208
+ "eval_validation_runtime": 383.1539,
209
+ "eval_validation_samples_per_second": 3.022,
210
+ "eval_validation_steps_per_second": 0.378,
211
+ "step": 1400
212
+ },
213
+ {
214
+ "epoch": 0.51,
215
+ "eval_benchmark_loss": 0.3509116470813751,
216
+ "eval_benchmark_runtime": 9.4708,
217
+ "eval_benchmark_samples_per_second": 2.745,
218
+ "eval_benchmark_steps_per_second": 0.422,
219
+ "step": 1400
220
+ },
221
+ {
222
+ "epoch": 0.55,
223
+ "grad_norm": 1.6281362771987915,
224
+ "learning_rate": 9.985e-07,
225
+ "loss": 0.1192,
226
+ "step": 1500
227
+ },
228
+ {
229
+ "epoch": 0.58,
230
+ "grad_norm": 1.7361140251159668,
231
+ "learning_rate": 9.983999999999998e-07,
232
+ "loss": 0.1114,
233
+ "step": 1600
234
+ },
235
+ {
236
+ "epoch": 0.58,
237
+ "eval_validation_loss": 0.11662287265062332,
238
+ "eval_validation_runtime": 383.2086,
239
+ "eval_validation_samples_per_second": 3.022,
240
+ "eval_validation_steps_per_second": 0.378,
241
+ "step": 1600
242
+ },
243
+ {
244
+ "epoch": 0.58,
245
+ "eval_benchmark_loss": 0.3516782522201538,
246
+ "eval_benchmark_runtime": 9.4592,
247
+ "eval_benchmark_samples_per_second": 2.749,
248
+ "eval_benchmark_steps_per_second": 0.423,
249
+ "step": 1600
250
+ },
251
+ {
252
+ "epoch": 0.62,
253
+ "grad_norm": 2.332932710647583,
254
+ "learning_rate": 9.982999999999998e-07,
255
+ "loss": 0.1128,
256
+ "step": 1700
257
+ },
258
+ {
259
+ "epoch": 0.66,
260
+ "grad_norm": 1.4797985553741455,
261
+ "learning_rate": 9.982e-07,
262
+ "loss": 0.1036,
263
+ "step": 1800
264
+ },
265
+ {
266
+ "epoch": 0.66,
267
+ "eval_validation_loss": 0.11438746005296707,
268
+ "eval_validation_runtime": 383.1904,
269
+ "eval_validation_samples_per_second": 3.022,
270
+ "eval_validation_steps_per_second": 0.378,
271
+ "step": 1800
272
+ },
273
+ {
274
+ "epoch": 0.66,
275
+ "eval_benchmark_loss": 0.34705713391304016,
276
+ "eval_benchmark_runtime": 9.4646,
277
+ "eval_benchmark_samples_per_second": 2.747,
278
+ "eval_benchmark_steps_per_second": 0.423,
279
+ "step": 1800
280
+ },
281
+ {
282
+ "epoch": 0.69,
283
+ "grad_norm": 1.546353816986084,
284
+ "learning_rate": 9.981e-07,
285
+ "loss": 0.1132,
286
+ "step": 1900
287
+ },
288
+ {
289
+ "epoch": 0.73,
290
+ "grad_norm": 2.1084446907043457,
291
+ "learning_rate": 9.98e-07,
292
+ "loss": 0.1125,
293
+ "step": 2000
294
+ },
295
+ {
296
+ "epoch": 0.73,
297
+ "eval_validation_loss": 0.11280551552772522,
298
+ "eval_validation_runtime": 383.2179,
299
+ "eval_validation_samples_per_second": 3.022,
300
+ "eval_validation_steps_per_second": 0.378,
301
+ "step": 2000
302
+ },
303
+ {
304
+ "epoch": 0.73,
305
+ "eval_benchmark_loss": 0.34782856702804565,
306
+ "eval_benchmark_runtime": 9.4577,
307
+ "eval_benchmark_samples_per_second": 2.749,
308
+ "eval_benchmark_steps_per_second": 0.423,
309
+ "step": 2000
310
+ },
311
+ {
312
+ "epoch": 0.76,
313
+ "grad_norm": 1.3358793258666992,
314
+ "learning_rate": 9.979e-07,
315
+ "loss": 0.1155,
316
+ "step": 2100
317
+ },
318
+ {
319
+ "epoch": 0.8,
320
+ "grad_norm": 1.6584503650665283,
321
+ "learning_rate": 9.978e-07,
322
+ "loss": 0.1034,
323
+ "step": 2200
324
+ },
325
+ {
326
+ "epoch": 0.8,
327
+ "eval_validation_loss": 0.11192985624074936,
328
+ "eval_validation_runtime": 383.2409,
329
+ "eval_validation_samples_per_second": 3.022,
330
+ "eval_validation_steps_per_second": 0.378,
331
+ "step": 2200
332
+ },
333
+ {
334
+ "epoch": 0.8,
335
+ "eval_benchmark_loss": 0.3532721698284149,
336
+ "eval_benchmark_runtime": 9.4718,
337
+ "eval_benchmark_samples_per_second": 2.745,
338
+ "eval_benchmark_steps_per_second": 0.422,
339
+ "step": 2200
340
+ },
341
+ {
342
+ "epoch": 0.84,
343
+ "grad_norm": 1.2788842916488647,
344
+ "learning_rate": 9.977e-07,
345
+ "loss": 0.1091,
346
+ "step": 2300
347
+ },
348
+ {
349
+ "epoch": 0.87,
350
+ "grad_norm": 1.3015437126159668,
351
+ "learning_rate": 9.976e-07,
352
+ "loss": 0.1024,
353
+ "step": 2400
354
+ },
355
+ {
356
+ "epoch": 0.87,
357
+ "eval_validation_loss": 0.11032804101705551,
358
+ "eval_validation_runtime": 383.3623,
359
+ "eval_validation_samples_per_second": 3.021,
360
+ "eval_validation_steps_per_second": 0.378,
361
+ "step": 2400
362
+ },
363
+ {
364
+ "epoch": 0.87,
365
+ "eval_benchmark_loss": 0.34818902611732483,
366
+ "eval_benchmark_runtime": 9.4633,
367
+ "eval_benchmark_samples_per_second": 2.747,
368
+ "eval_benchmark_steps_per_second": 0.423,
369
+ "step": 2400
370
+ },
371
+ {
372
+ "epoch": 0.91,
373
+ "grad_norm": 1.2690401077270508,
374
+ "learning_rate": 9.975e-07,
375
+ "loss": 0.0981,
376
+ "step": 2500
377
+ },
378
+ {
379
+ "epoch": 0.95,
380
+ "grad_norm": 1.4752624034881592,
381
+ "learning_rate": 9.974e-07,
382
+ "loss": 0.1036,
383
+ "step": 2600
384
+ },
385
+ {
386
+ "epoch": 0.95,
387
+ "eval_validation_loss": 0.10945109277963638,
388
+ "eval_validation_runtime": 383.3638,
389
+ "eval_validation_samples_per_second": 3.021,
390
+ "eval_validation_steps_per_second": 0.378,
391
+ "step": 2600
392
+ },
393
+ {
394
+ "epoch": 0.95,
395
+ "eval_benchmark_loss": 0.3477522134780884,
396
+ "eval_benchmark_runtime": 9.4688,
397
+ "eval_benchmark_samples_per_second": 2.746,
398
+ "eval_benchmark_steps_per_second": 0.422,
399
+ "step": 2600
400
+ },
401
+ {
402
+ "epoch": 0.98,
403
+ "grad_norm": 1.502576231956482,
404
+ "learning_rate": 9.973e-07,
405
+ "loss": 0.11,
406
+ "step": 2700
407
+ },
408
+ {
409
+ "epoch": 1.02,
410
+ "grad_norm": 0.9190167784690857,
411
+ "learning_rate": 9.972e-07,
412
+ "loss": 0.086,
413
+ "step": 2800
414
+ },
415
+ {
416
+ "epoch": 1.02,
417
+ "eval_validation_loss": 0.11021959781646729,
418
+ "eval_validation_runtime": 383.4523,
419
+ "eval_validation_samples_per_second": 3.02,
420
+ "eval_validation_steps_per_second": 0.378,
421
+ "step": 2800
422
+ },
423
+ {
424
+ "epoch": 1.02,
425
+ "eval_benchmark_loss": 0.35620927810668945,
426
+ "eval_benchmark_runtime": 9.4715,
427
+ "eval_benchmark_samples_per_second": 2.745,
428
+ "eval_benchmark_steps_per_second": 0.422,
429
+ "step": 2800
430
+ },
431
+ {
432
+ "epoch": 1.06,
433
+ "grad_norm": 2.0933382511138916,
434
+ "learning_rate": 9.971e-07,
435
+ "loss": 0.0849,
436
+ "step": 2900
437
+ },
438
+ {
439
+ "epoch": 1.09,
440
+ "grad_norm": 1.020569920539856,
441
+ "learning_rate": 9.97e-07,
442
+ "loss": 0.0858,
443
+ "step": 3000
444
+ },
445
+ {
446
+ "epoch": 1.09,
447
+ "eval_validation_loss": 0.10909596085548401,
448
+ "eval_validation_runtime": 383.3001,
449
+ "eval_validation_samples_per_second": 3.021,
450
+ "eval_validation_steps_per_second": 0.378,
451
+ "step": 3000
452
+ },
453
+ {
454
+ "epoch": 1.09,
455
+ "eval_benchmark_loss": 0.3547273278236389,
456
+ "eval_benchmark_runtime": 9.4578,
457
+ "eval_benchmark_samples_per_second": 2.749,
458
+ "eval_benchmark_steps_per_second": 0.423,
459
+ "step": 3000
460
+ },
461
+ {
462
+ "epoch": 1.13,
463
+ "grad_norm": 1.2632180452346802,
464
+ "learning_rate": 9.969e-07,
465
+ "loss": 0.0834,
466
+ "step": 3100
467
+ },
468
+ {
469
+ "epoch": 1.16,
470
+ "grad_norm": 1.0163979530334473,
471
+ "learning_rate": 9.968e-07,
472
+ "loss": 0.0786,
473
+ "step": 3200
474
+ },
475
+ {
476
+ "epoch": 1.16,
477
+ "eval_validation_loss": 0.11020828038454056,
478
+ "eval_validation_runtime": 383.4418,
479
+ "eval_validation_samples_per_second": 3.02,
480
+ "eval_validation_steps_per_second": 0.378,
481
+ "step": 3200
482
+ },
483
+ {
484
+ "epoch": 1.16,
485
+ "eval_benchmark_loss": 0.35691550374031067,
486
+ "eval_benchmark_runtime": 9.4772,
487
+ "eval_benchmark_samples_per_second": 2.743,
488
+ "eval_benchmark_steps_per_second": 0.422,
489
+ "step": 3200
490
+ },
491
+ {
492
+ "epoch": 1.2,
493
+ "grad_norm": 1.0731770992279053,
494
+ "learning_rate": 9.967e-07,
495
+ "loss": 0.0817,
496
+ "step": 3300
497
+ },
498
+ {
499
+ "epoch": 1.24,
500
+ "grad_norm": 1.4652243852615356,
501
+ "learning_rate": 9.966e-07,
502
+ "loss": 0.0885,
503
+ "step": 3400
504
+ },
505
+ {
506
+ "epoch": 1.24,
507
+ "eval_validation_loss": 0.10934263467788696,
508
+ "eval_validation_runtime": 383.3702,
509
+ "eval_validation_samples_per_second": 3.021,
510
+ "eval_validation_steps_per_second": 0.378,
511
+ "step": 3400
512
+ },
513
+ {
514
+ "epoch": 1.24,
515
+ "eval_benchmark_loss": 0.35489001870155334,
516
+ "eval_benchmark_runtime": 9.4613,
517
+ "eval_benchmark_samples_per_second": 2.748,
518
+ "eval_benchmark_steps_per_second": 0.423,
519
+ "step": 3400
520
+ },
521
+ {
522
+ "epoch": 1.27,
523
+ "grad_norm": 1.1992924213409424,
524
+ "learning_rate": 9.965e-07,
525
+ "loss": 0.078,
526
+ "step": 3500
527
+ },
528
+ {
529
+ "epoch": 1.31,
530
+ "grad_norm": 1.3917715549468994,
531
+ "learning_rate": 9.964e-07,
532
+ "loss": 0.0857,
533
+ "step": 3600
534
+ },
535
+ {
536
+ "epoch": 1.31,
537
+ "eval_validation_loss": 0.10760623216629028,
538
+ "eval_validation_runtime": 383.3793,
539
+ "eval_validation_samples_per_second": 3.021,
540
+ "eval_validation_steps_per_second": 0.378,
541
+ "step": 3600
542
+ },
543
+ {
544
+ "epoch": 1.31,
545
+ "eval_benchmark_loss": 0.35664084553718567,
546
+ "eval_benchmark_runtime": 9.4529,
547
+ "eval_benchmark_samples_per_second": 2.75,
548
+ "eval_benchmark_steps_per_second": 0.423,
549
+ "step": 3600
550
+ },
551
+ {
552
+ "epoch": 1.35,
553
+ "grad_norm": 0.9880099892616272,
554
+ "learning_rate": 9.962999999999999e-07,
555
+ "loss": 0.084,
556
+ "step": 3700
557
+ },
558
+ {
559
+ "epoch": 1.38,
560
+ "grad_norm": 1.15889310836792,
561
+ "learning_rate": 9.961999999999999e-07,
562
+ "loss": 0.0833,
563
+ "step": 3800
564
+ },
565
+ {
566
+ "epoch": 1.38,
567
+ "eval_validation_loss": 0.10785996913909912,
568
+ "eval_validation_runtime": 383.3503,
569
+ "eval_validation_samples_per_second": 3.021,
570
+ "eval_validation_steps_per_second": 0.378,
571
+ "step": 3800
572
+ },
573
+ {
574
+ "epoch": 1.38,
575
+ "eval_benchmark_loss": 0.3560557961463928,
576
+ "eval_benchmark_runtime": 9.457,
577
+ "eval_benchmark_samples_per_second": 2.749,
578
+ "eval_benchmark_steps_per_second": 0.423,
579
+ "step": 3800
580
+ },
581
+ {
582
+ "epoch": 1.42,
583
+ "grad_norm": 1.4265196323394775,
584
+ "learning_rate": 9.960999999999999e-07,
585
+ "loss": 0.0831,
586
+ "step": 3900
587
+ },
588
+ {
589
+ "epoch": 1.46,
590
+ "grad_norm": 1.4567248821258545,
591
+ "learning_rate": 9.959999999999999e-07,
592
+ "loss": 0.084,
593
+ "step": 4000
594
+ },
595
+ {
596
+ "epoch": 1.46,
597
+ "eval_validation_loss": 0.1076769158244133,
598
+ "eval_validation_runtime": 383.3556,
599
+ "eval_validation_samples_per_second": 3.021,
600
+ "eval_validation_steps_per_second": 0.378,
601
+ "step": 4000
602
+ },
603
+ {
604
+ "epoch": 1.46,
605
+ "eval_benchmark_loss": 0.3554157614707947,
606
+ "eval_benchmark_runtime": 9.4598,
607
+ "eval_benchmark_samples_per_second": 2.748,
608
+ "eval_benchmark_steps_per_second": 0.423,
609
+ "step": 4000
610
+ },
611
+ {
612
+ "epoch": 1.49,
613
+ "grad_norm": 1.7788864374160767,
614
+ "learning_rate": 9.958999999999999e-07,
615
+ "loss": 0.0813,
616
+ "step": 4100
617
+ },
618
+ {
619
+ "epoch": 1.53,
620
+ "grad_norm": 1.4983494281768799,
621
+ "learning_rate": 9.958e-07,
622
+ "loss": 0.0815,
623
+ "step": 4200
624
+ },
625
+ {
626
+ "epoch": 1.53,
627
+ "eval_validation_loss": 0.10664806514978409,
628
+ "eval_validation_runtime": 383.2552,
629
+ "eval_validation_samples_per_second": 3.021,
630
+ "eval_validation_steps_per_second": 0.378,
631
+ "step": 4200
632
+ },
633
+ {
634
+ "epoch": 1.53,
635
+ "eval_benchmark_loss": 0.35523322224617004,
636
+ "eval_benchmark_runtime": 9.4557,
637
+ "eval_benchmark_samples_per_second": 2.75,
638
+ "eval_benchmark_steps_per_second": 0.423,
639
+ "step": 4200
640
+ },
641
+ {
642
+ "epoch": 1.57,
643
+ "grad_norm": 0.7813511490821838,
644
+ "learning_rate": 9.957e-07,
645
+ "loss": 0.0877,
646
+ "step": 4300
647
+ },
648
+ {
649
+ "epoch": 1.6,
650
+ "grad_norm": 1.3383861780166626,
651
+ "learning_rate": 9.956e-07,
652
+ "loss": 0.0812,
653
+ "step": 4400
654
+ },
655
+ {
656
+ "epoch": 1.6,
657
+ "eval_validation_loss": 0.10593847930431366,
658
+ "eval_validation_runtime": 383.4244,
659
+ "eval_validation_samples_per_second": 3.02,
660
+ "eval_validation_steps_per_second": 0.378,
661
+ "step": 4400
662
+ },
663
+ {
664
+ "epoch": 1.6,
665
+ "eval_benchmark_loss": 0.3536509871482849,
666
+ "eval_benchmark_runtime": 9.4751,
667
+ "eval_benchmark_samples_per_second": 2.744,
668
+ "eval_benchmark_steps_per_second": 0.422,
669
+ "step": 4400
670
+ },
671
+ {
672
+ "epoch": 1.64,
673
+ "grad_norm": 1.5067020654678345,
674
+ "learning_rate": 9.955e-07,
675
+ "loss": 0.0821,
676
+ "step": 4500
677
+ },
678
+ {
679
+ "epoch": 1.67,
680
+ "grad_norm": 1.46514093875885,
681
+ "learning_rate": 9.953999999999998e-07,
682
+ "loss": 0.0841,
683
+ "step": 4600
684
+ },
685
+ {
686
+ "epoch": 1.67,
687
+ "eval_validation_loss": 0.10573519766330719,
688
+ "eval_validation_runtime": 383.3608,
689
+ "eval_validation_samples_per_second": 3.021,
690
+ "eval_validation_steps_per_second": 0.378,
691
+ "step": 4600
692
+ },
693
+ {
694
+ "epoch": 1.67,
695
+ "eval_benchmark_loss": 0.35240989923477173,
696
+ "eval_benchmark_runtime": 9.4682,
697
+ "eval_benchmark_samples_per_second": 2.746,
698
+ "eval_benchmark_steps_per_second": 0.422,
699
+ "step": 4600
700
+ },
701
+ {
702
+ "epoch": 1.71,
703
+ "grad_norm": 1.124515175819397,
704
+ "learning_rate": 9.952999999999998e-07,
705
+ "loss": 0.0833,
706
+ "step": 4700
707
+ },
708
+ {
709
+ "epoch": 1.75,
710
+ "grad_norm": 1.2566176652908325,
711
+ "learning_rate": 9.952e-07,
712
+ "loss": 0.0822,
713
+ "step": 4800
714
+ },
715
+ {
716
+ "epoch": 1.75,
717
+ "eval_validation_loss": 0.10490844398736954,
718
+ "eval_validation_runtime": 383.3754,
719
+ "eval_validation_samples_per_second": 3.021,
720
+ "eval_validation_steps_per_second": 0.378,
721
+ "step": 4800
722
+ },
723
+ {
724
+ "epoch": 1.75,
725
+ "eval_benchmark_loss": 0.3501981496810913,
726
+ "eval_benchmark_runtime": 9.4492,
727
+ "eval_benchmark_samples_per_second": 2.752,
728
+ "eval_benchmark_steps_per_second": 0.423,
729
+ "step": 4800
730
+ },
731
+ {
732
+ "epoch": 1.78,
733
+ "grad_norm": 1.4075120687484741,
734
+ "learning_rate": 9.951e-07,
735
+ "loss": 0.0824,
736
+ "step": 4900
737
+ },
738
+ {
739
+ "epoch": 1.82,
740
+ "grad_norm": 0.9538173079490662,
741
+ "learning_rate": 9.95e-07,
742
+ "loss": 0.0809,
743
+ "step": 5000
744
+ },
745
+ {
746
+ "epoch": 1.82,
747
+ "eval_validation_loss": 0.10531202703714371,
748
+ "eval_validation_runtime": 383.2794,
749
+ "eval_validation_samples_per_second": 3.021,
750
+ "eval_validation_steps_per_second": 0.378,
751
+ "step": 5000
752
+ },
753
+ {
754
+ "epoch": 1.82,
755
+ "eval_benchmark_loss": 0.3513197898864746,
756
+ "eval_benchmark_runtime": 9.4616,
757
+ "eval_benchmark_samples_per_second": 2.748,
758
+ "eval_benchmark_steps_per_second": 0.423,
759
+ "step": 5000
760
+ }
761
+ ],
762
+ "logging_steps": 100,
763
+ "max_steps": 1000000,
764
+ "num_input_tokens_seen": 0,
765
+ "num_train_epochs": 365,
766
+ "save_steps": 200,
767
+ "total_flos": 1.617395756875776e+17,
768
+ "train_batch_size": 1,
769
+ "trial_name": null,
770
+ "trial_params": null
771
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7fded323245cded55947594fb1c831172129c67b57f24c2454fa1eea80158d7a
3
+ size 4856