keeeeenw commited on
Commit
7a26ba2
·
verified ·
1 Parent(s): 88e8460

Delete checkpoints

Browse files
Files changed (23) hide show
  1. checkpoints/checkpoint-900/config.json +0 -40
  2. checkpoints/checkpoint-900/generation_config.json +0 -12
  3. checkpoints/checkpoint-900/global_step900/bf16_zero_pp_rank_0_mp_rank_00_optim_states.pt +0 -3
  4. checkpoints/checkpoint-900/global_step900/bf16_zero_pp_rank_1_mp_rank_00_optim_states.pt +0 -3
  5. checkpoints/checkpoint-900/global_step900/bf16_zero_pp_rank_2_mp_rank_00_optim_states.pt +0 -3
  6. checkpoints/checkpoint-900/global_step900/bf16_zero_pp_rank_3_mp_rank_00_optim_states.pt +0 -3
  7. checkpoints/checkpoint-900/global_step900/zero_pp_rank_0_mp_rank_00_model_states.pt +0 -3
  8. checkpoints/checkpoint-900/global_step900/zero_pp_rank_1_mp_rank_00_model_states.pt +0 -3
  9. checkpoints/checkpoint-900/global_step900/zero_pp_rank_2_mp_rank_00_model_states.pt +0 -3
  10. checkpoints/checkpoint-900/global_step900/zero_pp_rank_3_mp_rank_00_model_states.pt +0 -3
  11. checkpoints/checkpoint-900/latest +0 -1
  12. checkpoints/checkpoint-900/model.safetensors +0 -3
  13. checkpoints/checkpoint-900/rng_state_0.pth +0 -3
  14. checkpoints/checkpoint-900/rng_state_1.pth +0 -3
  15. checkpoints/checkpoint-900/rng_state_2.pth +0 -3
  16. checkpoints/checkpoint-900/rng_state_3.pth +0 -3
  17. checkpoints/checkpoint-900/scheduler.pt +0 -3
  18. checkpoints/checkpoint-900/special_tokens_map.json +0 -17
  19. checkpoints/checkpoint-900/tokenizer.json +0 -3
  20. checkpoints/checkpoint-900/tokenizer_config.json +0 -2064
  21. checkpoints/checkpoint-900/trainer_state.json +0 -1365
  22. checkpoints/checkpoint-900/training_args.bin +0 -3
  23. checkpoints/checkpoint-900/zero_to_fp32.py +0 -674
checkpoints/checkpoint-900/config.json DELETED
@@ -1,40 +0,0 @@
1
- {
2
- "_name_or_path": "/root/open-r1/checkpoints/meta-llama/Llama-3.2-1B-Instruct/",
3
- "architectures": [
4
- "LlamaForCausalLM"
5
- ],
6
- "attention_bias": false,
7
- "attention_dropout": 0.0,
8
- "bos_token_id": 128000,
9
- "eos_token_id": [
10
- 128001,
11
- 128008,
12
- 128009
13
- ],
14
- "head_dim": 64,
15
- "hidden_act": "silu",
16
- "hidden_size": 2048,
17
- "initializer_range": 0.02,
18
- "intermediate_size": 8192,
19
- "max_position_embeddings": 131072,
20
- "mlp_bias": false,
21
- "model_type": "llama",
22
- "num_attention_heads": 32,
23
- "num_hidden_layers": 16,
24
- "num_key_value_heads": 8,
25
- "pretraining_tp": 1,
26
- "rms_norm_eps": 1e-05,
27
- "rope_scaling": {
28
- "factor": 32.0,
29
- "high_freq_factor": 4.0,
30
- "low_freq_factor": 1.0,
31
- "original_max_position_embeddings": 8192,
32
- "rope_type": "llama3"
33
- },
34
- "rope_theta": 500000.0,
35
- "tie_word_embeddings": true,
36
- "torch_dtype": "bfloat16",
37
- "transformers_version": "4.49.0.dev0",
38
- "use_cache": false,
39
- "vocab_size": 128256
40
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
checkpoints/checkpoint-900/generation_config.json DELETED
@@ -1,12 +0,0 @@
1
- {
2
- "bos_token_id": 128000,
3
- "do_sample": true,
4
- "eos_token_id": [
5
- 128001,
6
- 128008,
7
- 128009
8
- ],
9
- "temperature": 0.6,
10
- "top_p": 0.9,
11
- "transformers_version": "4.49.0.dev0"
12
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
checkpoints/checkpoint-900/global_step900/bf16_zero_pp_rank_0_mp_rank_00_optim_states.pt DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:7d6bf6cee495523ac6fc19c1ae85cc66c0402e494e79e8273ee9223bacd1f0c2
3
- size 3707447984
 
 
 
 
checkpoints/checkpoint-900/global_step900/bf16_zero_pp_rank_1_mp_rank_00_optim_states.pt DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:12ff302b03e315bc3dadefc12b331b9d8f441c2d434dcd7ace6988b3a4899bbf
3
- size 3707447984
 
 
 
 
checkpoints/checkpoint-900/global_step900/bf16_zero_pp_rank_2_mp_rank_00_optim_states.pt DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:e89cb27bfc2b714613219126b07eee6a1dd5339104061549056618ffcfbc31d3
3
- size 3707447984
 
 
 
 
checkpoints/checkpoint-900/global_step900/bf16_zero_pp_rank_3_mp_rank_00_optim_states.pt DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:65b65a4e29c8aefc1a051b7ce234d0897d40280fc01c4e2bbcc91e86037881dc
3
- size 3707447984
 
 
 
 
checkpoints/checkpoint-900/global_step900/zero_pp_rank_0_mp_rank_00_model_states.pt DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:f4f0586d4eeef8444a5875b013789df4929aca9fc3d6e79715825bb5eff91ab2
3
- size 75384
 
 
 
 
checkpoints/checkpoint-900/global_step900/zero_pp_rank_1_mp_rank_00_model_states.pt DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:602896dfa0ba233ca45d33ebba6047a7b959d7bed67329f29ca4b1395486af8d
3
- size 75320
 
 
 
 
checkpoints/checkpoint-900/global_step900/zero_pp_rank_2_mp_rank_00_model_states.pt DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:c4e64bfaae681ce6019b60875fd5f3513a6ce2d5d3beb8c4dad8ff3c68f3905d
3
- size 75320
 
 
 
 
checkpoints/checkpoint-900/global_step900/zero_pp_rank_3_mp_rank_00_model_states.pt DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:a9dcb5835d47b7453eb34258d7b6589f5b17eb159d2e2d3a96193cd34077e161
3
- size 75320
 
 
 
 
checkpoints/checkpoint-900/latest DELETED
@@ -1 +0,0 @@
1
- global_step900
 
 
checkpoints/checkpoint-900/model.safetensors DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:084a75537bb0b59dce7c866406a247d7976305d52645a56edc44b8e8f755c30f
3
- size 2471645608
 
 
 
 
checkpoints/checkpoint-900/rng_state_0.pth DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:ae78313eb528c8d3695eebaf4de3539bd0a0bc6ee18c66af1ee183442f1758a0
3
- size 15024
 
 
 
 
checkpoints/checkpoint-900/rng_state_1.pth DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:1b38031f60d9e88601d369ef46bcdcf2b5b03f2cb4ba93853bcb2328df7ebb7c
3
- size 15024
 
 
 
 
checkpoints/checkpoint-900/rng_state_2.pth DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:f58092375c93d237cd0e3149aecfbf83e2acdae46279e07a32920d01cb507e64
3
- size 15024
 
 
 
 
checkpoints/checkpoint-900/rng_state_3.pth DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:83cd4bbff9962da7ec6787fcea8d65df7096917f9a5902e249ba7aee8887fe5f
3
- size 15024
 
 
 
 
checkpoints/checkpoint-900/scheduler.pt DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:03746929464b54010bdb23b85c112256bc4e1d2793521314f052727b3ea9e68f
3
- size 1064
 
 
 
 
checkpoints/checkpoint-900/special_tokens_map.json DELETED
@@ -1,17 +0,0 @@
1
- {
2
- "bos_token": {
3
- "content": "<|begin_of_text|>",
4
- "lstrip": false,
5
- "normalized": false,
6
- "rstrip": false,
7
- "single_word": false
8
- },
9
- "eos_token": {
10
- "content": "<|eot_id|>",
11
- "lstrip": false,
12
- "normalized": false,
13
- "rstrip": false,
14
- "single_word": false
15
- },
16
- "pad_token": "<|eot_id|>"
17
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
checkpoints/checkpoint-900/tokenizer.json DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:6b9e4e7fb171f92fd137b777cc2714bf87d11576700a1dcd7a399e7bbe39537b
3
- size 17209920
 
 
 
 
checkpoints/checkpoint-900/tokenizer_config.json DELETED
@@ -1,2064 +0,0 @@
1
- {
2
- "added_tokens_decoder": {
3
- "128000": {
4
- "content": "<|begin_of_text|>",
5
- "lstrip": false,
6
- "normalized": false,
7
- "rstrip": false,
8
- "single_word": false,
9
- "special": true
10
- },
11
- "128001": {
12
- "content": "<|end_of_text|>",
13
- "lstrip": false,
14
- "normalized": false,
15
- "rstrip": false,
16
- "single_word": false,
17
- "special": true
18
- },
19
- "128002": {
20
- "content": "<|reserved_special_token_0|>",
21
- "lstrip": false,
22
- "normalized": false,
23
- "rstrip": false,
24
- "single_word": false,
25
- "special": true
26
- },
27
- "128003": {
28
- "content": "<|reserved_special_token_1|>",
29
- "lstrip": false,
30
- "normalized": false,
31
- "rstrip": false,
32
- "single_word": false,
33
- "special": true
34
- },
35
- "128004": {
36
- "content": "<|finetune_right_pad_id|>",
37
- "lstrip": false,
38
- "normalized": false,
39
- "rstrip": false,
40
- "single_word": false,
41
- "special": true
42
- },
43
- "128005": {
44
- "content": "<|reserved_special_token_2|>",
45
- "lstrip": false,
46
- "normalized": false,
47
- "rstrip": false,
48
- "single_word": false,
49
- "special": true
50
- },
51
- "128006": {
52
- "content": "<|start_header_id|>",
53
- "lstrip": false,
54
- "normalized": false,
55
- "rstrip": false,
56
- "single_word": false,
57
- "special": true
58
- },
59
- "128007": {
60
- "content": "<|end_header_id|>",
61
- "lstrip": false,
62
- "normalized": false,
63
- "rstrip": false,
64
- "single_word": false,
65
- "special": true
66
- },
67
- "128008": {
68
- "content": "<|eom_id|>",
69
- "lstrip": false,
70
- "normalized": false,
71
- "rstrip": false,
72
- "single_word": false,
73
- "special": true
74
- },
75
- "128009": {
76
- "content": "<|eot_id|>",
77
- "lstrip": false,
78
- "normalized": false,
79
- "rstrip": false,
80
- "single_word": false,
81
- "special": true
82
- },
83
- "128010": {
84
- "content": "<|python_tag|>",
85
- "lstrip": false,
86
- "normalized": false,
87
- "rstrip": false,
88
- "single_word": false,
89
- "special": true
90
- },
91
- "128011": {
92
- "content": "<|reserved_special_token_3|>",
93
- "lstrip": false,
94
- "normalized": false,
95
- "rstrip": false,
96
- "single_word": false,
97
- "special": true
98
- },
99
- "128012": {
100
- "content": "<|reserved_special_token_4|>",
101
- "lstrip": false,
102
- "normalized": false,
103
- "rstrip": false,
104
- "single_word": false,
105
- "special": true
106
- },
107
- "128013": {
108
- "content": "<|reserved_special_token_5|>",
109
- "lstrip": false,
110
- "normalized": false,
111
- "rstrip": false,
112
- "single_word": false,
113
- "special": true
114
- },
115
- "128014": {
116
- "content": "<|reserved_special_token_6|>",
117
- "lstrip": false,
118
- "normalized": false,
119
- "rstrip": false,
120
- "single_word": false,
121
- "special": true
122
- },
123
- "128015": {
124
- "content": "<|reserved_special_token_7|>",
125
- "lstrip": false,
126
- "normalized": false,
127
- "rstrip": false,
128
- "single_word": false,
129
- "special": true
130
- },
131
- "128016": {
132
- "content": "<|reserved_special_token_8|>",
133
- "lstrip": false,
134
- "normalized": false,
135
- "rstrip": false,
136
- "single_word": false,
137
- "special": true
138
- },
139
- "128017": {
140
- "content": "<|reserved_special_token_9|>",
141
- "lstrip": false,
142
- "normalized": false,
143
- "rstrip": false,
144
- "single_word": false,
145
- "special": true
146
- },
147
- "128018": {
148
- "content": "<|reserved_special_token_10|>",
149
- "lstrip": false,
150
- "normalized": false,
151
- "rstrip": false,
152
- "single_word": false,
153
- "special": true
154
- },
155
- "128019": {
156
- "content": "<|reserved_special_token_11|>",
157
- "lstrip": false,
158
- "normalized": false,
159
- "rstrip": false,
160
- "single_word": false,
161
- "special": true
162
- },
163
- "128020": {
164
- "content": "<|reserved_special_token_12|>",
165
- "lstrip": false,
166
- "normalized": false,
167
- "rstrip": false,
168
- "single_word": false,
169
- "special": true
170
- },
171
- "128021": {
172
- "content": "<|reserved_special_token_13|>",
173
- "lstrip": false,
174
- "normalized": false,
175
- "rstrip": false,
176
- "single_word": false,
177
- "special": true
178
- },
179
- "128022": {
180
- "content": "<|reserved_special_token_14|>",
181
- "lstrip": false,
182
- "normalized": false,
183
- "rstrip": false,
184
- "single_word": false,
185
- "special": true
186
- },
187
- "128023": {
188
- "content": "<|reserved_special_token_15|>",
189
- "lstrip": false,
190
- "normalized": false,
191
- "rstrip": false,
192
- "single_word": false,
193
- "special": true
194
- },
195
- "128024": {
196
- "content": "<|reserved_special_token_16|>",
197
- "lstrip": false,
198
- "normalized": false,
199
- "rstrip": false,
200
- "single_word": false,
201
- "special": true
202
- },
203
- "128025": {
204
- "content": "<|reserved_special_token_17|>",
205
- "lstrip": false,
206
- "normalized": false,
207
- "rstrip": false,
208
- "single_word": false,
209
- "special": true
210
- },
211
- "128026": {
212
- "content": "<|reserved_special_token_18|>",
213
- "lstrip": false,
214
- "normalized": false,
215
- "rstrip": false,
216
- "single_word": false,
217
- "special": true
218
- },
219
- "128027": {
220
- "content": "<|reserved_special_token_19|>",
221
- "lstrip": false,
222
- "normalized": false,
223
- "rstrip": false,
224
- "single_word": false,
225
- "special": true
226
- },
227
- "128028": {
228
- "content": "<|reserved_special_token_20|>",
229
- "lstrip": false,
230
- "normalized": false,
231
- "rstrip": false,
232
- "single_word": false,
233
- "special": true
234
- },
235
- "128029": {
236
- "content": "<|reserved_special_token_21|>",
237
- "lstrip": false,
238
- "normalized": false,
239
- "rstrip": false,
240
- "single_word": false,
241
- "special": true
242
- },
243
- "128030": {
244
- "content": "<|reserved_special_token_22|>",
245
- "lstrip": false,
246
- "normalized": false,
247
- "rstrip": false,
248
- "single_word": false,
249
- "special": true
250
- },
251
- "128031": {
252
- "content": "<|reserved_special_token_23|>",
253
- "lstrip": false,
254
- "normalized": false,
255
- "rstrip": false,
256
- "single_word": false,
257
- "special": true
258
- },
259
- "128032": {
260
- "content": "<|reserved_special_token_24|>",
261
- "lstrip": false,
262
- "normalized": false,
263
- "rstrip": false,
264
- "single_word": false,
265
- "special": true
266
- },
267
- "128033": {
268
- "content": "<|reserved_special_token_25|>",
269
- "lstrip": false,
270
- "normalized": false,
271
- "rstrip": false,
272
- "single_word": false,
273
- "special": true
274
- },
275
- "128034": {
276
- "content": "<|reserved_special_token_26|>",
277
- "lstrip": false,
278
- "normalized": false,
279
- "rstrip": false,
280
- "single_word": false,
281
- "special": true
282
- },
283
- "128035": {
284
- "content": "<|reserved_special_token_27|>",
285
- "lstrip": false,
286
- "normalized": false,
287
- "rstrip": false,
288
- "single_word": false,
289
- "special": true
290
- },
291
- "128036": {
292
- "content": "<|reserved_special_token_28|>",
293
- "lstrip": false,
294
- "normalized": false,
295
- "rstrip": false,
296
- "single_word": false,
297
- "special": true
298
- },
299
- "128037": {
300
- "content": "<|reserved_special_token_29|>",
301
- "lstrip": false,
302
- "normalized": false,
303
- "rstrip": false,
304
- "single_word": false,
305
- "special": true
306
- },
307
- "128038": {
308
- "content": "<|reserved_special_token_30|>",
309
- "lstrip": false,
310
- "normalized": false,
311
- "rstrip": false,
312
- "single_word": false,
313
- "special": true
314
- },
315
- "128039": {
316
- "content": "<|reserved_special_token_31|>",
317
- "lstrip": false,
318
- "normalized": false,
319
- "rstrip": false,
320
- "single_word": false,
321
- "special": true
322
- },
323
- "128040": {
324
- "content": "<|reserved_special_token_32|>",
325
- "lstrip": false,
326
- "normalized": false,
327
- "rstrip": false,
328
- "single_word": false,
329
- "special": true
330
- },
331
- "128041": {
332
- "content": "<|reserved_special_token_33|>",
333
- "lstrip": false,
334
- "normalized": false,
335
- "rstrip": false,
336
- "single_word": false,
337
- "special": true
338
- },
339
- "128042": {
340
- "content": "<|reserved_special_token_34|>",
341
- "lstrip": false,
342
- "normalized": false,
343
- "rstrip": false,
344
- "single_word": false,
345
- "special": true
346
- },
347
- "128043": {
348
- "content": "<|reserved_special_token_35|>",
349
- "lstrip": false,
350
- "normalized": false,
351
- "rstrip": false,
352
- "single_word": false,
353
- "special": true
354
- },
355
- "128044": {
356
- "content": "<|reserved_special_token_36|>",
357
- "lstrip": false,
358
- "normalized": false,
359
- "rstrip": false,
360
- "single_word": false,
361
- "special": true
362
- },
363
- "128045": {
364
- "content": "<|reserved_special_token_37|>",
365
- "lstrip": false,
366
- "normalized": false,
367
- "rstrip": false,
368
- "single_word": false,
369
- "special": true
370
- },
371
- "128046": {
372
- "content": "<|reserved_special_token_38|>",
373
- "lstrip": false,
374
- "normalized": false,
375
- "rstrip": false,
376
- "single_word": false,
377
- "special": true
378
- },
379
- "128047": {
380
- "content": "<|reserved_special_token_39|>",
381
- "lstrip": false,
382
- "normalized": false,
383
- "rstrip": false,
384
- "single_word": false,
385
- "special": true
386
- },
387
- "128048": {
388
- "content": "<|reserved_special_token_40|>",
389
- "lstrip": false,
390
- "normalized": false,
391
- "rstrip": false,
392
- "single_word": false,
393
- "special": true
394
- },
395
- "128049": {
396
- "content": "<|reserved_special_token_41|>",
397
- "lstrip": false,
398
- "normalized": false,
399
- "rstrip": false,
400
- "single_word": false,
401
- "special": true
402
- },
403
- "128050": {
404
- "content": "<|reserved_special_token_42|>",
405
- "lstrip": false,
406
- "normalized": false,
407
- "rstrip": false,
408
- "single_word": false,
409
- "special": true
410
- },
411
- "128051": {
412
- "content": "<|reserved_special_token_43|>",
413
- "lstrip": false,
414
- "normalized": false,
415
- "rstrip": false,
416
- "single_word": false,
417
- "special": true
418
- },
419
- "128052": {
420
- "content": "<|reserved_special_token_44|>",
421
- "lstrip": false,
422
- "normalized": false,
423
- "rstrip": false,
424
- "single_word": false,
425
- "special": true
426
- },
427
- "128053": {
428
- "content": "<|reserved_special_token_45|>",
429
- "lstrip": false,
430
- "normalized": false,
431
- "rstrip": false,
432
- "single_word": false,
433
- "special": true
434
- },
435
- "128054": {
436
- "content": "<|reserved_special_token_46|>",
437
- "lstrip": false,
438
- "normalized": false,
439
- "rstrip": false,
440
- "single_word": false,
441
- "special": true
442
- },
443
- "128055": {
444
- "content": "<|reserved_special_token_47|>",
445
- "lstrip": false,
446
- "normalized": false,
447
- "rstrip": false,
448
- "single_word": false,
449
- "special": true
450
- },
451
- "128056": {
452
- "content": "<|reserved_special_token_48|>",
453
- "lstrip": false,
454
- "normalized": false,
455
- "rstrip": false,
456
- "single_word": false,
457
- "special": true
458
- },
459
- "128057": {
460
- "content": "<|reserved_special_token_49|>",
461
- "lstrip": false,
462
- "normalized": false,
463
- "rstrip": false,
464
- "single_word": false,
465
- "special": true
466
- },
467
- "128058": {
468
- "content": "<|reserved_special_token_50|>",
469
- "lstrip": false,
470
- "normalized": false,
471
- "rstrip": false,
472
- "single_word": false,
473
- "special": true
474
- },
475
- "128059": {
476
- "content": "<|reserved_special_token_51|>",
477
- "lstrip": false,
478
- "normalized": false,
479
- "rstrip": false,
480
- "single_word": false,
481
- "special": true
482
- },
483
- "128060": {
484
- "content": "<|reserved_special_token_52|>",
485
- "lstrip": false,
486
- "normalized": false,
487
- "rstrip": false,
488
- "single_word": false,
489
- "special": true
490
- },
491
- "128061": {
492
- "content": "<|reserved_special_token_53|>",
493
- "lstrip": false,
494
- "normalized": false,
495
- "rstrip": false,
496
- "single_word": false,
497
- "special": true
498
- },
499
- "128062": {
500
- "content": "<|reserved_special_token_54|>",
501
- "lstrip": false,
502
- "normalized": false,
503
- "rstrip": false,
504
- "single_word": false,
505
- "special": true
506
- },
507
- "128063": {
508
- "content": "<|reserved_special_token_55|>",
509
- "lstrip": false,
510
- "normalized": false,
511
- "rstrip": false,
512
- "single_word": false,
513
- "special": true
514
- },
515
- "128064": {
516
- "content": "<|reserved_special_token_56|>",
517
- "lstrip": false,
518
- "normalized": false,
519
- "rstrip": false,
520
- "single_word": false,
521
- "special": true
522
- },
523
- "128065": {
524
- "content": "<|reserved_special_token_57|>",
525
- "lstrip": false,
526
- "normalized": false,
527
- "rstrip": false,
528
- "single_word": false,
529
- "special": true
530
- },
531
- "128066": {
532
- "content": "<|reserved_special_token_58|>",
533
- "lstrip": false,
534
- "normalized": false,
535
- "rstrip": false,
536
- "single_word": false,
537
- "special": true
538
- },
539
- "128067": {
540
- "content": "<|reserved_special_token_59|>",
541
- "lstrip": false,
542
- "normalized": false,
543
- "rstrip": false,
544
- "single_word": false,
545
- "special": true
546
- },
547
- "128068": {
548
- "content": "<|reserved_special_token_60|>",
549
- "lstrip": false,
550
- "normalized": false,
551
- "rstrip": false,
552
- "single_word": false,
553
- "special": true
554
- },
555
- "128069": {
556
- "content": "<|reserved_special_token_61|>",
557
- "lstrip": false,
558
- "normalized": false,
559
- "rstrip": false,
560
- "single_word": false,
561
- "special": true
562
- },
563
- "128070": {
564
- "content": "<|reserved_special_token_62|>",
565
- "lstrip": false,
566
- "normalized": false,
567
- "rstrip": false,
568
- "single_word": false,
569
- "special": true
570
- },
571
- "128071": {
572
- "content": "<|reserved_special_token_63|>",
573
- "lstrip": false,
574
- "normalized": false,
575
- "rstrip": false,
576
- "single_word": false,
577
- "special": true
578
- },
579
- "128072": {
580
- "content": "<|reserved_special_token_64|>",
581
- "lstrip": false,
582
- "normalized": false,
583
- "rstrip": false,
584
- "single_word": false,
585
- "special": true
586
- },
587
- "128073": {
588
- "content": "<|reserved_special_token_65|>",
589
- "lstrip": false,
590
- "normalized": false,
591
- "rstrip": false,
592
- "single_word": false,
593
- "special": true
594
- },
595
- "128074": {
596
- "content": "<|reserved_special_token_66|>",
597
- "lstrip": false,
598
- "normalized": false,
599
- "rstrip": false,
600
- "single_word": false,
601
- "special": true
602
- },
603
- "128075": {
604
- "content": "<|reserved_special_token_67|>",
605
- "lstrip": false,
606
- "normalized": false,
607
- "rstrip": false,
608
- "single_word": false,
609
- "special": true
610
- },
611
- "128076": {
612
- "content": "<|reserved_special_token_68|>",
613
- "lstrip": false,
614
- "normalized": false,
615
- "rstrip": false,
616
- "single_word": false,
617
- "special": true
618
- },
619
- "128077": {
620
- "content": "<|reserved_special_token_69|>",
621
- "lstrip": false,
622
- "normalized": false,
623
- "rstrip": false,
624
- "single_word": false,
625
- "special": true
626
- },
627
- "128078": {
628
- "content": "<|reserved_special_token_70|>",
629
- "lstrip": false,
630
- "normalized": false,
631
- "rstrip": false,
632
- "single_word": false,
633
- "special": true
634
- },
635
- "128079": {
636
- "content": "<|reserved_special_token_71|>",
637
- "lstrip": false,
638
- "normalized": false,
639
- "rstrip": false,
640
- "single_word": false,
641
- "special": true
642
- },
643
- "128080": {
644
- "content": "<|reserved_special_token_72|>",
645
- "lstrip": false,
646
- "normalized": false,
647
- "rstrip": false,
648
- "single_word": false,
649
- "special": true
650
- },
651
- "128081": {
652
- "content": "<|reserved_special_token_73|>",
653
- "lstrip": false,
654
- "normalized": false,
655
- "rstrip": false,
656
- "single_word": false,
657
- "special": true
658
- },
659
- "128082": {
660
- "content": "<|reserved_special_token_74|>",
661
- "lstrip": false,
662
- "normalized": false,
663
- "rstrip": false,
664
- "single_word": false,
665
- "special": true
666
- },
667
- "128083": {
668
- "content": "<|reserved_special_token_75|>",
669
- "lstrip": false,
670
- "normalized": false,
671
- "rstrip": false,
672
- "single_word": false,
673
- "special": true
674
- },
675
- "128084": {
676
- "content": "<|reserved_special_token_76|>",
677
- "lstrip": false,
678
- "normalized": false,
679
- "rstrip": false,
680
- "single_word": false,
681
- "special": true
682
- },
683
- "128085": {
684
- "content": "<|reserved_special_token_77|>",
685
- "lstrip": false,
686
- "normalized": false,
687
- "rstrip": false,
688
- "single_word": false,
689
- "special": true
690
- },
691
- "128086": {
692
- "content": "<|reserved_special_token_78|>",
693
- "lstrip": false,
694
- "normalized": false,
695
- "rstrip": false,
696
- "single_word": false,
697
- "special": true
698
- },
699
- "128087": {
700
- "content": "<|reserved_special_token_79|>",
701
- "lstrip": false,
702
- "normalized": false,
703
- "rstrip": false,
704
- "single_word": false,
705
- "special": true
706
- },
707
- "128088": {
708
- "content": "<|reserved_special_token_80|>",
709
- "lstrip": false,
710
- "normalized": false,
711
- "rstrip": false,
712
- "single_word": false,
713
- "special": true
714
- },
715
- "128089": {
716
- "content": "<|reserved_special_token_81|>",
717
- "lstrip": false,
718
- "normalized": false,
719
- "rstrip": false,
720
- "single_word": false,
721
- "special": true
722
- },
723
- "128090": {
724
- "content": "<|reserved_special_token_82|>",
725
- "lstrip": false,
726
- "normalized": false,
727
- "rstrip": false,
728
- "single_word": false,
729
- "special": true
730
- },
731
- "128091": {
732
- "content": "<|reserved_special_token_83|>",
733
- "lstrip": false,
734
- "normalized": false,
735
- "rstrip": false,
736
- "single_word": false,
737
- "special": true
738
- },
739
- "128092": {
740
- "content": "<|reserved_special_token_84|>",
741
- "lstrip": false,
742
- "normalized": false,
743
- "rstrip": false,
744
- "single_word": false,
745
- "special": true
746
- },
747
- "128093": {
748
- "content": "<|reserved_special_token_85|>",
749
- "lstrip": false,
750
- "normalized": false,
751
- "rstrip": false,
752
- "single_word": false,
753
- "special": true
754
- },
755
- "128094": {
756
- "content": "<|reserved_special_token_86|>",
757
- "lstrip": false,
758
- "normalized": false,
759
- "rstrip": false,
760
- "single_word": false,
761
- "special": true
762
- },
763
- "128095": {
764
- "content": "<|reserved_special_token_87|>",
765
- "lstrip": false,
766
- "normalized": false,
767
- "rstrip": false,
768
- "single_word": false,
769
- "special": true
770
- },
771
- "128096": {
772
- "content": "<|reserved_special_token_88|>",
773
- "lstrip": false,
774
- "normalized": false,
775
- "rstrip": false,
776
- "single_word": false,
777
- "special": true
778
- },
779
- "128097": {
780
- "content": "<|reserved_special_token_89|>",
781
- "lstrip": false,
782
- "normalized": false,
783
- "rstrip": false,
784
- "single_word": false,
785
- "special": true
786
- },
787
- "128098": {
788
- "content": "<|reserved_special_token_90|>",
789
- "lstrip": false,
790
- "normalized": false,
791
- "rstrip": false,
792
- "single_word": false,
793
- "special": true
794
- },
795
- "128099": {
796
- "content": "<|reserved_special_token_91|>",
797
- "lstrip": false,
798
- "normalized": false,
799
- "rstrip": false,
800
- "single_word": false,
801
- "special": true
802
- },
803
- "128100": {
804
- "content": "<|reserved_special_token_92|>",
805
- "lstrip": false,
806
- "normalized": false,
807
- "rstrip": false,
808
- "single_word": false,
809
- "special": true
810
- },
811
- "128101": {
812
- "content": "<|reserved_special_token_93|>",
813
- "lstrip": false,
814
- "normalized": false,
815
- "rstrip": false,
816
- "single_word": false,
817
- "special": true
818
- },
819
- "128102": {
820
- "content": "<|reserved_special_token_94|>",
821
- "lstrip": false,
822
- "normalized": false,
823
- "rstrip": false,
824
- "single_word": false,
825
- "special": true
826
- },
827
- "128103": {
828
- "content": "<|reserved_special_token_95|>",
829
- "lstrip": false,
830
- "normalized": false,
831
- "rstrip": false,
832
- "single_word": false,
833
- "special": true
834
- },
835
- "128104": {
836
- "content": "<|reserved_special_token_96|>",
837
- "lstrip": false,
838
- "normalized": false,
839
- "rstrip": false,
840
- "single_word": false,
841
- "special": true
842
- },
843
- "128105": {
844
- "content": "<|reserved_special_token_97|>",
845
- "lstrip": false,
846
- "normalized": false,
847
- "rstrip": false,
848
- "single_word": false,
849
- "special": true
850
- },
851
- "128106": {
852
- "content": "<|reserved_special_token_98|>",
853
- "lstrip": false,
854
- "normalized": false,
855
- "rstrip": false,
856
- "single_word": false,
857
- "special": true
858
- },
859
- "128107": {
860
- "content": "<|reserved_special_token_99|>",
861
- "lstrip": false,
862
- "normalized": false,
863
- "rstrip": false,
864
- "single_word": false,
865
- "special": true
866
- },
867
- "128108": {
868
- "content": "<|reserved_special_token_100|>",
869
- "lstrip": false,
870
- "normalized": false,
871
- "rstrip": false,
872
- "single_word": false,
873
- "special": true
874
- },
875
- "128109": {
876
- "content": "<|reserved_special_token_101|>",
877
- "lstrip": false,
878
- "normalized": false,
879
- "rstrip": false,
880
- "single_word": false,
881
- "special": true
882
- },
883
- "128110": {
884
- "content": "<|reserved_special_token_102|>",
885
- "lstrip": false,
886
- "normalized": false,
887
- "rstrip": false,
888
- "single_word": false,
889
- "special": true
890
- },
891
- "128111": {
892
- "content": "<|reserved_special_token_103|>",
893
- "lstrip": false,
894
- "normalized": false,
895
- "rstrip": false,
896
- "single_word": false,
897
- "special": true
898
- },
899
- "128112": {
900
- "content": "<|reserved_special_token_104|>",
901
- "lstrip": false,
902
- "normalized": false,
903
- "rstrip": false,
904
- "single_word": false,
905
- "special": true
906
- },
907
- "128113": {
908
- "content": "<|reserved_special_token_105|>",
909
- "lstrip": false,
910
- "normalized": false,
911
- "rstrip": false,
912
- "single_word": false,
913
- "special": true
914
- },
915
- "128114": {
916
- "content": "<|reserved_special_token_106|>",
917
- "lstrip": false,
918
- "normalized": false,
919
- "rstrip": false,
920
- "single_word": false,
921
- "special": true
922
- },
923
- "128115": {
924
- "content": "<|reserved_special_token_107|>",
925
- "lstrip": false,
926
- "normalized": false,
927
- "rstrip": false,
928
- "single_word": false,
929
- "special": true
930
- },
931
- "128116": {
932
- "content": "<|reserved_special_token_108|>",
933
- "lstrip": false,
934
- "normalized": false,
935
- "rstrip": false,
936
- "single_word": false,
937
- "special": true
938
- },
939
- "128117": {
940
- "content": "<|reserved_special_token_109|>",
941
- "lstrip": false,
942
- "normalized": false,
943
- "rstrip": false,
944
- "single_word": false,
945
- "special": true
946
- },
947
- "128118": {
948
- "content": "<|reserved_special_token_110|>",
949
- "lstrip": false,
950
- "normalized": false,
951
- "rstrip": false,
952
- "single_word": false,
953
- "special": true
954
- },
955
- "128119": {
956
- "content": "<|reserved_special_token_111|>",
957
- "lstrip": false,
958
- "normalized": false,
959
- "rstrip": false,
960
- "single_word": false,
961
- "special": true
962
- },
963
- "128120": {
964
- "content": "<|reserved_special_token_112|>",
965
- "lstrip": false,
966
- "normalized": false,
967
- "rstrip": false,
968
- "single_word": false,
969
- "special": true
970
- },
971
- "128121": {
972
- "content": "<|reserved_special_token_113|>",
973
- "lstrip": false,
974
- "normalized": false,
975
- "rstrip": false,
976
- "single_word": false,
977
- "special": true
978
- },
979
- "128122": {
980
- "content": "<|reserved_special_token_114|>",
981
- "lstrip": false,
982
- "normalized": false,
983
- "rstrip": false,
984
- "single_word": false,
985
- "special": true
986
- },
987
- "128123": {
988
- "content": "<|reserved_special_token_115|>",
989
- "lstrip": false,
990
- "normalized": false,
991
- "rstrip": false,
992
- "single_word": false,
993
- "special": true
994
- },
995
- "128124": {
996
- "content": "<|reserved_special_token_116|>",
997
- "lstrip": false,
998
- "normalized": false,
999
- "rstrip": false,
1000
- "single_word": false,
1001
- "special": true
1002
- },
1003
- "128125": {
1004
- "content": "<|reserved_special_token_117|>",
1005
- "lstrip": false,
1006
- "normalized": false,
1007
- "rstrip": false,
1008
- "single_word": false,
1009
- "special": true
1010
- },
1011
- "128126": {
1012
- "content": "<|reserved_special_token_118|>",
1013
- "lstrip": false,
1014
- "normalized": false,
1015
- "rstrip": false,
1016
- "single_word": false,
1017
- "special": true
1018
- },
1019
- "128127": {
1020
- "content": "<|reserved_special_token_119|>",
1021
- "lstrip": false,
1022
- "normalized": false,
1023
- "rstrip": false,
1024
- "single_word": false,
1025
- "special": true
1026
- },
1027
- "128128": {
1028
- "content": "<|reserved_special_token_120|>",
1029
- "lstrip": false,
1030
- "normalized": false,
1031
- "rstrip": false,
1032
- "single_word": false,
1033
- "special": true
1034
- },
1035
- "128129": {
1036
- "content": "<|reserved_special_token_121|>",
1037
- "lstrip": false,
1038
- "normalized": false,
1039
- "rstrip": false,
1040
- "single_word": false,
1041
- "special": true
1042
- },
1043
- "128130": {
1044
- "content": "<|reserved_special_token_122|>",
1045
- "lstrip": false,
1046
- "normalized": false,
1047
- "rstrip": false,
1048
- "single_word": false,
1049
- "special": true
1050
- },
1051
- "128131": {
1052
- "content": "<|reserved_special_token_123|>",
1053
- "lstrip": false,
1054
- "normalized": false,
1055
- "rstrip": false,
1056
- "single_word": false,
1057
- "special": true
1058
- },
1059
- "128132": {
1060
- "content": "<|reserved_special_token_124|>",
1061
- "lstrip": false,
1062
- "normalized": false,
1063
- "rstrip": false,
1064
- "single_word": false,
1065
- "special": true
1066
- },
1067
- "128133": {
1068
- "content": "<|reserved_special_token_125|>",
1069
- "lstrip": false,
1070
- "normalized": false,
1071
- "rstrip": false,
1072
- "single_word": false,
1073
- "special": true
1074
- },
1075
- "128134": {
1076
- "content": "<|reserved_special_token_126|>",
1077
- "lstrip": false,
1078
- "normalized": false,
1079
- "rstrip": false,
1080
- "single_word": false,
1081
- "special": true
1082
- },
1083
- "128135": {
1084
- "content": "<|reserved_special_token_127|>",
1085
- "lstrip": false,
1086
- "normalized": false,
1087
- "rstrip": false,
1088
- "single_word": false,
1089
- "special": true
1090
- },
1091
- "128136": {
1092
- "content": "<|reserved_special_token_128|>",
1093
- "lstrip": false,
1094
- "normalized": false,
1095
- "rstrip": false,
1096
- "single_word": false,
1097
- "special": true
1098
- },
1099
- "128137": {
1100
- "content": "<|reserved_special_token_129|>",
1101
- "lstrip": false,
1102
- "normalized": false,
1103
- "rstrip": false,
1104
- "single_word": false,
1105
- "special": true
1106
- },
1107
- "128138": {
1108
- "content": "<|reserved_special_token_130|>",
1109
- "lstrip": false,
1110
- "normalized": false,
1111
- "rstrip": false,
1112
- "single_word": false,
1113
- "special": true
1114
- },
1115
- "128139": {
1116
- "content": "<|reserved_special_token_131|>",
1117
- "lstrip": false,
1118
- "normalized": false,
1119
- "rstrip": false,
1120
- "single_word": false,
1121
- "special": true
1122
- },
1123
- "128140": {
1124
- "content": "<|reserved_special_token_132|>",
1125
- "lstrip": false,
1126
- "normalized": false,
1127
- "rstrip": false,
1128
- "single_word": false,
1129
- "special": true
1130
- },
1131
- "128141": {
1132
- "content": "<|reserved_special_token_133|>",
1133
- "lstrip": false,
1134
- "normalized": false,
1135
- "rstrip": false,
1136
- "single_word": false,
1137
- "special": true
1138
- },
1139
- "128142": {
1140
- "content": "<|reserved_special_token_134|>",
1141
- "lstrip": false,
1142
- "normalized": false,
1143
- "rstrip": false,
1144
- "single_word": false,
1145
- "special": true
1146
- },
1147
- "128143": {
1148
- "content": "<|reserved_special_token_135|>",
1149
- "lstrip": false,
1150
- "normalized": false,
1151
- "rstrip": false,
1152
- "single_word": false,
1153
- "special": true
1154
- },
1155
- "128144": {
1156
- "content": "<|reserved_special_token_136|>",
1157
- "lstrip": false,
1158
- "normalized": false,
1159
- "rstrip": false,
1160
- "single_word": false,
1161
- "special": true
1162
- },
1163
- "128145": {
1164
- "content": "<|reserved_special_token_137|>",
1165
- "lstrip": false,
1166
- "normalized": false,
1167
- "rstrip": false,
1168
- "single_word": false,
1169
- "special": true
1170
- },
1171
- "128146": {
1172
- "content": "<|reserved_special_token_138|>",
1173
- "lstrip": false,
1174
- "normalized": false,
1175
- "rstrip": false,
1176
- "single_word": false,
1177
- "special": true
1178
- },
1179
- "128147": {
1180
- "content": "<|reserved_special_token_139|>",
1181
- "lstrip": false,
1182
- "normalized": false,
1183
- "rstrip": false,
1184
- "single_word": false,
1185
- "special": true
1186
- },
1187
- "128148": {
1188
- "content": "<|reserved_special_token_140|>",
1189
- "lstrip": false,
1190
- "normalized": false,
1191
- "rstrip": false,
1192
- "single_word": false,
1193
- "special": true
1194
- },
1195
- "128149": {
1196
- "content": "<|reserved_special_token_141|>",
1197
- "lstrip": false,
1198
- "normalized": false,
1199
- "rstrip": false,
1200
- "single_word": false,
1201
- "special": true
1202
- },
1203
- "128150": {
1204
- "content": "<|reserved_special_token_142|>",
1205
- "lstrip": false,
1206
- "normalized": false,
1207
- "rstrip": false,
1208
- "single_word": false,
1209
- "special": true
1210
- },
1211
- "128151": {
1212
- "content": "<|reserved_special_token_143|>",
1213
- "lstrip": false,
1214
- "normalized": false,
1215
- "rstrip": false,
1216
- "single_word": false,
1217
- "special": true
1218
- },
1219
- "128152": {
1220
- "content": "<|reserved_special_token_144|>",
1221
- "lstrip": false,
1222
- "normalized": false,
1223
- "rstrip": false,
1224
- "single_word": false,
1225
- "special": true
1226
- },
1227
- "128153": {
1228
- "content": "<|reserved_special_token_145|>",
1229
- "lstrip": false,
1230
- "normalized": false,
1231
- "rstrip": false,
1232
- "single_word": false,
1233
- "special": true
1234
- },
1235
- "128154": {
1236
- "content": "<|reserved_special_token_146|>",
1237
- "lstrip": false,
1238
- "normalized": false,
1239
- "rstrip": false,
1240
- "single_word": false,
1241
- "special": true
1242
- },
1243
- "128155": {
1244
- "content": "<|reserved_special_token_147|>",
1245
- "lstrip": false,
1246
- "normalized": false,
1247
- "rstrip": false,
1248
- "single_word": false,
1249
- "special": true
1250
- },
1251
- "128156": {
1252
- "content": "<|reserved_special_token_148|>",
1253
- "lstrip": false,
1254
- "normalized": false,
1255
- "rstrip": false,
1256
- "single_word": false,
1257
- "special": true
1258
- },
1259
- "128157": {
1260
- "content": "<|reserved_special_token_149|>",
1261
- "lstrip": false,
1262
- "normalized": false,
1263
- "rstrip": false,
1264
- "single_word": false,
1265
- "special": true
1266
- },
1267
- "128158": {
1268
- "content": "<|reserved_special_token_150|>",
1269
- "lstrip": false,
1270
- "normalized": false,
1271
- "rstrip": false,
1272
- "single_word": false,
1273
- "special": true
1274
- },
1275
- "128159": {
1276
- "content": "<|reserved_special_token_151|>",
1277
- "lstrip": false,
1278
- "normalized": false,
1279
- "rstrip": false,
1280
- "single_word": false,
1281
- "special": true
1282
- },
1283
- "128160": {
1284
- "content": "<|reserved_special_token_152|>",
1285
- "lstrip": false,
1286
- "normalized": false,
1287
- "rstrip": false,
1288
- "single_word": false,
1289
- "special": true
1290
- },
1291
- "128161": {
1292
- "content": "<|reserved_special_token_153|>",
1293
- "lstrip": false,
1294
- "normalized": false,
1295
- "rstrip": false,
1296
- "single_word": false,
1297
- "special": true
1298
- },
1299
- "128162": {
1300
- "content": "<|reserved_special_token_154|>",
1301
- "lstrip": false,
1302
- "normalized": false,
1303
- "rstrip": false,
1304
- "single_word": false,
1305
- "special": true
1306
- },
1307
- "128163": {
1308
- "content": "<|reserved_special_token_155|>",
1309
- "lstrip": false,
1310
- "normalized": false,
1311
- "rstrip": false,
1312
- "single_word": false,
1313
- "special": true
1314
- },
1315
- "128164": {
1316
- "content": "<|reserved_special_token_156|>",
1317
- "lstrip": false,
1318
- "normalized": false,
1319
- "rstrip": false,
1320
- "single_word": false,
1321
- "special": true
1322
- },
1323
- "128165": {
1324
- "content": "<|reserved_special_token_157|>",
1325
- "lstrip": false,
1326
- "normalized": false,
1327
- "rstrip": false,
1328
- "single_word": false,
1329
- "special": true
1330
- },
1331
- "128166": {
1332
- "content": "<|reserved_special_token_158|>",
1333
- "lstrip": false,
1334
- "normalized": false,
1335
- "rstrip": false,
1336
- "single_word": false,
1337
- "special": true
1338
- },
1339
- "128167": {
1340
- "content": "<|reserved_special_token_159|>",
1341
- "lstrip": false,
1342
- "normalized": false,
1343
- "rstrip": false,
1344
- "single_word": false,
1345
- "special": true
1346
- },
1347
- "128168": {
1348
- "content": "<|reserved_special_token_160|>",
1349
- "lstrip": false,
1350
- "normalized": false,
1351
- "rstrip": false,
1352
- "single_word": false,
1353
- "special": true
1354
- },
1355
- "128169": {
1356
- "content": "<|reserved_special_token_161|>",
1357
- "lstrip": false,
1358
- "normalized": false,
1359
- "rstrip": false,
1360
- "single_word": false,
1361
- "special": true
1362
- },
1363
- "128170": {
1364
- "content": "<|reserved_special_token_162|>",
1365
- "lstrip": false,
1366
- "normalized": false,
1367
- "rstrip": false,
1368
- "single_word": false,
1369
- "special": true
1370
- },
1371
- "128171": {
1372
- "content": "<|reserved_special_token_163|>",
1373
- "lstrip": false,
1374
- "normalized": false,
1375
- "rstrip": false,
1376
- "single_word": false,
1377
- "special": true
1378
- },
1379
- "128172": {
1380
- "content": "<|reserved_special_token_164|>",
1381
- "lstrip": false,
1382
- "normalized": false,
1383
- "rstrip": false,
1384
- "single_word": false,
1385
- "special": true
1386
- },
1387
- "128173": {
1388
- "content": "<|reserved_special_token_165|>",
1389
- "lstrip": false,
1390
- "normalized": false,
1391
- "rstrip": false,
1392
- "single_word": false,
1393
- "special": true
1394
- },
1395
- "128174": {
1396
- "content": "<|reserved_special_token_166|>",
1397
- "lstrip": false,
1398
- "normalized": false,
1399
- "rstrip": false,
1400
- "single_word": false,
1401
- "special": true
1402
- },
1403
- "128175": {
1404
- "content": "<|reserved_special_token_167|>",
1405
- "lstrip": false,
1406
- "normalized": false,
1407
- "rstrip": false,
1408
- "single_word": false,
1409
- "special": true
1410
- },
1411
- "128176": {
1412
- "content": "<|reserved_special_token_168|>",
1413
- "lstrip": false,
1414
- "normalized": false,
1415
- "rstrip": false,
1416
- "single_word": false,
1417
- "special": true
1418
- },
1419
- "128177": {
1420
- "content": "<|reserved_special_token_169|>",
1421
- "lstrip": false,
1422
- "normalized": false,
1423
- "rstrip": false,
1424
- "single_word": false,
1425
- "special": true
1426
- },
1427
- "128178": {
1428
- "content": "<|reserved_special_token_170|>",
1429
- "lstrip": false,
1430
- "normalized": false,
1431
- "rstrip": false,
1432
- "single_word": false,
1433
- "special": true
1434
- },
1435
- "128179": {
1436
- "content": "<|reserved_special_token_171|>",
1437
- "lstrip": false,
1438
- "normalized": false,
1439
- "rstrip": false,
1440
- "single_word": false,
1441
- "special": true
1442
- },
1443
- "128180": {
1444
- "content": "<|reserved_special_token_172|>",
1445
- "lstrip": false,
1446
- "normalized": false,
1447
- "rstrip": false,
1448
- "single_word": false,
1449
- "special": true
1450
- },
1451
- "128181": {
1452
- "content": "<|reserved_special_token_173|>",
1453
- "lstrip": false,
1454
- "normalized": false,
1455
- "rstrip": false,
1456
- "single_word": false,
1457
- "special": true
1458
- },
1459
- "128182": {
1460
- "content": "<|reserved_special_token_174|>",
1461
- "lstrip": false,
1462
- "normalized": false,
1463
- "rstrip": false,
1464
- "single_word": false,
1465
- "special": true
1466
- },
1467
- "128183": {
1468
- "content": "<|reserved_special_token_175|>",
1469
- "lstrip": false,
1470
- "normalized": false,
1471
- "rstrip": false,
1472
- "single_word": false,
1473
- "special": true
1474
- },
1475
- "128184": {
1476
- "content": "<|reserved_special_token_176|>",
1477
- "lstrip": false,
1478
- "normalized": false,
1479
- "rstrip": false,
1480
- "single_word": false,
1481
- "special": true
1482
- },
1483
- "128185": {
1484
- "content": "<|reserved_special_token_177|>",
1485
- "lstrip": false,
1486
- "normalized": false,
1487
- "rstrip": false,
1488
- "single_word": false,
1489
- "special": true
1490
- },
1491
- "128186": {
1492
- "content": "<|reserved_special_token_178|>",
1493
- "lstrip": false,
1494
- "normalized": false,
1495
- "rstrip": false,
1496
- "single_word": false,
1497
- "special": true
1498
- },
1499
- "128187": {
1500
- "content": "<|reserved_special_token_179|>",
1501
- "lstrip": false,
1502
- "normalized": false,
1503
- "rstrip": false,
1504
- "single_word": false,
1505
- "special": true
1506
- },
1507
- "128188": {
1508
- "content": "<|reserved_special_token_180|>",
1509
- "lstrip": false,
1510
- "normalized": false,
1511
- "rstrip": false,
1512
- "single_word": false,
1513
- "special": true
1514
- },
1515
- "128189": {
1516
- "content": "<|reserved_special_token_181|>",
1517
- "lstrip": false,
1518
- "normalized": false,
1519
- "rstrip": false,
1520
- "single_word": false,
1521
- "special": true
1522
- },
1523
- "128190": {
1524
- "content": "<|reserved_special_token_182|>",
1525
- "lstrip": false,
1526
- "normalized": false,
1527
- "rstrip": false,
1528
- "single_word": false,
1529
- "special": true
1530
- },
1531
- "128191": {
1532
- "content": "<|reserved_special_token_183|>",
1533
- "lstrip": false,
1534
- "normalized": false,
1535
- "rstrip": false,
1536
- "single_word": false,
1537
- "special": true
1538
- },
1539
- "128192": {
1540
- "content": "<|reserved_special_token_184|>",
1541
- "lstrip": false,
1542
- "normalized": false,
1543
- "rstrip": false,
1544
- "single_word": false,
1545
- "special": true
1546
- },
1547
- "128193": {
1548
- "content": "<|reserved_special_token_185|>",
1549
- "lstrip": false,
1550
- "normalized": false,
1551
- "rstrip": false,
1552
- "single_word": false,
1553
- "special": true
1554
- },
1555
- "128194": {
1556
- "content": "<|reserved_special_token_186|>",
1557
- "lstrip": false,
1558
- "normalized": false,
1559
- "rstrip": false,
1560
- "single_word": false,
1561
- "special": true
1562
- },
1563
- "128195": {
1564
- "content": "<|reserved_special_token_187|>",
1565
- "lstrip": false,
1566
- "normalized": false,
1567
- "rstrip": false,
1568
- "single_word": false,
1569
- "special": true
1570
- },
1571
- "128196": {
1572
- "content": "<|reserved_special_token_188|>",
1573
- "lstrip": false,
1574
- "normalized": false,
1575
- "rstrip": false,
1576
- "single_word": false,
1577
- "special": true
1578
- },
1579
- "128197": {
1580
- "content": "<|reserved_special_token_189|>",
1581
- "lstrip": false,
1582
- "normalized": false,
1583
- "rstrip": false,
1584
- "single_word": false,
1585
- "special": true
1586
- },
1587
- "128198": {
1588
- "content": "<|reserved_special_token_190|>",
1589
- "lstrip": false,
1590
- "normalized": false,
1591
- "rstrip": false,
1592
- "single_word": false,
1593
- "special": true
1594
- },
1595
- "128199": {
1596
- "content": "<|reserved_special_token_191|>",
1597
- "lstrip": false,
1598
- "normalized": false,
1599
- "rstrip": false,
1600
- "single_word": false,
1601
- "special": true
1602
- },
1603
- "128200": {
1604
- "content": "<|reserved_special_token_192|>",
1605
- "lstrip": false,
1606
- "normalized": false,
1607
- "rstrip": false,
1608
- "single_word": false,
1609
- "special": true
1610
- },
1611
- "128201": {
1612
- "content": "<|reserved_special_token_193|>",
1613
- "lstrip": false,
1614
- "normalized": false,
1615
- "rstrip": false,
1616
- "single_word": false,
1617
- "special": true
1618
- },
1619
- "128202": {
1620
- "content": "<|reserved_special_token_194|>",
1621
- "lstrip": false,
1622
- "normalized": false,
1623
- "rstrip": false,
1624
- "single_word": false,
1625
- "special": true
1626
- },
1627
- "128203": {
1628
- "content": "<|reserved_special_token_195|>",
1629
- "lstrip": false,
1630
- "normalized": false,
1631
- "rstrip": false,
1632
- "single_word": false,
1633
- "special": true
1634
- },
1635
- "128204": {
1636
- "content": "<|reserved_special_token_196|>",
1637
- "lstrip": false,
1638
- "normalized": false,
1639
- "rstrip": false,
1640
- "single_word": false,
1641
- "special": true
1642
- },
1643
- "128205": {
1644
- "content": "<|reserved_special_token_197|>",
1645
- "lstrip": false,
1646
- "normalized": false,
1647
- "rstrip": false,
1648
- "single_word": false,
1649
- "special": true
1650
- },
1651
- "128206": {
1652
- "content": "<|reserved_special_token_198|>",
1653
- "lstrip": false,
1654
- "normalized": false,
1655
- "rstrip": false,
1656
- "single_word": false,
1657
- "special": true
1658
- },
1659
- "128207": {
1660
- "content": "<|reserved_special_token_199|>",
1661
- "lstrip": false,
1662
- "normalized": false,
1663
- "rstrip": false,
1664
- "single_word": false,
1665
- "special": true
1666
- },
1667
- "128208": {
1668
- "content": "<|reserved_special_token_200|>",
1669
- "lstrip": false,
1670
- "normalized": false,
1671
- "rstrip": false,
1672
- "single_word": false,
1673
- "special": true
1674
- },
1675
- "128209": {
1676
- "content": "<|reserved_special_token_201|>",
1677
- "lstrip": false,
1678
- "normalized": false,
1679
- "rstrip": false,
1680
- "single_word": false,
1681
- "special": true
1682
- },
1683
- "128210": {
1684
- "content": "<|reserved_special_token_202|>",
1685
- "lstrip": false,
1686
- "normalized": false,
1687
- "rstrip": false,
1688
- "single_word": false,
1689
- "special": true
1690
- },
1691
- "128211": {
1692
- "content": "<|reserved_special_token_203|>",
1693
- "lstrip": false,
1694
- "normalized": false,
1695
- "rstrip": false,
1696
- "single_word": false,
1697
- "special": true
1698
- },
1699
- "128212": {
1700
- "content": "<|reserved_special_token_204|>",
1701
- "lstrip": false,
1702
- "normalized": false,
1703
- "rstrip": false,
1704
- "single_word": false,
1705
- "special": true
1706
- },
1707
- "128213": {
1708
- "content": "<|reserved_special_token_205|>",
1709
- "lstrip": false,
1710
- "normalized": false,
1711
- "rstrip": false,
1712
- "single_word": false,
1713
- "special": true
1714
- },
1715
- "128214": {
1716
- "content": "<|reserved_special_token_206|>",
1717
- "lstrip": false,
1718
- "normalized": false,
1719
- "rstrip": false,
1720
- "single_word": false,
1721
- "special": true
1722
- },
1723
- "128215": {
1724
- "content": "<|reserved_special_token_207|>",
1725
- "lstrip": false,
1726
- "normalized": false,
1727
- "rstrip": false,
1728
- "single_word": false,
1729
- "special": true
1730
- },
1731
- "128216": {
1732
- "content": "<|reserved_special_token_208|>",
1733
- "lstrip": false,
1734
- "normalized": false,
1735
- "rstrip": false,
1736
- "single_word": false,
1737
- "special": true
1738
- },
1739
- "128217": {
1740
- "content": "<|reserved_special_token_209|>",
1741
- "lstrip": false,
1742
- "normalized": false,
1743
- "rstrip": false,
1744
- "single_word": false,
1745
- "special": true
1746
- },
1747
- "128218": {
1748
- "content": "<|reserved_special_token_210|>",
1749
- "lstrip": false,
1750
- "normalized": false,
1751
- "rstrip": false,
1752
- "single_word": false,
1753
- "special": true
1754
- },
1755
- "128219": {
1756
- "content": "<|reserved_special_token_211|>",
1757
- "lstrip": false,
1758
- "normalized": false,
1759
- "rstrip": false,
1760
- "single_word": false,
1761
- "special": true
1762
- },
1763
- "128220": {
1764
- "content": "<|reserved_special_token_212|>",
1765
- "lstrip": false,
1766
- "normalized": false,
1767
- "rstrip": false,
1768
- "single_word": false,
1769
- "special": true
1770
- },
1771
- "128221": {
1772
- "content": "<|reserved_special_token_213|>",
1773
- "lstrip": false,
1774
- "normalized": false,
1775
- "rstrip": false,
1776
- "single_word": false,
1777
- "special": true
1778
- },
1779
- "128222": {
1780
- "content": "<|reserved_special_token_214|>",
1781
- "lstrip": false,
1782
- "normalized": false,
1783
- "rstrip": false,
1784
- "single_word": false,
1785
- "special": true
1786
- },
1787
- "128223": {
1788
- "content": "<|reserved_special_token_215|>",
1789
- "lstrip": false,
1790
- "normalized": false,
1791
- "rstrip": false,
1792
- "single_word": false,
1793
- "special": true
1794
- },
1795
- "128224": {
1796
- "content": "<|reserved_special_token_216|>",
1797
- "lstrip": false,
1798
- "normalized": false,
1799
- "rstrip": false,
1800
- "single_word": false,
1801
- "special": true
1802
- },
1803
- "128225": {
1804
- "content": "<|reserved_special_token_217|>",
1805
- "lstrip": false,
1806
- "normalized": false,
1807
- "rstrip": false,
1808
- "single_word": false,
1809
- "special": true
1810
- },
1811
- "128226": {
1812
- "content": "<|reserved_special_token_218|>",
1813
- "lstrip": false,
1814
- "normalized": false,
1815
- "rstrip": false,
1816
- "single_word": false,
1817
- "special": true
1818
- },
1819
- "128227": {
1820
- "content": "<|reserved_special_token_219|>",
1821
- "lstrip": false,
1822
- "normalized": false,
1823
- "rstrip": false,
1824
- "single_word": false,
1825
- "special": true
1826
- },
1827
- "128228": {
1828
- "content": "<|reserved_special_token_220|>",
1829
- "lstrip": false,
1830
- "normalized": false,
1831
- "rstrip": false,
1832
- "single_word": false,
1833
- "special": true
1834
- },
1835
- "128229": {
1836
- "content": "<|reserved_special_token_221|>",
1837
- "lstrip": false,
1838
- "normalized": false,
1839
- "rstrip": false,
1840
- "single_word": false,
1841
- "special": true
1842
- },
1843
- "128230": {
1844
- "content": "<|reserved_special_token_222|>",
1845
- "lstrip": false,
1846
- "normalized": false,
1847
- "rstrip": false,
1848
- "single_word": false,
1849
- "special": true
1850
- },
1851
- "128231": {
1852
- "content": "<|reserved_special_token_223|>",
1853
- "lstrip": false,
1854
- "normalized": false,
1855
- "rstrip": false,
1856
- "single_word": false,
1857
- "special": true
1858
- },
1859
- "128232": {
1860
- "content": "<|reserved_special_token_224|>",
1861
- "lstrip": false,
1862
- "normalized": false,
1863
- "rstrip": false,
1864
- "single_word": false,
1865
- "special": true
1866
- },
1867
- "128233": {
1868
- "content": "<|reserved_special_token_225|>",
1869
- "lstrip": false,
1870
- "normalized": false,
1871
- "rstrip": false,
1872
- "single_word": false,
1873
- "special": true
1874
- },
1875
- "128234": {
1876
- "content": "<|reserved_special_token_226|>",
1877
- "lstrip": false,
1878
- "normalized": false,
1879
- "rstrip": false,
1880
- "single_word": false,
1881
- "special": true
1882
- },
1883
- "128235": {
1884
- "content": "<|reserved_special_token_227|>",
1885
- "lstrip": false,
1886
- "normalized": false,
1887
- "rstrip": false,
1888
- "single_word": false,
1889
- "special": true
1890
- },
1891
- "128236": {
1892
- "content": "<|reserved_special_token_228|>",
1893
- "lstrip": false,
1894
- "normalized": false,
1895
- "rstrip": false,
1896
- "single_word": false,
1897
- "special": true
1898
- },
1899
- "128237": {
1900
- "content": "<|reserved_special_token_229|>",
1901
- "lstrip": false,
1902
- "normalized": false,
1903
- "rstrip": false,
1904
- "single_word": false,
1905
- "special": true
1906
- },
1907
- "128238": {
1908
- "content": "<|reserved_special_token_230|>",
1909
- "lstrip": false,
1910
- "normalized": false,
1911
- "rstrip": false,
1912
- "single_word": false,
1913
- "special": true
1914
- },
1915
- "128239": {
1916
- "content": "<|reserved_special_token_231|>",
1917
- "lstrip": false,
1918
- "normalized": false,
1919
- "rstrip": false,
1920
- "single_word": false,
1921
- "special": true
1922
- },
1923
- "128240": {
1924
- "content": "<|reserved_special_token_232|>",
1925
- "lstrip": false,
1926
- "normalized": false,
1927
- "rstrip": false,
1928
- "single_word": false,
1929
- "special": true
1930
- },
1931
- "128241": {
1932
- "content": "<|reserved_special_token_233|>",
1933
- "lstrip": false,
1934
- "normalized": false,
1935
- "rstrip": false,
1936
- "single_word": false,
1937
- "special": true
1938
- },
1939
- "128242": {
1940
- "content": "<|reserved_special_token_234|>",
1941
- "lstrip": false,
1942
- "normalized": false,
1943
- "rstrip": false,
1944
- "single_word": false,
1945
- "special": true
1946
- },
1947
- "128243": {
1948
- "content": "<|reserved_special_token_235|>",
1949
- "lstrip": false,
1950
- "normalized": false,
1951
- "rstrip": false,
1952
- "single_word": false,
1953
- "special": true
1954
- },
1955
- "128244": {
1956
- "content": "<|reserved_special_token_236|>",
1957
- "lstrip": false,
1958
- "normalized": false,
1959
- "rstrip": false,
1960
- "single_word": false,
1961
- "special": true
1962
- },
1963
- "128245": {
1964
- "content": "<|reserved_special_token_237|>",
1965
- "lstrip": false,
1966
- "normalized": false,
1967
- "rstrip": false,
1968
- "single_word": false,
1969
- "special": true
1970
- },
1971
- "128246": {
1972
- "content": "<|reserved_special_token_238|>",
1973
- "lstrip": false,
1974
- "normalized": false,
1975
- "rstrip": false,
1976
- "single_word": false,
1977
- "special": true
1978
- },
1979
- "128247": {
1980
- "content": "<|reserved_special_token_239|>",
1981
- "lstrip": false,
1982
- "normalized": false,
1983
- "rstrip": false,
1984
- "single_word": false,
1985
- "special": true
1986
- },
1987
- "128248": {
1988
- "content": "<|reserved_special_token_240|>",
1989
- "lstrip": false,
1990
- "normalized": false,
1991
- "rstrip": false,
1992
- "single_word": false,
1993
- "special": true
1994
- },
1995
- "128249": {
1996
- "content": "<|reserved_special_token_241|>",
1997
- "lstrip": false,
1998
- "normalized": false,
1999
- "rstrip": false,
2000
- "single_word": false,
2001
- "special": true
2002
- },
2003
- "128250": {
2004
- "content": "<|reserved_special_token_242|>",
2005
- "lstrip": false,
2006
- "normalized": false,
2007
- "rstrip": false,
2008
- "single_word": false,
2009
- "special": true
2010
- },
2011
- "128251": {
2012
- "content": "<|reserved_special_token_243|>",
2013
- "lstrip": false,
2014
- "normalized": false,
2015
- "rstrip": false,
2016
- "single_word": false,
2017
- "special": true
2018
- },
2019
- "128252": {
2020
- "content": "<|reserved_special_token_244|>",
2021
- "lstrip": false,
2022
- "normalized": false,
2023
- "rstrip": false,
2024
- "single_word": false,
2025
- "special": true
2026
- },
2027
- "128253": {
2028
- "content": "<|reserved_special_token_245|>",
2029
- "lstrip": false,
2030
- "normalized": false,
2031
- "rstrip": false,
2032
- "single_word": false,
2033
- "special": true
2034
- },
2035
- "128254": {
2036
- "content": "<|reserved_special_token_246|>",
2037
- "lstrip": false,
2038
- "normalized": false,
2039
- "rstrip": false,
2040
- "single_word": false,
2041
- "special": true
2042
- },
2043
- "128255": {
2044
- "content": "<|reserved_special_token_247|>",
2045
- "lstrip": false,
2046
- "normalized": false,
2047
- "rstrip": false,
2048
- "single_word": false,
2049
- "special": true
2050
- }
2051
- },
2052
- "bos_token": "<|begin_of_text|>",
2053
- "chat_template": "{{- bos_token }}\n{%- if custom_tools is defined %}\n {%- set tools = custom_tools %}\n{%- endif %}\n{%- if not tools_in_user_message is defined %}\n {%- set tools_in_user_message = true %}\n{%- endif %}\n{%- if not date_string is defined %}\n {%- if strftime_now is defined %}\n {%- set date_string = strftime_now(\"%d %b %Y\") %}\n {%- else %}\n {%- set date_string = \"26 Jul 2024\" %}\n {%- endif %}\n{%- endif %}\n{%- if not tools is defined %}\n {%- set tools = none %}\n{%- endif %}\n\n{#- This block extracts the system message, so we can slot it into the right place. #}\n{%- if messages[0]['role'] == 'system' %}\n {%- set system_message = messages[0]['content']|trim %}\n {%- set messages = messages[1:] %}\n{%- else %}\n {%- set system_message = \"\" %}\n{%- endif %}\n\n{#- System message #}\n{{- \"<|start_header_id|>system<|end_header_id|>\\n\\n\" }}\n{%- if tools is not none %}\n {{- \"Environment: ipython\\n\" }}\n{%- endif %}\n{{- \"Cutting Knowledge Date: December 2023\\n\" }}\n{{- \"Today Date: \" + date_string + \"\\n\\n\" }}\n{%- if tools is not none and not tools_in_user_message %}\n {{- \"You have access to the following functions. To call a function, please respond with JSON for a function call.\" }}\n {{- 'Respond in the format {\"name\": function name, \"parameters\": dictionary of argument name and its value}.' }}\n {{- \"Do not use variables.\\n\\n\" }}\n {%- for t in tools %}\n {{- t | tojson(indent=4) }}\n {{- \"\\n\\n\" }}\n {%- endfor %}\n{%- endif %}\n{{- system_message }}\n{{- \"<|eot_id|>\" }}\n\n{#- Custom tools are passed in a user message with some extra guidance #}\n{%- if tools_in_user_message and not tools is none %}\n {#- Extract the first user message so we can plug it in here #}\n {%- if messages | length != 0 %}\n {%- set first_user_message = messages[0]['content']|trim %}\n {%- set messages = messages[1:] %}\n {%- else %}\n {{- raise_exception(\"Cannot put tools in the first user message when there's no first user message!\") }}\n{%- endif %}\n {{- '<|start_header_id|>user<|end_header_id|>\\n\\n' -}}\n {{- \"Given the following functions, please respond with a JSON for a function call \" }}\n {{- \"with its proper arguments that best answers the given prompt.\\n\\n\" }}\n {{- 'Respond in the format {\"name\": function name, \"parameters\": dictionary of argument name and its value}.' }}\n {{- \"Do not use variables.\\n\\n\" }}\n {%- for t in tools %}\n {{- t | tojson(indent=4) }}\n {{- \"\\n\\n\" }}\n {%- endfor %}\n {{- first_user_message + \"<|eot_id|>\"}}\n{%- endif %}\n\n{%- for message in messages %}\n {%- if not (message.role == 'ipython' or message.role == 'tool' or 'tool_calls' in message) %}\n {{- '<|start_header_id|>' + message['role'] + '<|end_header_id|>\\n\\n'+ message['content'] | trim + '<|eot_id|>' }}\n {%- elif 'tool_calls' in message %}\n {%- if not message.tool_calls|length == 1 %}\n {{- raise_exception(\"This model only supports single tool-calls at once!\") }}\n {%- endif %}\n {%- set tool_call = message.tool_calls[0].function %}\n {{- '<|start_header_id|>assistant<|end_header_id|>\\n\\n' -}}\n {{- '{\"name\": \"' + tool_call.name + '\", ' }}\n {{- '\"parameters\": ' }}\n {{- tool_call.arguments | tojson }}\n {{- \"}\" }}\n {{- \"<|eot_id|>\" }}\n {%- elif message.role == \"tool\" or message.role == \"ipython\" %}\n {{- \"<|start_header_id|>ipython<|end_header_id|>\\n\\n\" }}\n {%- if message.content is mapping or message.content is iterable %}\n {{- message.content | tojson }}\n {%- else %}\n {{- message.content }}\n {%- endif %}\n {{- \"<|eot_id|>\" }}\n {%- endif %}\n{%- endfor %}\n{%- if add_generation_prompt %}\n {{- '<|start_header_id|>assistant<|end_header_id|>\\n\\n' }}\n{%- endif %}\n",
2054
- "clean_up_tokenization_spaces": true,
2055
- "eos_token": "<|eot_id|>",
2056
- "extra_special_tokens": {},
2057
- "model_input_names": [
2058
- "input_ids",
2059
- "attention_mask"
2060
- ],
2061
- "model_max_length": 131072,
2062
- "pad_token": "<|eot_id|>",
2063
- "tokenizer_class": "PreTrainedTokenizerFast"
2064
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
checkpoints/checkpoint-900/trainer_state.json DELETED
@@ -1,1365 +0,0 @@
1
- {
2
- "best_metric": null,
3
- "best_model_checkpoint": null,
4
- "epoch": 2.759571209800919,
5
- "eval_steps": 100,
6
- "global_step": 900,
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.015313935681470138,
13
- "grad_norm": 3.8326407564846536,
14
- "learning_rate": 6.134969325153375e-07,
15
- "loss": 1.3339,
16
- "step": 5
17
- },
18
- {
19
- "epoch": 0.030627871362940276,
20
- "grad_norm": 3.5496788557416803,
21
- "learning_rate": 1.226993865030675e-06,
22
- "loss": 1.3517,
23
- "step": 10
24
- },
25
- {
26
- "epoch": 0.045941807044410414,
27
- "grad_norm": 2.292172802820395,
28
- "learning_rate": 1.8404907975460124e-06,
29
- "loss": 1.3142,
30
- "step": 15
31
- },
32
- {
33
- "epoch": 0.06125574272588055,
34
- "grad_norm": 1.7775340318261912,
35
- "learning_rate": 2.45398773006135e-06,
36
- "loss": 1.2477,
37
- "step": 20
38
- },
39
- {
40
- "epoch": 0.07656967840735068,
41
- "grad_norm": 1.4368794370978828,
42
- "learning_rate": 3.0674846625766875e-06,
43
- "loss": 1.2037,
44
- "step": 25
45
- },
46
- {
47
- "epoch": 0.09188361408882083,
48
- "grad_norm": 1.061354555879628,
49
- "learning_rate": 3.680981595092025e-06,
50
- "loss": 1.1505,
51
- "step": 30
52
- },
53
- {
54
- "epoch": 0.10719754977029096,
55
- "grad_norm": 0.8377616484610274,
56
- "learning_rate": 4.294478527607362e-06,
57
- "loss": 1.1179,
58
- "step": 35
59
- },
60
- {
61
- "epoch": 0.1225114854517611,
62
- "grad_norm": 0.7884987790196414,
63
- "learning_rate": 4.9079754601227e-06,
64
- "loss": 1.0735,
65
- "step": 40
66
- },
67
- {
68
- "epoch": 0.13782542113323124,
69
- "grad_norm": 0.7279957151761491,
70
- "learning_rate": 5.521472392638038e-06,
71
- "loss": 1.0556,
72
- "step": 45
73
- },
74
- {
75
- "epoch": 0.15313935681470137,
76
- "grad_norm": 0.6692858930956382,
77
- "learning_rate": 6.134969325153375e-06,
78
- "loss": 1.0294,
79
- "step": 50
80
- },
81
- {
82
- "epoch": 0.16845329249617153,
83
- "grad_norm": 0.6401840707040195,
84
- "learning_rate": 6.748466257668712e-06,
85
- "loss": 1.0175,
86
- "step": 55
87
- },
88
- {
89
- "epoch": 0.18376722817764166,
90
- "grad_norm": 0.6584204446837412,
91
- "learning_rate": 7.36196319018405e-06,
92
- "loss": 1.0283,
93
- "step": 60
94
- },
95
- {
96
- "epoch": 0.1990811638591118,
97
- "grad_norm": 0.6647811794468166,
98
- "learning_rate": 7.975460122699386e-06,
99
- "loss": 0.9799,
100
- "step": 65
101
- },
102
- {
103
- "epoch": 0.21439509954058192,
104
- "grad_norm": 0.6774675226534475,
105
- "learning_rate": 8.588957055214725e-06,
106
- "loss": 0.969,
107
- "step": 70
108
- },
109
- {
110
- "epoch": 0.22970903522205208,
111
- "grad_norm": 0.6918496726012217,
112
- "learning_rate": 9.202453987730062e-06,
113
- "loss": 0.9787,
114
- "step": 75
115
- },
116
- {
117
- "epoch": 0.2450229709035222,
118
- "grad_norm": 0.6740082248654927,
119
- "learning_rate": 9.8159509202454e-06,
120
- "loss": 0.9596,
121
- "step": 80
122
- },
123
- {
124
- "epoch": 0.26033690658499237,
125
- "grad_norm": 0.6791518891702232,
126
- "learning_rate": 1.0429447852760737e-05,
127
- "loss": 0.9813,
128
- "step": 85
129
- },
130
- {
131
- "epoch": 0.27565084226646247,
132
- "grad_norm": 0.7029695620068616,
133
- "learning_rate": 1.1042944785276076e-05,
134
- "loss": 0.9703,
135
- "step": 90
136
- },
137
- {
138
- "epoch": 0.29096477794793263,
139
- "grad_norm": 0.6355433423695775,
140
- "learning_rate": 1.1656441717791411e-05,
141
- "loss": 0.9375,
142
- "step": 95
143
- },
144
- {
145
- "epoch": 0.30627871362940273,
146
- "grad_norm": 0.6551855479751819,
147
- "learning_rate": 1.226993865030675e-05,
148
- "loss": 0.952,
149
- "step": 100
150
- },
151
- {
152
- "epoch": 0.30627871362940273,
153
- "eval_loss": 0.964697003364563,
154
- "eval_runtime": 3.4019,
155
- "eval_samples_per_second": 36.45,
156
- "eval_steps_per_second": 2.352,
157
- "step": 100
158
- },
159
- {
160
- "epoch": 0.3215926493108729,
161
- "grad_norm": 0.6655305965463972,
162
- "learning_rate": 1.2883435582822085e-05,
163
- "loss": 0.9237,
164
- "step": 105
165
- },
166
- {
167
- "epoch": 0.33690658499234305,
168
- "grad_norm": 0.7257201177401903,
169
- "learning_rate": 1.3496932515337424e-05,
170
- "loss": 0.9511,
171
- "step": 110
172
- },
173
- {
174
- "epoch": 0.35222052067381315,
175
- "grad_norm": 0.7424667090609683,
176
- "learning_rate": 1.4110429447852763e-05,
177
- "loss": 0.9323,
178
- "step": 115
179
- },
180
- {
181
- "epoch": 0.3675344563552833,
182
- "grad_norm": 0.7336582339806711,
183
- "learning_rate": 1.47239263803681e-05,
184
- "loss": 0.9449,
185
- "step": 120
186
- },
187
- {
188
- "epoch": 0.38284839203675347,
189
- "grad_norm": 0.6851538651091853,
190
- "learning_rate": 1.5337423312883436e-05,
191
- "loss": 0.9133,
192
- "step": 125
193
- },
194
- {
195
- "epoch": 0.3981623277182236,
196
- "grad_norm": 0.7424313941044212,
197
- "learning_rate": 1.5950920245398772e-05,
198
- "loss": 0.9173,
199
- "step": 130
200
- },
201
- {
202
- "epoch": 0.41347626339969373,
203
- "grad_norm": 0.7937170928985942,
204
- "learning_rate": 1.656441717791411e-05,
205
- "loss": 0.9348,
206
- "step": 135
207
- },
208
- {
209
- "epoch": 0.42879019908116384,
210
- "grad_norm": 0.7668730085289572,
211
- "learning_rate": 1.717791411042945e-05,
212
- "loss": 0.9091,
213
- "step": 140
214
- },
215
- {
216
- "epoch": 0.444104134762634,
217
- "grad_norm": 0.713212902268727,
218
- "learning_rate": 1.7791411042944788e-05,
219
- "loss": 0.9054,
220
- "step": 145
221
- },
222
- {
223
- "epoch": 0.45941807044410415,
224
- "grad_norm": 0.8266274957992115,
225
- "learning_rate": 1.8404907975460123e-05,
226
- "loss": 0.9165,
227
- "step": 150
228
- },
229
- {
230
- "epoch": 0.47473200612557426,
231
- "grad_norm": 0.7773849012810377,
232
- "learning_rate": 1.9018404907975462e-05,
233
- "loss": 0.9261,
234
- "step": 155
235
- },
236
- {
237
- "epoch": 0.4900459418070444,
238
- "grad_norm": 0.7698543962501324,
239
- "learning_rate": 1.96319018404908e-05,
240
- "loss": 0.8848,
241
- "step": 160
242
- },
243
- {
244
- "epoch": 0.5053598774885145,
245
- "grad_norm": 0.7326492641075325,
246
- "learning_rate": 1.9999908278985548e-05,
247
- "loss": 0.9002,
248
- "step": 165
249
- },
250
- {
251
- "epoch": 0.5206738131699847,
252
- "grad_norm": 0.725915155149828,
253
- "learning_rate": 1.9998876436895888e-05,
254
- "loss": 0.9064,
255
- "step": 170
256
- },
257
- {
258
- "epoch": 0.5359877488514548,
259
- "grad_norm": 0.7137681931522117,
260
- "learning_rate": 1.99966982201439e-05,
261
- "loss": 0.9083,
262
- "step": 175
263
- },
264
- {
265
- "epoch": 0.5513016845329249,
266
- "grad_norm": 0.8244990854189328,
267
- "learning_rate": 1.9993373878462894e-05,
268
- "loss": 0.8988,
269
- "step": 180
270
- },
271
- {
272
- "epoch": 0.5666156202143952,
273
- "grad_norm": 0.7883005110592801,
274
- "learning_rate": 1.99889037929898e-05,
275
- "loss": 0.9021,
276
- "step": 185
277
- },
278
- {
279
- "epoch": 0.5819295558958653,
280
- "grad_norm": 0.7224060026030372,
281
- "learning_rate": 1.9983288476221482e-05,
282
- "loss": 0.8881,
283
- "step": 190
284
- },
285
- {
286
- "epoch": 0.5972434915773354,
287
- "grad_norm": 0.7010662098507998,
288
- "learning_rate": 1.9976528571955946e-05,
289
- "loss": 0.8889,
290
- "step": 195
291
- },
292
- {
293
- "epoch": 0.6125574272588055,
294
- "grad_norm": 0.7154815490781321,
295
- "learning_rate": 1.9968624855218578e-05,
296
- "loss": 0.8884,
297
- "step": 200
298
- },
299
- {
300
- "epoch": 0.6125574272588055,
301
- "eval_loss": 0.9119902849197388,
302
- "eval_runtime": 3.4604,
303
- "eval_samples_per_second": 35.834,
304
- "eval_steps_per_second": 2.312,
305
- "step": 200
306
- },
307
- {
308
- "epoch": 0.6278713629402757,
309
- "grad_norm": 0.6984184509316927,
310
- "learning_rate": 1.995957823217325e-05,
311
- "loss": 0.9033,
312
- "step": 205
313
- },
314
- {
315
- "epoch": 0.6431852986217458,
316
- "grad_norm": 0.7120998555383481,
317
- "learning_rate": 1.9949389740018438e-05,
318
- "loss": 0.8988,
319
- "step": 210
320
- },
321
- {
322
- "epoch": 0.6584992343032159,
323
- "grad_norm": 0.66696186884436,
324
- "learning_rate": 1.993806054686832e-05,
325
- "loss": 0.8735,
326
- "step": 215
327
- },
328
- {
329
- "epoch": 0.6738131699846861,
330
- "grad_norm": 0.7395425655626215,
331
- "learning_rate": 1.9925591951618822e-05,
332
- "loss": 0.9127,
333
- "step": 220
334
- },
335
- {
336
- "epoch": 0.6891271056661562,
337
- "grad_norm": 0.6960401538019414,
338
- "learning_rate": 1.9911985383798737e-05,
339
- "loss": 0.875,
340
- "step": 225
341
- },
342
- {
343
- "epoch": 0.7044410413476263,
344
- "grad_norm": 0.7195874036136765,
345
- "learning_rate": 1.9897242403405792e-05,
346
- "loss": 0.8782,
347
- "step": 230
348
- },
349
- {
350
- "epoch": 0.7197549770290965,
351
- "grad_norm": 0.7129931764921758,
352
- "learning_rate": 1.9881364700727827e-05,
353
- "loss": 0.8943,
354
- "step": 235
355
- },
356
- {
357
- "epoch": 0.7350689127105666,
358
- "grad_norm": 0.690645634714852,
359
- "learning_rate": 1.9864354096148966e-05,
360
- "loss": 0.8782,
361
- "step": 240
362
- },
363
- {
364
- "epoch": 0.7503828483920367,
365
- "grad_norm": 0.7055958514936376,
366
- "learning_rate": 1.9846212539940955e-05,
367
- "loss": 0.8867,
368
- "step": 245
369
- },
370
- {
371
- "epoch": 0.7656967840735069,
372
- "grad_norm": 0.7517885221069497,
373
- "learning_rate": 1.982694211203952e-05,
374
- "loss": 0.8843,
375
- "step": 250
376
- },
377
- {
378
- "epoch": 0.781010719754977,
379
- "grad_norm": 0.7270711746255435,
380
- "learning_rate": 1.9806545021805922e-05,
381
- "loss": 0.8757,
382
- "step": 255
383
- },
384
- {
385
- "epoch": 0.7963246554364471,
386
- "grad_norm": 0.707252502344583,
387
- "learning_rate": 1.9785023607773655e-05,
388
- "loss": 0.8842,
389
- "step": 260
390
- },
391
- {
392
- "epoch": 0.8116385911179173,
393
- "grad_norm": 0.8083132789219781,
394
- "learning_rate": 1.976238033738033e-05,
395
- "loss": 0.8578,
396
- "step": 265
397
- },
398
- {
399
- "epoch": 0.8269525267993875,
400
- "grad_norm": 0.7287731737231247,
401
- "learning_rate": 1.9738617806684767e-05,
402
- "loss": 0.911,
403
- "step": 270
404
- },
405
- {
406
- "epoch": 0.8422664624808576,
407
- "grad_norm": 0.7683997901360534,
408
- "learning_rate": 1.9713738740069384e-05,
409
- "loss": 0.8607,
410
- "step": 275
411
- },
412
- {
413
- "epoch": 0.8575803981623277,
414
- "grad_norm": 0.6883196631913017,
415
- "learning_rate": 1.9687745989927823e-05,
416
- "loss": 0.8781,
417
- "step": 280
418
- },
419
- {
420
- "epoch": 0.8728943338437979,
421
- "grad_norm": 0.7137416791751114,
422
- "learning_rate": 1.966064253633793e-05,
423
- "loss": 0.8675,
424
- "step": 285
425
- },
426
- {
427
- "epoch": 0.888208269525268,
428
- "grad_norm": 0.800711538711323,
429
- "learning_rate": 1.963243148672009e-05,
430
- "loss": 0.8696,
431
- "step": 290
432
- },
433
- {
434
- "epoch": 0.9035222052067381,
435
- "grad_norm": 0.7017619169785259,
436
- "learning_rate": 1.960311607548096e-05,
437
- "loss": 0.8724,
438
- "step": 295
439
- },
440
- {
441
- "epoch": 0.9188361408882083,
442
- "grad_norm": 0.7347091076360852,
443
- "learning_rate": 1.957269966364263e-05,
444
- "loss": 0.8753,
445
- "step": 300
446
- },
447
- {
448
- "epoch": 0.9188361408882083,
449
- "eval_loss": 0.8846515417098999,
450
- "eval_runtime": 3.4431,
451
- "eval_samples_per_second": 36.014,
452
- "eval_steps_per_second": 2.323,
453
- "step": 300
454
- },
455
- {
456
- "epoch": 0.9341500765696784,
457
- "grad_norm": 0.6840257399223311,
458
- "learning_rate": 1.9541185738457304e-05,
459
- "loss": 0.8661,
460
- "step": 305
461
- },
462
- {
463
- "epoch": 0.9494640122511485,
464
- "grad_norm": 0.6350789314835292,
465
- "learning_rate": 1.9508577913007475e-05,
466
- "loss": 0.866,
467
- "step": 310
468
- },
469
- {
470
- "epoch": 0.9647779479326187,
471
- "grad_norm": 0.742666392443459,
472
- "learning_rate": 1.9474879925791665e-05,
473
- "loss": 0.8712,
474
- "step": 315
475
- },
476
- {
477
- "epoch": 0.9800918836140888,
478
- "grad_norm": 0.7707700113231902,
479
- "learning_rate": 1.944009564029584e-05,
480
- "loss": 0.8656,
481
- "step": 320
482
- },
483
- {
484
- "epoch": 0.9954058192955589,
485
- "grad_norm": 0.7026681496550549,
486
- "learning_rate": 1.9404229044550432e-05,
487
- "loss": 0.8707,
488
- "step": 325
489
- },
490
- {
491
- "epoch": 1.0122511485451762,
492
- "grad_norm": 0.8698658929763603,
493
- "learning_rate": 1.9367284250673126e-05,
494
- "loss": 1.0103,
495
- "step": 330
496
- },
497
- {
498
- "epoch": 1.0275650842266462,
499
- "grad_norm": 0.7696705674515646,
500
- "learning_rate": 1.9329265494397386e-05,
501
- "loss": 0.8254,
502
- "step": 335
503
- },
504
- {
505
- "epoch": 1.0428790199081164,
506
- "grad_norm": 0.703085270940529,
507
- "learning_rate": 1.929017713458685e-05,
508
- "loss": 0.7994,
509
- "step": 340
510
- },
511
- {
512
- "epoch": 1.0581929555895866,
513
- "grad_norm": 0.7281489106195598,
514
- "learning_rate": 1.9250023652735573e-05,
515
- "loss": 0.7997,
516
- "step": 345
517
- },
518
- {
519
- "epoch": 1.0735068912710566,
520
- "grad_norm": 0.6721859356177987,
521
- "learning_rate": 1.920880965245422e-05,
522
- "loss": 0.8028,
523
- "step": 350
524
- },
525
- {
526
- "epoch": 1.0888208269525268,
527
- "grad_norm": 0.6999456629939552,
528
- "learning_rate": 1.9166539858942258e-05,
529
- "loss": 0.7923,
530
- "step": 355
531
- },
532
- {
533
- "epoch": 1.104134762633997,
534
- "grad_norm": 0.761057382772568,
535
- "learning_rate": 1.912321911844622e-05,
536
- "loss": 0.8023,
537
- "step": 360
538
- },
539
- {
540
- "epoch": 1.119448698315467,
541
- "grad_norm": 0.7613985157569584,
542
- "learning_rate": 1.907885239770408e-05,
543
- "loss": 0.8171,
544
- "step": 365
545
- },
546
- {
547
- "epoch": 1.1347626339969372,
548
- "grad_norm": 0.7130602486133167,
549
- "learning_rate": 1.9033444783375806e-05,
550
- "loss": 0.8136,
551
- "step": 370
552
- },
553
- {
554
- "epoch": 1.1500765696784074,
555
- "grad_norm": 0.7290477717213912,
556
- "learning_rate": 1.8987001481460177e-05,
557
- "loss": 0.7956,
558
- "step": 375
559
- },
560
- {
561
- "epoch": 1.1653905053598774,
562
- "grad_norm": 0.6942113670274667,
563
- "learning_rate": 1.8939527816697917e-05,
564
- "loss": 0.795,
565
- "step": 380
566
- },
567
- {
568
- "epoch": 1.1807044410413476,
569
- "grad_norm": 0.7278550945084655,
570
- "learning_rate": 1.8891029231961208e-05,
571
- "loss": 0.8038,
572
- "step": 385
573
- },
574
- {
575
- "epoch": 1.1960183767228179,
576
- "grad_norm": 0.7637922924754829,
577
- "learning_rate": 1.8841511287629667e-05,
578
- "loss": 0.7941,
579
- "step": 390
580
- },
581
- {
582
- "epoch": 1.2113323124042878,
583
- "grad_norm": 0.7227667767094055,
584
- "learning_rate": 1.8790979660952832e-05,
585
- "loss": 0.7899,
586
- "step": 395
587
- },
588
- {
589
- "epoch": 1.226646248085758,
590
- "grad_norm": 0.6856723635729023,
591
- "learning_rate": 1.8739440145399295e-05,
592
- "loss": 0.8037,
593
- "step": 400
594
- },
595
- {
596
- "epoch": 1.226646248085758,
597
- "eval_loss": 0.8724085092544556,
598
- "eval_runtime": 3.6883,
599
- "eval_samples_per_second": 33.619,
600
- "eval_steps_per_second": 2.169,
601
- "step": 400
602
- },
603
- {
604
- "epoch": 1.2419601837672283,
605
- "grad_norm": 0.6585093868236329,
606
- "learning_rate": 1.8686898649992437e-05,
607
- "loss": 0.7969,
608
- "step": 405
609
- },
610
- {
611
- "epoch": 1.2572741194486983,
612
- "grad_norm": 0.6787675213944868,
613
- "learning_rate": 1.8633361198632987e-05,
614
- "loss": 0.8077,
615
- "step": 410
616
- },
617
- {
618
- "epoch": 1.2725880551301685,
619
- "grad_norm": 0.6808724401749536,
620
- "learning_rate": 1.857883392940837e-05,
621
- "loss": 0.8077,
622
- "step": 415
623
- },
624
- {
625
- "epoch": 1.2879019908116387,
626
- "grad_norm": 0.6824672080638723,
627
- "learning_rate": 1.8523323093888983e-05,
628
- "loss": 0.8025,
629
- "step": 420
630
- },
631
- {
632
- "epoch": 1.3032159264931087,
633
- "grad_norm": 0.6611632916751219,
634
- "learning_rate": 1.8466835056411422e-05,
635
- "loss": 0.7875,
636
- "step": 425
637
- },
638
- {
639
- "epoch": 1.318529862174579,
640
- "grad_norm": 0.701746007873801,
641
- "learning_rate": 1.8409376293348836e-05,
642
- "loss": 0.794,
643
- "step": 430
644
- },
645
- {
646
- "epoch": 1.3338437978560491,
647
- "grad_norm": 0.6668057827943388,
648
- "learning_rate": 1.8350953392368408e-05,
649
- "loss": 0.797,
650
- "step": 435
651
- },
652
- {
653
- "epoch": 1.349157733537519,
654
- "grad_norm": 0.7077024335619939,
655
- "learning_rate": 1.8291573051676063e-05,
656
- "loss": 0.8257,
657
- "step": 440
658
- },
659
- {
660
- "epoch": 1.3644716692189893,
661
- "grad_norm": 0.7168213984013295,
662
- "learning_rate": 1.8231242079248512e-05,
663
- "loss": 0.7934,
664
- "step": 445
665
- },
666
- {
667
- "epoch": 1.3797856049004595,
668
- "grad_norm": 0.7308258264034365,
669
- "learning_rate": 1.816996739205274e-05,
670
- "loss": 0.8078,
671
- "step": 450
672
- },
673
- {
674
- "epoch": 1.3950995405819295,
675
- "grad_norm": 0.7935757091141241,
676
- "learning_rate": 1.810775601525296e-05,
677
- "loss": 0.8172,
678
- "step": 455
679
- },
680
- {
681
- "epoch": 1.4104134762633997,
682
- "grad_norm": 0.7960723044086886,
683
- "learning_rate": 1.8044615081405153e-05,
684
- "loss": 0.7973,
685
- "step": 460
686
- },
687
- {
688
- "epoch": 1.42572741194487,
689
- "grad_norm": 0.7013219126306894,
690
- "learning_rate": 1.7980551829639357e-05,
691
- "loss": 0.7928,
692
- "step": 465
693
- },
694
- {
695
- "epoch": 1.44104134762634,
696
- "grad_norm": 0.7030096980783208,
697
- "learning_rate": 1.7915573604829684e-05,
698
- "loss": 0.7891,
699
- "step": 470
700
- },
701
- {
702
- "epoch": 1.4563552833078102,
703
- "grad_norm": 0.7129952798275541,
704
- "learning_rate": 1.784968785675221e-05,
705
- "loss": 0.7964,
706
- "step": 475
707
- },
708
- {
709
- "epoch": 1.4716692189892804,
710
- "grad_norm": 0.7042351661303197,
711
- "learning_rate": 1.7782902139230876e-05,
712
- "loss": 0.7821,
713
- "step": 480
714
- },
715
- {
716
- "epoch": 1.4869831546707504,
717
- "grad_norm": 0.7412846210916867,
718
- "learning_rate": 1.7715224109271426e-05,
719
- "loss": 0.7983,
720
- "step": 485
721
- },
722
- {
723
- "epoch": 1.5022970903522204,
724
- "grad_norm": 0.7586380261564012,
725
- "learning_rate": 1.764666152618355e-05,
726
- "loss": 0.8167,
727
- "step": 490
728
- },
729
- {
730
- "epoch": 1.5176110260336908,
731
- "grad_norm": 0.784243612804958,
732
- "learning_rate": 1.7577222250691254e-05,
733
- "loss": 0.8086,
734
- "step": 495
735
- },
736
- {
737
- "epoch": 1.5329249617151608,
738
- "grad_norm": 0.6901397300587065,
739
- "learning_rate": 1.7506914244031627e-05,
740
- "loss": 0.7966,
741
- "step": 500
742
- },
743
- {
744
- "epoch": 1.5329249617151608,
745
- "eval_loss": 0.8608717918395996,
746
- "eval_runtime": 4.0604,
747
- "eval_samples_per_second": 30.539,
748
- "eval_steps_per_second": 1.97,
749
- "step": 500
750
- },
751
- {
752
- "epoch": 1.5482388973966308,
753
- "grad_norm": 0.6826627810241662,
754
- "learning_rate": 1.7435745567042096e-05,
755
- "loss": 0.7871,
756
- "step": 505
757
- },
758
- {
759
- "epoch": 1.5635528330781012,
760
- "grad_norm": 0.7329202799692207,
761
- "learning_rate": 1.7363724379236237e-05,
762
- "loss": 0.806,
763
- "step": 510
764
- },
765
- {
766
- "epoch": 1.5788667687595712,
767
- "grad_norm": 0.8605855369920612,
768
- "learning_rate": 1.7290858937868296e-05,
769
- "loss": 0.8134,
770
- "step": 515
771
- },
772
- {
773
- "epoch": 1.5941807044410412,
774
- "grad_norm": 0.6963610579257828,
775
- "learning_rate": 1.7217157596986474e-05,
776
- "loss": 0.8057,
777
- "step": 520
778
- },
779
- {
780
- "epoch": 1.6094946401225116,
781
- "grad_norm": 0.715201778907277,
782
- "learning_rate": 1.7142628806475144e-05,
783
- "loss": 0.806,
784
- "step": 525
785
- },
786
- {
787
- "epoch": 1.6248085758039816,
788
- "grad_norm": 0.7191782284781755,
789
- "learning_rate": 1.706728111108607e-05,
790
- "loss": 0.8027,
791
- "step": 530
792
- },
793
- {
794
- "epoch": 1.6401225114854516,
795
- "grad_norm": 0.6872215034145709,
796
- "learning_rate": 1.699112314945874e-05,
797
- "loss": 0.7953,
798
- "step": 535
799
- },
800
- {
801
- "epoch": 1.655436447166922,
802
- "grad_norm": 0.7008995671254957,
803
- "learning_rate": 1.691416365312996e-05,
804
- "loss": 0.8096,
805
- "step": 540
806
- },
807
- {
808
- "epoch": 1.670750382848392,
809
- "grad_norm": 0.6521306963395784,
810
- "learning_rate": 1.683641144553275e-05,
811
- "loss": 0.7921,
812
- "step": 545
813
- },
814
- {
815
- "epoch": 1.686064318529862,
816
- "grad_norm": 0.6632532080286483,
817
- "learning_rate": 1.675787544098477e-05,
818
- "loss": 0.8056,
819
- "step": 550
820
- },
821
- {
822
- "epoch": 1.7013782542113323,
823
- "grad_norm": 0.6547969582532698,
824
- "learning_rate": 1.667856464366626e-05,
825
- "loss": 0.7819,
826
- "step": 555
827
- },
828
- {
829
- "epoch": 1.7166921898928025,
830
- "grad_norm": 0.7046901763937415,
831
- "learning_rate": 1.6598488146587733e-05,
832
- "loss": 0.8061,
833
- "step": 560
834
- },
835
- {
836
- "epoch": 1.7320061255742725,
837
- "grad_norm": 0.6788820639858653,
838
- "learning_rate": 1.6517655130547435e-05,
839
- "loss": 0.7997,
840
- "step": 565
841
- },
842
- {
843
- "epoch": 1.7473200612557427,
844
- "grad_norm": 0.6894021837769623,
845
- "learning_rate": 1.6436074863078783e-05,
846
- "loss": 0.803,
847
- "step": 570
848
- },
849
- {
850
- "epoch": 1.762633996937213,
851
- "grad_norm": 0.7360294460315284,
852
- "learning_rate": 1.635375669738782e-05,
853
- "loss": 0.7975,
854
- "step": 575
855
- },
856
- {
857
- "epoch": 1.777947932618683,
858
- "grad_norm": 0.6850098881910617,
859
- "learning_rate": 1.627071007128089e-05,
860
- "loss": 0.7825,
861
- "step": 580
862
- },
863
- {
864
- "epoch": 1.793261868300153,
865
- "grad_norm": 0.7278655150611072,
866
- "learning_rate": 1.618694450608256e-05,
867
- "loss": 0.7955,
868
- "step": 585
869
- },
870
- {
871
- "epoch": 1.8085758039816233,
872
- "grad_norm": 0.7122173514963583,
873
- "learning_rate": 1.6102469605544036e-05,
874
- "loss": 0.808,
875
- "step": 590
876
- },
877
- {
878
- "epoch": 1.8238897396630933,
879
- "grad_norm": 0.6925118679066572,
880
- "learning_rate": 1.6017295054742045e-05,
881
- "loss": 0.7875,
882
- "step": 595
883
- },
884
- {
885
- "epoch": 1.8392036753445635,
886
- "grad_norm": 0.6857769438421079,
887
- "learning_rate": 1.5931430618968476e-05,
888
- "loss": 0.7884,
889
- "step": 600
890
- },
891
- {
892
- "epoch": 1.8392036753445635,
893
- "eval_loss": 0.8516013622283936,
894
- "eval_runtime": 4.2451,
895
- "eval_samples_per_second": 29.21,
896
- "eval_steps_per_second": 1.885,
897
- "step": 600
898
- },
899
- {
900
- "epoch": 1.8545176110260337,
901
- "grad_norm": 0.6326947636704211,
902
- "learning_rate": 1.5844886142610763e-05,
903
- "loss": 0.7819,
904
- "step": 605
905
- },
906
- {
907
- "epoch": 1.8698315467075037,
908
- "grad_norm": 0.666694114425699,
909
- "learning_rate": 1.575767154802323e-05,
910
- "loss": 0.7777,
911
- "step": 610
912
- },
913
- {
914
- "epoch": 1.885145482388974,
915
- "grad_norm": 0.6182711467351976,
916
- "learning_rate": 1.5669796834389496e-05,
917
- "loss": 0.7889,
918
- "step": 615
919
- },
920
- {
921
- "epoch": 1.9004594180704442,
922
- "grad_norm": 0.6402744830423595,
923
- "learning_rate": 1.5581272076576047e-05,
924
- "loss": 0.7839,
925
- "step": 620
926
- },
927
- {
928
- "epoch": 1.9157733537519142,
929
- "grad_norm": 0.6612456968952057,
930
- "learning_rate": 1.5492107423977167e-05,
931
- "loss": 0.7884,
932
- "step": 625
933
- },
934
- {
935
- "epoch": 1.9310872894333844,
936
- "grad_norm": 0.6757984187665057,
937
- "learning_rate": 1.5402313099351302e-05,
938
- "loss": 0.7784,
939
- "step": 630
940
- },
941
- {
942
- "epoch": 1.9464012251148546,
943
- "grad_norm": 0.6707442527908725,
944
- "learning_rate": 1.5311899397649e-05,
945
- "loss": 0.7935,
946
- "step": 635
947
- },
948
- {
949
- "epoch": 1.9617151607963246,
950
- "grad_norm": 0.787792189502549,
951
- "learning_rate": 1.522087668483264e-05,
952
- "loss": 0.779,
953
- "step": 640
954
- },
955
- {
956
- "epoch": 1.9770290964777948,
957
- "grad_norm": 0.6925473058613637,
958
- "learning_rate": 1.5129255396687899e-05,
959
- "loss": 0.7891,
960
- "step": 645
961
- },
962
- {
963
- "epoch": 1.992343032159265,
964
- "grad_norm": 0.6890667970589922,
965
- "learning_rate": 1.503704603762734e-05,
966
- "loss": 0.7693,
967
- "step": 650
968
- },
969
- {
970
- "epoch": 2.009188361408882,
971
- "grad_norm": 0.8106643045129424,
972
- "learning_rate": 1.4944259179486068e-05,
973
- "loss": 0.9039,
974
- "step": 655
975
- },
976
- {
977
- "epoch": 2.0245022970903523,
978
- "grad_norm": 0.7728005094053498,
979
- "learning_rate": 1.4850905460309648e-05,
980
- "loss": 0.7116,
981
- "step": 660
982
- },
983
- {
984
- "epoch": 2.0398162327718223,
985
- "grad_norm": 0.7301328106464207,
986
- "learning_rate": 1.4756995583134463e-05,
987
- "loss": 0.7313,
988
- "step": 665
989
- },
990
- {
991
- "epoch": 2.0551301684532923,
992
- "grad_norm": 0.702048383564715,
993
- "learning_rate": 1.4662540314760608e-05,
994
- "loss": 0.7222,
995
- "step": 670
996
- },
997
- {
998
- "epoch": 2.0704441041347628,
999
- "grad_norm": 0.7081084536553046,
1000
- "learning_rate": 1.4567550484517456e-05,
1001
- "loss": 0.7214,
1002
- "step": 675
1003
- },
1004
- {
1005
- "epoch": 2.0857580398162328,
1006
- "grad_norm": 0.7492389283409402,
1007
- "learning_rate": 1.4472036983022106e-05,
1008
- "loss": 0.7251,
1009
- "step": 680
1010
- },
1011
- {
1012
- "epoch": 2.1010719754977027,
1013
- "grad_norm": 0.7183839129962212,
1014
- "learning_rate": 1.437601076093073e-05,
1015
- "loss": 0.71,
1016
- "step": 685
1017
- },
1018
- {
1019
- "epoch": 2.116385911179173,
1020
- "grad_norm": 0.6904647110507196,
1021
- "learning_rate": 1.4279482827683095e-05,
1022
- "loss": 0.724,
1023
- "step": 690
1024
- },
1025
- {
1026
- "epoch": 2.131699846860643,
1027
- "grad_norm": 0.6745352346306862,
1028
- "learning_rate": 1.4182464250240341e-05,
1029
- "loss": 0.7127,
1030
- "step": 695
1031
- },
1032
- {
1033
- "epoch": 2.147013782542113,
1034
- "grad_norm": 0.7254396412798532,
1035
- "learning_rate": 1.4084966151816124e-05,
1036
- "loss": 0.7071,
1037
- "step": 700
1038
- },
1039
- {
1040
- "epoch": 2.147013782542113,
1041
- "eval_loss": 0.8550077676773071,
1042
- "eval_runtime": 3.8848,
1043
- "eval_samples_per_second": 31.919,
1044
- "eval_steps_per_second": 2.059,
1045
- "step": 700
1046
- },
1047
- {
1048
- "epoch": 2.1623277182235836,
1049
- "grad_norm": 0.73349139216254,
1050
- "learning_rate": 1.3986999710601348e-05,
1051
- "loss": 0.7298,
1052
- "step": 705
1053
- },
1054
- {
1055
- "epoch": 2.1776416539050536,
1056
- "grad_norm": 0.6655753573647281,
1057
- "learning_rate": 1.3888576158482586e-05,
1058
- "loss": 0.7124,
1059
- "step": 710
1060
- },
1061
- {
1062
- "epoch": 2.1929555895865236,
1063
- "grad_norm": 0.6987464871327375,
1064
- "learning_rate": 1.3789706779754326e-05,
1065
- "loss": 0.712,
1066
- "step": 715
1067
- },
1068
- {
1069
- "epoch": 2.208269525267994,
1070
- "grad_norm": 0.7104463485359258,
1071
- "learning_rate": 1.3690402909825245e-05,
1072
- "loss": 0.7256,
1073
- "step": 720
1074
- },
1075
- {
1076
- "epoch": 2.223583460949464,
1077
- "grad_norm": 0.786597658435966,
1078
- "learning_rate": 1.3590675933918578e-05,
1079
- "loss": 0.7265,
1080
- "step": 725
1081
- },
1082
- {
1083
- "epoch": 2.238897396630934,
1084
- "grad_norm": 0.7359585689347169,
1085
- "learning_rate": 1.3490537285766809e-05,
1086
- "loss": 0.7167,
1087
- "step": 730
1088
- },
1089
- {
1090
- "epoch": 2.2542113323124044,
1091
- "grad_norm": 0.6905264840508457,
1092
- "learning_rate": 1.3389998446300791e-05,
1093
- "loss": 0.7046,
1094
- "step": 735
1095
- },
1096
- {
1097
- "epoch": 2.2695252679938744,
1098
- "grad_norm": 0.7110824744193723,
1099
- "learning_rate": 1.3289070942333448e-05,
1100
- "loss": 0.7113,
1101
- "step": 740
1102
- },
1103
- {
1104
- "epoch": 2.2848392036753444,
1105
- "grad_norm": 0.774570053362907,
1106
- "learning_rate": 1.3187766345238222e-05,
1107
- "loss": 0.7097,
1108
- "step": 745
1109
- },
1110
- {
1111
- "epoch": 2.300153139356815,
1112
- "grad_norm": 0.7178949231829396,
1113
- "learning_rate": 1.308609626962242e-05,
1114
- "loss": 0.6951,
1115
- "step": 750
1116
- },
1117
- {
1118
- "epoch": 2.315467075038285,
1119
- "grad_norm": 0.6778307950742112,
1120
- "learning_rate": 1.2984072371995581e-05,
1121
- "loss": 0.7119,
1122
- "step": 755
1123
- },
1124
- {
1125
- "epoch": 2.330781010719755,
1126
- "grad_norm": 0.7381317901698428,
1127
- "learning_rate": 1.288170634943307e-05,
1128
- "loss": 0.7179,
1129
- "step": 760
1130
- },
1131
- {
1132
- "epoch": 2.3460949464012253,
1133
- "grad_norm": 0.688339196574423,
1134
- "learning_rate": 1.2779009938234986e-05,
1135
- "loss": 0.7372,
1136
- "step": 765
1137
- },
1138
- {
1139
- "epoch": 2.3614088820826953,
1140
- "grad_norm": 0.7309911889901869,
1141
- "learning_rate": 1.2675994912580601e-05,
1142
- "loss": 0.7217,
1143
- "step": 770
1144
- },
1145
- {
1146
- "epoch": 2.3767228177641653,
1147
- "grad_norm": 2.627453466768546,
1148
- "learning_rate": 1.2572673083178448e-05,
1149
- "loss": 0.7208,
1150
- "step": 775
1151
- },
1152
- {
1153
- "epoch": 2.3920367534456357,
1154
- "grad_norm": 0.6825595176377366,
1155
- "learning_rate": 1.2469056295912216e-05,
1156
- "loss": 0.7173,
1157
- "step": 780
1158
- },
1159
- {
1160
- "epoch": 2.4073506891271057,
1161
- "grad_norm": 0.681764187496552,
1162
- "learning_rate": 1.2365156430482621e-05,
1163
- "loss": 0.7172,
1164
- "step": 785
1165
- },
1166
- {
1167
- "epoch": 2.4226646248085757,
1168
- "grad_norm": 0.6957949104606218,
1169
- "learning_rate": 1.2260985399045379e-05,
1170
- "loss": 0.7144,
1171
- "step": 790
1172
- },
1173
- {
1174
- "epoch": 2.437978560490046,
1175
- "grad_norm": 0.6810069800053375,
1176
- "learning_rate": 1.2156555144845489e-05,
1177
- "loss": 0.7153,
1178
- "step": 795
1179
- },
1180
- {
1181
- "epoch": 2.453292496171516,
1182
- "grad_norm": 0.6879227501131829,
1183
- "learning_rate": 1.2051877640847929e-05,
1184
- "loss": 0.7208,
1185
- "step": 800
1186
- },
1187
- {
1188
- "epoch": 2.453292496171516,
1189
- "eval_loss": 0.8501381278038025,
1190
- "eval_runtime": 3.7489,
1191
- "eval_samples_per_second": 33.076,
1192
- "eval_steps_per_second": 2.134,
1193
- "step": 800
1194
- },
1195
- {
1196
- "epoch": 2.468606431852986,
1197
- "grad_norm": 0.7074177591745171,
1198
- "learning_rate": 1.1946964888364949e-05,
1199
- "loss": 0.7293,
1200
- "step": 805
1201
- },
1202
- {
1203
- "epoch": 2.4839203675344566,
1204
- "grad_norm": 0.7100957198483567,
1205
- "learning_rate": 1.1841828915680127e-05,
1206
- "loss": 0.7088,
1207
- "step": 810
1208
- },
1209
- {
1210
- "epoch": 2.4992343032159265,
1211
- "grad_norm": 0.7065376294005609,
1212
- "learning_rate": 1.1736481776669307e-05,
1213
- "loss": 0.7083,
1214
- "step": 815
1215
- },
1216
- {
1217
- "epoch": 2.5145482388973965,
1218
- "grad_norm": 0.699699294689246,
1219
- "learning_rate": 1.1630935549418627e-05,
1220
- "loss": 0.7239,
1221
- "step": 820
1222
- },
1223
- {
1224
- "epoch": 2.5298621745788665,
1225
- "grad_norm": 0.6931742613638052,
1226
- "learning_rate": 1.1525202334839771e-05,
1227
- "loss": 0.7257,
1228
- "step": 825
1229
- },
1230
- {
1231
- "epoch": 2.545176110260337,
1232
- "grad_norm": 0.6626821835687099,
1233
- "learning_rate": 1.1419294255282574e-05,
1234
- "loss": 0.7158,
1235
- "step": 830
1236
- },
1237
- {
1238
- "epoch": 2.560490045941807,
1239
- "grad_norm": 0.7476424454423357,
1240
- "learning_rate": 1.1313223453145202e-05,
1241
- "loss": 0.7231,
1242
- "step": 835
1243
- },
1244
- {
1245
- "epoch": 2.5758039816232774,
1246
- "grad_norm": 0.7121987313639168,
1247
- "learning_rate": 1.1207002089482026e-05,
1248
- "loss": 0.7272,
1249
- "step": 840
1250
- },
1251
- {
1252
- "epoch": 2.5911179173047474,
1253
- "grad_norm": 0.6842806839970864,
1254
- "learning_rate": 1.1100642342609352e-05,
1255
- "loss": 0.7215,
1256
- "step": 845
1257
- },
1258
- {
1259
- "epoch": 2.6064318529862174,
1260
- "grad_norm": 0.671282856198737,
1261
- "learning_rate": 1.0994156406709155e-05,
1262
- "loss": 0.7086,
1263
- "step": 850
1264
- },
1265
- {
1266
- "epoch": 2.6217457886676874,
1267
- "grad_norm": 0.7127796423271462,
1268
- "learning_rate": 1.088755649043104e-05,
1269
- "loss": 0.7113,
1270
- "step": 855
1271
- },
1272
- {
1273
- "epoch": 2.637059724349158,
1274
- "grad_norm": 0.6642491628332302,
1275
- "learning_rate": 1.0780854815492496e-05,
1276
- "loss": 0.7247,
1277
- "step": 860
1278
- },
1279
- {
1280
- "epoch": 2.652373660030628,
1281
- "grad_norm": 0.6698942933482407,
1282
- "learning_rate": 1.0674063615277681e-05,
1283
- "loss": 0.7301,
1284
- "step": 865
1285
- },
1286
- {
1287
- "epoch": 2.6676875957120982,
1288
- "grad_norm": 0.6690575090698347,
1289
- "learning_rate": 1.0567195133434851e-05,
1290
- "loss": 0.7318,
1291
- "step": 870
1292
- },
1293
- {
1294
- "epoch": 2.6830015313935682,
1295
- "grad_norm": 0.6647266330654462,
1296
- "learning_rate": 1.0460261622472631e-05,
1297
- "loss": 0.7143,
1298
- "step": 875
1299
- },
1300
- {
1301
- "epoch": 2.698315467075038,
1302
- "grad_norm": 0.6879144621587469,
1303
- "learning_rate": 1.0353275342355262e-05,
1304
- "loss": 0.7227,
1305
- "step": 880
1306
- },
1307
- {
1308
- "epoch": 2.713629402756508,
1309
- "grad_norm": 0.6775051673571305,
1310
- "learning_rate": 1.024624855909698e-05,
1311
- "loss": 0.715,
1312
- "step": 885
1313
- },
1314
- {
1315
- "epoch": 2.7289433384379786,
1316
- "grad_norm": 0.6785052740842338,
1317
- "learning_rate": 1.013919354335572e-05,
1318
- "loss": 0.7224,
1319
- "step": 890
1320
- },
1321
- {
1322
- "epoch": 2.7442572741194486,
1323
- "grad_norm": 0.6852717568024729,
1324
- "learning_rate": 1.0032122569026284e-05,
1325
- "loss": 0.7181,
1326
- "step": 895
1327
- },
1328
- {
1329
- "epoch": 2.759571209800919,
1330
- "grad_norm": 0.6966758449578118,
1331
- "learning_rate": 9.925047911833137e-06,
1332
- "loss": 0.726,
1333
- "step": 900
1334
- },
1335
- {
1336
- "epoch": 2.759571209800919,
1337
- "eval_loss": 0.8449206352233887,
1338
- "eval_runtime": 3.7457,
1339
- "eval_samples_per_second": 33.104,
1340
- "eval_steps_per_second": 2.136,
1341
- "step": 900
1342
- }
1343
- ],
1344
- "logging_steps": 5,
1345
- "max_steps": 1630,
1346
- "num_input_tokens_seen": 0,
1347
- "num_train_epochs": 5,
1348
- "save_steps": 100,
1349
- "stateful_callbacks": {
1350
- "TrainerControl": {
1351
- "args": {
1352
- "should_epoch_stop": false,
1353
- "should_evaluate": false,
1354
- "should_log": false,
1355
- "should_save": true,
1356
- "should_training_stop": false
1357
- },
1358
- "attributes": {}
1359
- }
1360
- },
1361
- "total_flos": 95696971628544.0,
1362
- "train_batch_size": 4,
1363
- "trial_name": null,
1364
- "trial_params": null
1365
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
checkpoints/checkpoint-900/training_args.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:8128bba117ecebe610aa8a93374390d0c029e4632d49ee23fa1bd8e6b39d6679
3
- size 7352
 
 
 
 
checkpoints/checkpoint-900/zero_to_fp32.py DELETED
@@ -1,674 +0,0 @@
1
- #!/usr/bin/env python
2
-
3
- # Copyright (c) Microsoft Corporation.
4
- # SPDX-License-Identifier: Apache-2.0
5
-
6
- # DeepSpeed Team
7
-
8
- # This script extracts fp32 consolidated weights from a zero 1, 2 and 3 DeepSpeed checkpoints. It gets
9
- # copied into the top level checkpoint dir, so the user can easily do the conversion at any point in
10
- # the future. Once extracted, the weights don't require DeepSpeed and can be used in any
11
- # application.
12
- #
13
- # example:
14
- # python zero_to_fp32.py . output_dir/
15
- # or
16
- # python zero_to_fp32.py . output_dir/ --safe_serialization
17
-
18
- import argparse
19
- import torch
20
- import glob
21
- import math
22
- import os
23
- import re
24
- import json
25
- from tqdm import tqdm
26
- from collections import OrderedDict
27
- from dataclasses import dataclass
28
-
29
- # while this script doesn't use deepspeed to recover data, since the checkpoints are pickled with
30
- # DeepSpeed data structures it has to be available in the current python environment.
31
- from deepspeed.utils import logger
32
- from deepspeed.checkpoint.constants import (DS_VERSION, OPTIMIZER_STATE_DICT, SINGLE_PARTITION_OF_FP32_GROUPS,
33
- FP32_FLAT_GROUPS, ZERO_STAGE, PARTITION_COUNT, PARAM_SHAPES, BUFFER_NAMES,
34
- FROZEN_PARAM_SHAPES, FROZEN_PARAM_FRAGMENTS)
35
-
36
-
37
- @dataclass
38
- class zero_model_state:
39
- buffers: dict()
40
- param_shapes: dict()
41
- shared_params: list
42
- ds_version: int
43
- frozen_param_shapes: dict()
44
- frozen_param_fragments: dict()
45
-
46
-
47
- debug = 0
48
-
49
- # load to cpu
50
- device = torch.device('cpu')
51
-
52
-
53
- def atoi(text):
54
- return int(text) if text.isdigit() else text
55
-
56
-
57
- def natural_keys(text):
58
- '''
59
- alist.sort(key=natural_keys) sorts in human order
60
- http://nedbatchelder.com/blog/200712/human_sorting.html
61
- (See Toothy's implementation in the comments)
62
- '''
63
- return [atoi(c) for c in re.split(r'(\d+)', text)]
64
-
65
-
66
- def get_model_state_file(checkpoint_dir, zero_stage):
67
- if not os.path.isdir(checkpoint_dir):
68
- raise FileNotFoundError(f"Directory '{checkpoint_dir}' doesn't exist")
69
-
70
- # there should be only one file
71
- if zero_stage <= 2:
72
- file = os.path.join(checkpoint_dir, "mp_rank_00_model_states.pt")
73
- elif zero_stage == 3:
74
- file = os.path.join(checkpoint_dir, "zero_pp_rank_0_mp_rank_00_model_states.pt")
75
-
76
- if not os.path.exists(file):
77
- raise FileNotFoundError(f"can't find model states file at '{file}'")
78
-
79
- return file
80
-
81
-
82
- def get_checkpoint_files(checkpoint_dir, glob_pattern):
83
- # XXX: need to test that this simple glob rule works for multi-node setup too
84
- ckpt_files = sorted(glob.glob(os.path.join(checkpoint_dir, glob_pattern)), key=natural_keys)
85
-
86
- if len(ckpt_files) == 0:
87
- raise FileNotFoundError(f"can't find {glob_pattern} files in directory '{checkpoint_dir}'")
88
-
89
- return ckpt_files
90
-
91
-
92
- def get_optim_files(checkpoint_dir):
93
- return get_checkpoint_files(checkpoint_dir, "*_optim_states.pt")
94
-
95
-
96
- def get_model_state_files(checkpoint_dir):
97
- return get_checkpoint_files(checkpoint_dir, "*_model_states.pt")
98
-
99
-
100
- def parse_model_states(files):
101
- zero_model_states = []
102
- for file in files:
103
- state_dict = torch.load(file, map_location=device)
104
-
105
- if BUFFER_NAMES not in state_dict:
106
- raise ValueError(f"{file} is not a model state checkpoint")
107
- buffer_names = state_dict[BUFFER_NAMES]
108
- if debug:
109
- print("Found buffers:", buffer_names)
110
-
111
- # recover just the buffers while restoring them to fp32 if they were saved in fp16
112
- buffers = {k: v.float() for k, v in state_dict["module"].items() if k in buffer_names}
113
- param_shapes = state_dict[PARAM_SHAPES]
114
-
115
- # collect parameters that are included in param_shapes
116
- param_names = []
117
- for s in param_shapes:
118
- for name in s.keys():
119
- param_names.append(name)
120
-
121
- # update with frozen parameters
122
- frozen_param_shapes = state_dict.get(FROZEN_PARAM_SHAPES, None)
123
- if frozen_param_shapes is not None:
124
- if debug:
125
- print(f"Found frozen_param_shapes: {frozen_param_shapes}")
126
- param_names += list(frozen_param_shapes.keys())
127
-
128
- # handle shared params
129
- shared_params = [[k, v] for k, v in state_dict["shared_params"].items()]
130
-
131
- ds_version = state_dict.get(DS_VERSION, None)
132
-
133
- frozen_param_fragments = state_dict.get(FROZEN_PARAM_FRAGMENTS, None)
134
-
135
- z_model_state = zero_model_state(buffers=buffers,
136
- param_shapes=param_shapes,
137
- shared_params=shared_params,
138
- ds_version=ds_version,
139
- frozen_param_shapes=frozen_param_shapes,
140
- frozen_param_fragments=frozen_param_fragments)
141
- zero_model_states.append(z_model_state)
142
-
143
- return zero_model_states
144
-
145
-
146
- def parse_optim_states(files, ds_checkpoint_dir):
147
- total_files = len(files)
148
- state_dicts = []
149
- for f in files:
150
- state_dict = torch.load(f, map_location=device)
151
- # immediately discard the potentially huge 2 optimizer states as we only care for fp32 master weights
152
- # and also handle the case where it was already removed by another helper script
153
- state_dict["optimizer_state_dict"].pop("optimizer_state_dict", None)
154
- state_dicts.append(state_dict)
155
-
156
- if not ZERO_STAGE in state_dicts[0][OPTIMIZER_STATE_DICT]:
157
- raise ValueError(f"{files[0]} is not a zero checkpoint")
158
- zero_stage = state_dicts[0][OPTIMIZER_STATE_DICT][ZERO_STAGE]
159
- world_size = state_dicts[0][OPTIMIZER_STATE_DICT][PARTITION_COUNT]
160
-
161
- # For ZeRO-2 each param group can have different partition_count as data parallelism for expert
162
- # parameters can be different from data parallelism for non-expert parameters. So we can just
163
- # use the max of the partition_count to get the dp world_size.
164
-
165
- if type(world_size) is list:
166
- world_size = max(world_size)
167
-
168
- if world_size != total_files:
169
- raise ValueError(
170
- f"Expected {world_size} of '*_optim_states.pt' under '{ds_checkpoint_dir}' but found {total_files} files. "
171
- "Possibly due to an overwrite of an old checkpoint, or a checkpoint didn't get saved by one or more processes."
172
- )
173
-
174
- # the groups are named differently in each stage
175
- if zero_stage <= 2:
176
- fp32_groups_key = SINGLE_PARTITION_OF_FP32_GROUPS
177
- elif zero_stage == 3:
178
- fp32_groups_key = FP32_FLAT_GROUPS
179
- else:
180
- raise ValueError(f"unknown zero stage {zero_stage}")
181
-
182
- if zero_stage <= 2:
183
- fp32_flat_groups = [state_dicts[i][OPTIMIZER_STATE_DICT][fp32_groups_key] for i in range(len(state_dicts))]
184
- elif zero_stage == 3:
185
- # if there is more than one param group, there will be multiple flattened tensors - one
186
- # flattened tensor per group - for simplicity merge them into a single tensor
187
- #
188
- # XXX: could make the script more memory efficient for when there are multiple groups - it
189
- # will require matching the sub-lists of param_shapes for each param group flattened tensor
190
-
191
- fp32_flat_groups = [
192
- torch.cat(state_dicts[i][OPTIMIZER_STATE_DICT][fp32_groups_key], 0) for i in range(len(state_dicts))
193
- ]
194
-
195
- return zero_stage, world_size, fp32_flat_groups
196
-
197
-
198
- def _get_fp32_state_dict_from_zero_checkpoint(ds_checkpoint_dir, exclude_frozen_parameters):
199
- """
200
- Returns fp32 state_dict reconstructed from ds checkpoint
201
-
202
- Args:
203
- - ``ds_checkpoint_dir``: path to the deepspeed checkpoint folder (where the optimizer files are)
204
-
205
- """
206
- print(f"Processing zero checkpoint '{ds_checkpoint_dir}'")
207
-
208
- optim_files = get_optim_files(ds_checkpoint_dir)
209
- zero_stage, world_size, fp32_flat_groups = parse_optim_states(optim_files, ds_checkpoint_dir)
210
- print(f"Detected checkpoint of type zero stage {zero_stage}, world_size: {world_size}")
211
-
212
- model_files = get_model_state_files(ds_checkpoint_dir)
213
-
214
- zero_model_states = parse_model_states(model_files)
215
- print(f'Parsing checkpoint created by deepspeed=={zero_model_states[0].ds_version}')
216
-
217
- if zero_stage <= 2:
218
- return _get_fp32_state_dict_from_zero2_checkpoint(world_size, fp32_flat_groups, zero_model_states,
219
- exclude_frozen_parameters)
220
- elif zero_stage == 3:
221
- return _get_fp32_state_dict_from_zero3_checkpoint(world_size, fp32_flat_groups, zero_model_states,
222
- exclude_frozen_parameters)
223
-
224
-
225
- def _zero2_merge_frozen_params(state_dict, zero_model_states):
226
- if zero_model_states[0].frozen_param_shapes is None or len(zero_model_states[0].frozen_param_shapes) == 0:
227
- return
228
-
229
- frozen_param_shapes = zero_model_states[0].frozen_param_shapes
230
- frozen_param_fragments = zero_model_states[0].frozen_param_fragments
231
-
232
- if debug:
233
- num_elem = sum(s.numel() for s in frozen_param_shapes.values())
234
- print(f'rank 0: {FROZEN_PARAM_SHAPES}.numel = {num_elem}')
235
-
236
- wanted_params = len(frozen_param_shapes)
237
- wanted_numel = sum(s.numel() for s in frozen_param_shapes.values())
238
- avail_numel = sum([p.numel() for p in frozen_param_fragments.values()])
239
- print(f'Frozen params: Have {avail_numel} numels to process.')
240
- print(f'Frozen params: Need {wanted_numel} numels in {wanted_params} params')
241
-
242
- total_params = 0
243
- total_numel = 0
244
- for name, shape in frozen_param_shapes.items():
245
- total_params += 1
246
- unpartitioned_numel = shape.numel()
247
- total_numel += unpartitioned_numel
248
-
249
- state_dict[name] = frozen_param_fragments[name]
250
-
251
- if debug:
252
- print(f"{name} full shape: {shape} unpartitioned numel {unpartitioned_numel} ")
253
-
254
- print(f"Reconstructed Frozen fp32 state dict with {total_params} params {total_numel} elements")
255
-
256
-
257
- def _has_callable(obj, fn):
258
- attr = getattr(obj, fn, None)
259
- return callable(attr)
260
-
261
-
262
- def _zero2_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states):
263
- param_shapes = zero_model_states[0].param_shapes
264
-
265
- # Reconstruction protocol:
266
- #
267
- # XXX: document this
268
-
269
- if debug:
270
- for i in range(world_size):
271
- for j in range(len(fp32_flat_groups[0])):
272
- print(f"{FP32_FLAT_GROUPS}[{i}][{j}].shape={fp32_flat_groups[i][j].shape}")
273
-
274
- # XXX: memory usage doubles here (zero2)
275
- num_param_groups = len(fp32_flat_groups[0])
276
- merged_single_partition_of_fp32_groups = []
277
- for i in range(num_param_groups):
278
- merged_partitions = [sd[i] for sd in fp32_flat_groups]
279
- full_single_fp32_vector = torch.cat(merged_partitions, 0)
280
- merged_single_partition_of_fp32_groups.append(full_single_fp32_vector)
281
- avail_numel = sum(
282
- [full_single_fp32_vector.numel() for full_single_fp32_vector in merged_single_partition_of_fp32_groups])
283
-
284
- if debug:
285
- wanted_params = sum([len(shapes) for shapes in param_shapes])
286
- wanted_numel = sum([sum(shape.numel() for shape in shapes.values()) for shapes in param_shapes])
287
- # not asserting if there is a mismatch due to possible padding
288
- print(f"Have {avail_numel} numels to process.")
289
- print(f"Need {wanted_numel} numels in {wanted_params} params.")
290
-
291
- # params
292
- # XXX: for huge models that can't fit into the host's RAM we will have to recode this to support
293
- # out-of-core computing solution
294
- total_numel = 0
295
- total_params = 0
296
- for shapes, full_single_fp32_vector in zip(param_shapes, merged_single_partition_of_fp32_groups):
297
- offset = 0
298
- avail_numel = full_single_fp32_vector.numel()
299
- for name, shape in shapes.items():
300
-
301
- unpartitioned_numel = shape.numel() if _has_callable(shape, 'numel') else math.prod(shape)
302
- total_numel += unpartitioned_numel
303
- total_params += 1
304
-
305
- if debug:
306
- print(f"{name} full shape: {shape} unpartitioned numel {unpartitioned_numel} ")
307
- state_dict[name] = full_single_fp32_vector.narrow(0, offset, unpartitioned_numel).view(shape)
308
- offset += unpartitioned_numel
309
-
310
- # Z2 started to align to 2*world_size to improve nccl performance. Therefore both offset and
311
- # avail_numel can differ by anywhere between 0..2*world_size. Due to two unrelated complex
312
- # paddings performed in the code it's almost impossible to predict the exact numbers w/o the
313
- # live optimizer object, so we are checking that the numbers are within the right range
314
- align_to = 2 * world_size
315
-
316
- def zero2_align(x):
317
- return align_to * math.ceil(x / align_to)
318
-
319
- if debug:
320
- print(f"original offset={offset}, avail_numel={avail_numel}")
321
-
322
- offset = zero2_align(offset)
323
- avail_numel = zero2_align(avail_numel)
324
-
325
- if debug:
326
- print(f"aligned offset={offset}, avail_numel={avail_numel}")
327
-
328
- # Sanity check
329
- if offset != avail_numel:
330
- raise ValueError(f"consumed {offset} numels out of {avail_numel} - something is wrong")
331
-
332
- print(f"Reconstructed fp32 state dict with {total_params} params {total_numel} elements")
333
-
334
-
335
- def _get_fp32_state_dict_from_zero2_checkpoint(world_size, fp32_flat_groups, zero_model_states,
336
- exclude_frozen_parameters):
337
- state_dict = OrderedDict()
338
-
339
- # buffers
340
- buffers = zero_model_states[0].buffers
341
- state_dict.update(buffers)
342
- if debug:
343
- print(f"added {len(buffers)} buffers")
344
-
345
- if not exclude_frozen_parameters:
346
- _zero2_merge_frozen_params(state_dict, zero_model_states)
347
-
348
- _zero2_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states)
349
-
350
- # recover shared parameters
351
- for pair in zero_model_states[0].shared_params:
352
- if pair[1] in state_dict:
353
- state_dict[pair[0]] = state_dict[pair[1]]
354
-
355
- return state_dict
356
-
357
-
358
- def zero3_partitioned_param_info(unpartitioned_numel, world_size):
359
- remainder = unpartitioned_numel % world_size
360
- padding_numel = (world_size - remainder) if remainder else 0
361
- partitioned_numel = math.ceil(unpartitioned_numel / world_size)
362
- return partitioned_numel, padding_numel
363
-
364
-
365
- def _zero3_merge_frozen_params(state_dict, world_size, zero_model_states):
366
- if zero_model_states[0].frozen_param_shapes is None or len(zero_model_states[0].frozen_param_shapes) == 0:
367
- return
368
-
369
- if debug:
370
- for i in range(world_size):
371
- num_elem = sum(s.numel() for s in zero_model_states[i].frozen_param_fragments.values())
372
- print(f'rank {i}: {FROZEN_PARAM_SHAPES}.numel = {num_elem}')
373
-
374
- frozen_param_shapes = zero_model_states[0].frozen_param_shapes
375
- wanted_params = len(frozen_param_shapes)
376
- wanted_numel = sum(s.numel() for s in frozen_param_shapes.values())
377
- avail_numel = sum([p.numel() for p in zero_model_states[0].frozen_param_fragments.values()]) * world_size
378
- print(f'Frozen params: Have {avail_numel} numels to process.')
379
- print(f'Frozen params: Need {wanted_numel} numels in {wanted_params} params')
380
-
381
- total_params = 0
382
- total_numel = 0
383
- for name, shape in zero_model_states[0].frozen_param_shapes.items():
384
- total_params += 1
385
- unpartitioned_numel = shape.numel()
386
- total_numel += unpartitioned_numel
387
-
388
- param_frags = tuple(model_state.frozen_param_fragments[name] for model_state in zero_model_states)
389
- state_dict[name] = torch.cat(param_frags, 0).narrow(0, 0, unpartitioned_numel).view(shape)
390
-
391
- partitioned_numel, partitioned_padding_numel = zero3_partitioned_param_info(unpartitioned_numel, world_size)
392
-
393
- if debug:
394
- print(
395
- f"Frozen params: {total_params} {name} full shape: {shape} partition0 numel={partitioned_numel} partitioned_padding_numel={partitioned_padding_numel}"
396
- )
397
-
398
- print(f"Reconstructed Frozen fp32 state dict with {total_params} params {total_numel} elements")
399
-
400
-
401
- def _zero3_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states):
402
- param_shapes = zero_model_states[0].param_shapes
403
- avail_numel = fp32_flat_groups[0].numel() * world_size
404
- # Reconstruction protocol: For zero3 we need to zip the partitions together at boundary of each
405
- # param, re-consolidating each param, while dealing with padding if any
406
-
407
- # merge list of dicts, preserving order
408
- param_shapes = {k: v for d in param_shapes for k, v in d.items()}
409
-
410
- if debug:
411
- for i in range(world_size):
412
- print(f"{FP32_FLAT_GROUPS}[{i}].shape={fp32_flat_groups[i].shape}")
413
-
414
- wanted_params = len(param_shapes)
415
- wanted_numel = sum(shape.numel() for shape in param_shapes.values())
416
- # not asserting if there is a mismatch due to possible padding
417
- avail_numel = fp32_flat_groups[0].numel() * world_size
418
- print(f"Trainable params: Have {avail_numel} numels to process.")
419
- print(f"Trainable params: Need {wanted_numel} numels in {wanted_params} params.")
420
-
421
- # params
422
- # XXX: for huge models that can't fit into the host's RAM we will have to recode this to support
423
- # out-of-core computing solution
424
- offset = 0
425
- total_numel = 0
426
- total_params = 0
427
- for name, shape in tqdm(param_shapes.items(), desc='Gathering Sharded Weights'):
428
- unpartitioned_numel = shape.numel()
429
- total_numel += unpartitioned_numel
430
- total_params += 1
431
- partitioned_numel, partitioned_padding_numel = zero3_partitioned_param_info(unpartitioned_numel, world_size)
432
-
433
- if debug:
434
- print(
435
- f"Trainable params: {total_params} {name} full shape: {shape} partition0 numel={partitioned_numel} partitioned_padding_numel={partitioned_padding_numel}"
436
- )
437
-
438
- # XXX: memory usage doubles here
439
- state_dict[name] = torch.cat(
440
- tuple(fp32_flat_groups[i].narrow(0, offset, partitioned_numel) for i in range(world_size)),
441
- 0).narrow(0, 0, unpartitioned_numel).view(shape)
442
- offset += partitioned_numel
443
-
444
- offset *= world_size
445
-
446
- # Sanity check
447
- if offset != avail_numel:
448
- raise ValueError(f"consumed {offset} numels out of {avail_numel} - something is wrong")
449
-
450
- print(f"Reconstructed Trainable fp32 state dict with {total_params} params {total_numel} elements")
451
-
452
-
453
- def _get_fp32_state_dict_from_zero3_checkpoint(world_size, fp32_flat_groups, zero_model_states,
454
- exclude_frozen_parameters):
455
- state_dict = OrderedDict()
456
-
457
- # buffers
458
- buffers = zero_model_states[0].buffers
459
- state_dict.update(buffers)
460
- if debug:
461
- print(f"added {len(buffers)} buffers")
462
-
463
- if not exclude_frozen_parameters:
464
- _zero3_merge_frozen_params(state_dict, world_size, zero_model_states)
465
-
466
- _zero3_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states)
467
-
468
- # recover shared parameters
469
- for pair in zero_model_states[0].shared_params:
470
- if pair[1] in state_dict:
471
- state_dict[pair[0]] = state_dict[pair[1]]
472
-
473
- return state_dict
474
-
475
-
476
- def get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir, tag=None, exclude_frozen_parameters=False):
477
- """
478
- Convert ZeRO 2 or 3 checkpoint into a single fp32 consolidated state_dict that can be loaded with
479
- ``load_state_dict()`` and used for training without DeepSpeed or shared with others, for example
480
- via a model hub.
481
-
482
- Args:
483
- - ``checkpoint_dir``: path to the desired checkpoint folder
484
- - ``tag``: checkpoint tag used as a unique identifier for checkpoint. If not provided will attempt to load tag in 'latest' file. e.g., ``global_step14``
485
- - ``exclude_frozen_parameters``: exclude frozen parameters
486
-
487
- Returns:
488
- - pytorch ``state_dict``
489
-
490
- Note: this approach may not work if your application doesn't have sufficient free CPU memory and
491
- you may need to use the offline approach using the ``zero_to_fp32.py`` script that is saved with
492
- the checkpoint.
493
-
494
- A typical usage might be ::
495
-
496
- from deepspeed.utils.zero_to_fp32 import get_fp32_state_dict_from_zero_checkpoint
497
- # do the training and checkpoint saving
498
- state_dict = get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir) # already on cpu
499
- model = model.cpu() # move to cpu
500
- model.load_state_dict(state_dict)
501
- # submit to model hub or save the model to share with others
502
-
503
- In this example the ``model`` will no longer be usable in the deepspeed context of the same
504
- application. i.e. you will need to re-initialize the deepspeed engine, since
505
- ``model.load_state_dict(state_dict)`` will remove all the deepspeed magic from it.
506
-
507
- If you want it all done for you, use ``load_state_dict_from_zero_checkpoint`` instead.
508
-
509
- """
510
- if tag is None:
511
- latest_path = os.path.join(checkpoint_dir, 'latest')
512
- if os.path.isfile(latest_path):
513
- with open(latest_path, 'r') as fd:
514
- tag = fd.read().strip()
515
- else:
516
- raise ValueError(f"Unable to find 'latest' file at {latest_path}")
517
-
518
- ds_checkpoint_dir = os.path.join(checkpoint_dir, tag)
519
-
520
- if not os.path.isdir(ds_checkpoint_dir):
521
- raise FileNotFoundError(f"Directory '{ds_checkpoint_dir}' doesn't exist")
522
-
523
- return _get_fp32_state_dict_from_zero_checkpoint(ds_checkpoint_dir, exclude_frozen_parameters)
524
-
525
-
526
- def convert_zero_checkpoint_to_fp32_state_dict(checkpoint_dir,
527
- output_dir,
528
- max_shard_size="5GB",
529
- safe_serialization=False,
530
- tag=None,
531
- exclude_frozen_parameters=False):
532
- """
533
- Convert ZeRO 2 or 3 checkpoint into a single fp32 consolidated ``state_dict`` file that can be
534
- loaded with ``torch.load(file)`` + ``load_state_dict()`` and used for training without DeepSpeed.
535
-
536
- Args:
537
- - ``checkpoint_dir``: path to the desired checkpoint folder. (one that contains the tag-folder, like ``global_step14``)
538
- - ``output_dir``: directory to the pytorch fp32 state_dict output files
539
- - ``max_shard_size``: the maximum size for a checkpoint before being sharded, default value is 5GB
540
- - ``safe_serialization``: whether to save the model using `safetensors` or the traditional PyTorch way (that uses `pickle`).
541
- - ``tag``: checkpoint tag used as a unique identifier for checkpoint. If not provided will attempt to load tag in the file named ``latest`` in the checkpoint folder, e.g., ``global_step14``
542
- - ``exclude_frozen_parameters``: exclude frozen parameters
543
- """
544
- # Dependency pre-check
545
- if safe_serialization:
546
- try:
547
- from safetensors.torch import save_file
548
- except ImportError:
549
- print('If you want to use `safe_serialization`, please `pip install safetensors`')
550
- raise
551
- if max_shard_size is not None:
552
- try:
553
- from huggingface_hub import split_torch_state_dict_into_shards
554
- except ImportError:
555
- print('If you want to use `max_shard_size`, please `pip install huggingface_hub`')
556
- raise
557
-
558
- # Convert zero checkpoint to state_dict
559
- state_dict = get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir, tag, exclude_frozen_parameters)
560
-
561
- # Shard the model if it is too big.
562
- weights_name = "model.safetensors" if safe_serialization else "pytorch_model.bin"
563
- if max_shard_size is not None:
564
- filename_pattern = weights_name.replace(".bin", "{suffix}.bin").replace(".safetensors", "{suffix}.safetensors")
565
- state_dict_split = split_torch_state_dict_into_shards(state_dict,
566
- filename_pattern=filename_pattern,
567
- max_shard_size=max_shard_size)
568
- else:
569
- from collections import namedtuple
570
- StateDictSplit = namedtuple("StateDictSplit", ["is_sharded", "filename_to_tensors"])
571
- state_dict_split = StateDictSplit(is_sharded=False,
572
- filename_to_tensors={weights_name: list(state_dict.keys())})
573
-
574
- # Save the model
575
- filename_to_tensors = state_dict_split.filename_to_tensors.items()
576
- for shard_file, tensors in tqdm(filename_to_tensors, desc="Saving checkpoint shards"):
577
- shard = {tensor: state_dict[tensor].contiguous() for tensor in tensors}
578
- output_path = os.path.join(output_dir, shard_file)
579
- if safe_serialization:
580
- save_file(shard, output_path, metadata={"format": "pt"})
581
- else:
582
- torch.save(shard, output_path)
583
-
584
- # Save index if sharded
585
- if state_dict_split.is_sharded:
586
- index = {
587
- "metadata": state_dict_split.metadata,
588
- "weight_map": state_dict_split.tensor_to_filename,
589
- }
590
- save_index_file = "model.safetensors.index.json" if safe_serialization else "pytorch_model.bin.index.json"
591
- save_index_file = os.path.join(output_dir, save_index_file)
592
- with open(save_index_file, "w", encoding="utf-8") as f:
593
- content = json.dumps(index, indent=2, sort_keys=True) + "\n"
594
- f.write(content)
595
-
596
-
597
- def load_state_dict_from_zero_checkpoint(model, checkpoint_dir, tag=None):
598
- """
599
- 1. Put the provided model to cpu
600
- 2. Convert ZeRO 2 or 3 checkpoint into a single fp32 consolidated ``state_dict``
601
- 3. Load it into the provided model
602
-
603
- Args:
604
- - ``model``: the model object to update
605
- - ``checkpoint_dir``: path to the desired checkpoint folder. (one that contains the tag-folder, like ``global_step14``)
606
- - ``tag``: checkpoint tag used as a unique identifier for checkpoint. If not provided will attempt to load tag in the file named ``latest`` in the checkpoint folder, e.g., ``global_step14``
607
-
608
- Returns:
609
- - ``model`: modified model
610
-
611
- Make sure you have plenty of CPU memory available before you call this function. If you don't
612
- have enough use the ``zero_to_fp32.py`` utility to do the conversion. You will find it
613
- conveniently placed for you in the checkpoint folder.
614
-
615
- A typical usage might be ::
616
-
617
- from deepspeed.utils.zero_to_fp32 import load_state_dict_from_zero_checkpoint
618
- model = load_state_dict_from_zero_checkpoint(trainer.model, checkpoint_dir)
619
- # submit to model hub or save the model to share with others
620
-
621
- Note, that once this was run, the ``model`` will no longer be usable in the deepspeed context
622
- of the same application. i.e. you will need to re-initialize the deepspeed engine, since
623
- ``model.load_state_dict(state_dict)`` will remove all the deepspeed magic from it.
624
-
625
- """
626
- logger.info(f"Extracting fp32 weights")
627
- state_dict = get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir, tag)
628
-
629
- logger.info(f"Overwriting model with fp32 weights")
630
- model = model.cpu()
631
- model.load_state_dict(state_dict, strict=False)
632
-
633
- return model
634
-
635
-
636
- if __name__ == "__main__":
637
- parser = argparse.ArgumentParser()
638
- parser.add_argument("checkpoint_dir",
639
- type=str,
640
- help="path to the desired checkpoint folder, e.g., path/checkpoint-12")
641
- parser.add_argument("output_dir",
642
- type=str,
643
- help="directory to the pytorch fp32 state_dict output files"
644
- "(e.g. path/checkpoint-12-output/)")
645
- parser.add_argument(
646
- "--max_shard_size",
647
- type=str,
648
- default="5GB",
649
- help="The maximum size for a checkpoint before being sharded. Checkpoints shard will then be each of size"
650
- "lower than this size. If expressed as a string, needs to be digits followed by a unit (like `5MB`"
651
- "We default it to 5GB in order for models to be able to run easily on free-tier google colab instances"
652
- "without CPU OOM issues.")
653
- parser.add_argument(
654
- "--safe_serialization",
655
- default=False,
656
- action='store_true',
657
- help="Whether to save the model using `safetensors` or the traditional PyTorch way (that uses `pickle`).")
658
- parser.add_argument("-t",
659
- "--tag",
660
- type=str,
661
- default=None,
662
- help="checkpoint tag used as a unique identifier for checkpoint. e.g., global_step1")
663
- parser.add_argument("--exclude_frozen_parameters", action='store_true', help="exclude frozen parameters")
664
- parser.add_argument("-d", "--debug", action='store_true', help="enable debug")
665
- args = parser.parse_args()
666
-
667
- debug = args.debug
668
-
669
- convert_zero_checkpoint_to_fp32_state_dict(args.checkpoint_dir,
670
- args.output_dir,
671
- max_shard_size=args.max_shard_size,
672
- safe_serialization=args.safe_serialization,
673
- tag=args.tag,
674
- exclude_frozen_parameters=args.exclude_frozen_parameters)