cogito233 commited on
Commit
0f8c792
·
verified ·
1 Parent(s): e377f91

Init: upload full folder recursively

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. .gitattributes +4 -0
  2. v0-20251012-211828/args.json +369 -0
  3. v0-20251012-211828/checkpoint-224/added_tokens.json +24 -0
  4. v0-20251012-211828/checkpoint-224/args.json +369 -0
  5. v0-20251012-211828/checkpoint-224/config.json +30 -0
  6. v0-20251012-211828/checkpoint-224/generation_config.json +6 -0
  7. v0-20251012-211828/checkpoint-224/latest +1 -0
  8. v0-20251012-211828/checkpoint-224/merges.txt +0 -0
  9. v0-20251012-211828/checkpoint-224/model-00001-of-00004.safetensors +3 -0
  10. v0-20251012-211828/checkpoint-224/model-00002-of-00004.safetensors +3 -0
  11. v0-20251012-211828/checkpoint-224/model-00003-of-00004.safetensors +3 -0
  12. v0-20251012-211828/checkpoint-224/model-00004-of-00004.safetensors +3 -0
  13. v0-20251012-211828/checkpoint-224/model.safetensors.index.json +346 -0
  14. v0-20251012-211828/checkpoint-224/rng_state_0.pth +3 -0
  15. v0-20251012-211828/checkpoint-224/rng_state_1.pth +3 -0
  16. v0-20251012-211828/checkpoint-224/rng_state_2.pth +3 -0
  17. v0-20251012-211828/checkpoint-224/rng_state_3.pth +3 -0
  18. v0-20251012-211828/checkpoint-224/rng_state_4.pth +3 -0
  19. v0-20251012-211828/checkpoint-224/rng_state_5.pth +3 -0
  20. v0-20251012-211828/checkpoint-224/rng_state_6.pth +3 -0
  21. v0-20251012-211828/checkpoint-224/rng_state_7.pth +3 -0
  22. v0-20251012-211828/checkpoint-224/scheduler.pt +3 -0
  23. v0-20251012-211828/checkpoint-224/special_tokens_map.json +31 -0
  24. v0-20251012-211828/checkpoint-224/tokenizer.json +3 -0
  25. v0-20251012-211828/checkpoint-224/tokenizer_config.json +208 -0
  26. v0-20251012-211828/checkpoint-224/trainer_state.json +2283 -0
  27. v0-20251012-211828/checkpoint-224/training_args.bin +3 -0
  28. v0-20251012-211828/checkpoint-224/vocab.json +0 -0
  29. v0-20251012-211828/checkpoint-224/zero_to_fp32.py +604 -0
  30. v0-20251012-211828/checkpoint-448/added_tokens.json +24 -0
  31. v0-20251012-211828/checkpoint-448/args.json +369 -0
  32. v0-20251012-211828/checkpoint-448/config.json +30 -0
  33. v0-20251012-211828/checkpoint-448/generation_config.json +6 -0
  34. v0-20251012-211828/checkpoint-448/latest +1 -0
  35. v0-20251012-211828/checkpoint-448/merges.txt +0 -0
  36. v0-20251012-211828/checkpoint-448/model-00001-of-00004.safetensors +3 -0
  37. v0-20251012-211828/checkpoint-448/model-00002-of-00004.safetensors +3 -0
  38. v0-20251012-211828/checkpoint-448/model-00003-of-00004.safetensors +3 -0
  39. v0-20251012-211828/checkpoint-448/model-00004-of-00004.safetensors +3 -0
  40. v0-20251012-211828/checkpoint-448/model.safetensors.index.json +346 -0
  41. v0-20251012-211828/checkpoint-448/rng_state_0.pth +3 -0
  42. v0-20251012-211828/checkpoint-448/rng_state_1.pth +3 -0
  43. v0-20251012-211828/checkpoint-448/rng_state_2.pth +3 -0
  44. v0-20251012-211828/checkpoint-448/rng_state_3.pth +3 -0
  45. v0-20251012-211828/checkpoint-448/rng_state_4.pth +3 -0
  46. v0-20251012-211828/checkpoint-448/rng_state_5.pth +3 -0
  47. v0-20251012-211828/checkpoint-448/rng_state_6.pth +3 -0
  48. v0-20251012-211828/checkpoint-448/rng_state_7.pth +3 -0
  49. v0-20251012-211828/checkpoint-448/scheduler.pt +3 -0
  50. v0-20251012-211828/checkpoint-448/special_tokens_map.json +31 -0
.gitattributes CHANGED
@@ -33,3 +33,7 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ v0-20251012-211828/checkpoint-224/tokenizer.json filter=lfs diff=lfs merge=lfs -text
37
+ v0-20251012-211828/checkpoint-448/tokenizer.json filter=lfs diff=lfs merge=lfs -text
38
+ v0-20251012-211828/checkpoint-672/tokenizer.json filter=lfs diff=lfs merge=lfs -text
39
+ v0-20251012-211828/checkpoint-896/tokenizer.json filter=lfs diff=lfs merge=lfs -text
v0-20251012-211828/args.json ADDED
@@ -0,0 +1,369 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model": "/data/minimax-dialogue/users/ruobai/cogito/base_model/Qwen2.5-7B",
3
+ "model_type": "qwen2_5",
4
+ "model_revision": null,
5
+ "task_type": "causal_lm",
6
+ "torch_dtype": "bfloat16",
7
+ "attn_impl": "flash_attn",
8
+ "num_labels": null,
9
+ "problem_type": null,
10
+ "rope_scaling": null,
11
+ "device_map": null,
12
+ "max_memory": {},
13
+ "local_repo_path": null,
14
+ "init_strategy": null,
15
+ "template": "qwen2_5",
16
+ "system": null,
17
+ "max_length": 16240,
18
+ "truncation_strategy": "delete",
19
+ "max_pixels": null,
20
+ "agent_template": null,
21
+ "norm_bbox": null,
22
+ "response_prefix": null,
23
+ "padding_side": "right",
24
+ "loss_scale": "default",
25
+ "sequence_parallel_size": 1,
26
+ "use_chat_template": true,
27
+ "template_backend": "swift",
28
+ "dataset": [
29
+ "/data/minimax-dialogue/users/ruobai/cogito/base_dataset/broswer_training_data/merged_data.jsonl"
30
+ ],
31
+ "val_dataset": [],
32
+ "split_dataset_ratio": 0.001,
33
+ "data_seed": 42,
34
+ "dataset_num_proc": 100,
35
+ "load_from_cache_file": true,
36
+ "dataset_shuffle": true,
37
+ "val_dataset_shuffle": false,
38
+ "streaming": false,
39
+ "interleave_prob": null,
40
+ "stopping_strategy": "first_exhausted",
41
+ "shuffle_buffer_size": 1000,
42
+ "download_mode": "reuse_dataset_if_exists",
43
+ "columns": {},
44
+ "strict": false,
45
+ "remove_unused_columns": false,
46
+ "model_name": [
47
+ null,
48
+ null
49
+ ],
50
+ "model_author": [
51
+ null,
52
+ null
53
+ ],
54
+ "custom_dataset_info": [],
55
+ "quant_method": null,
56
+ "quant_bits": null,
57
+ "hqq_axis": null,
58
+ "bnb_4bit_compute_dtype": "bfloat16",
59
+ "bnb_4bit_quant_type": "nf4",
60
+ "bnb_4bit_use_double_quant": true,
61
+ "bnb_4bit_quant_storage": null,
62
+ "max_new_tokens": 64,
63
+ "temperature": 0.0,
64
+ "top_k": null,
65
+ "top_p": null,
66
+ "repetition_penalty": null,
67
+ "num_beams": 1,
68
+ "stream": false,
69
+ "stop_words": [],
70
+ "logprobs": false,
71
+ "top_logprobs": null,
72
+ "ckpt_dir": null,
73
+ "lora_modules": [],
74
+ "tuner_backend": "peft",
75
+ "train_type": "full",
76
+ "adapters": [],
77
+ "external_plugins": [],
78
+ "seed": 42,
79
+ "model_kwargs": {},
80
+ "load_args": false,
81
+ "load_data_args": false,
82
+ "use_hf": true,
83
+ "hub_token": null,
84
+ "custom_register_path": [],
85
+ "ddp_timeout": 1800,
86
+ "ddp_backend": "nccl",
87
+ "ignore_args_error": false,
88
+ "use_swift_lora": false,
89
+ "output_dir": "/data/minimax-dialogue/users/ruobai/swift/model/qwen2p5_7B_ba_1012/v0-20251012-211828",
90
+ "overwrite_output_dir": false,
91
+ "do_train": false,
92
+ "do_eval": false,
93
+ "do_predict": false,
94
+ "eval_strategy": "epoch",
95
+ "prediction_loss_only": false,
96
+ "per_device_train_batch_size": 1,
97
+ "per_device_eval_batch_size": 1,
98
+ "per_gpu_train_batch_size": null,
99
+ "per_gpu_eval_batch_size": null,
100
+ "gradient_accumulation_steps": 2,
101
+ "eval_accumulation_steps": null,
102
+ "eval_delay": 0,
103
+ "torch_empty_cache_steps": null,
104
+ "learning_rate": 1e-05,
105
+ "weight_decay": 0.05,
106
+ "adam_beta1": 0.9,
107
+ "adam_beta2": 0.95,
108
+ "adam_epsilon": 1e-08,
109
+ "max_grad_norm": 1.0,
110
+ "num_train_epochs": 4.0,
111
+ "max_steps": -1,
112
+ "lr_scheduler_type": "cosine",
113
+ "lr_scheduler_kwargs": null,
114
+ "warmup_ratio": 0.05,
115
+ "warmup_steps": 0,
116
+ "log_level": "passive",
117
+ "log_level_replica": "warning",
118
+ "log_on_each_node": true,
119
+ "logging_dir": "/data/minimax-dialogue/users/ruobai/swift/model/qwen2p5_7B_ba_1012/v0-20251012-211828/runs",
120
+ "logging_strategy": "steps",
121
+ "logging_first_step": true,
122
+ "logging_steps": 1,
123
+ "logging_nan_inf_filter": true,
124
+ "save_strategy": "epoch",
125
+ "save_steps": 500,
126
+ "save_total_limit": null,
127
+ "save_safetensors": true,
128
+ "save_on_each_node": false,
129
+ "save_only_model": false,
130
+ "restore_callback_states_from_checkpoint": false,
131
+ "no_cuda": false,
132
+ "use_cpu": false,
133
+ "use_mps_device": false,
134
+ "jit_mode_eval": false,
135
+ "use_ipex": false,
136
+ "bf16": true,
137
+ "fp16": false,
138
+ "fp16_opt_level": "O1",
139
+ "half_precision_backend": "auto",
140
+ "bf16_full_eval": false,
141
+ "fp16_full_eval": false,
142
+ "tf32": null,
143
+ "local_rank": 0,
144
+ "tpu_num_cores": null,
145
+ "tpu_metrics_debug": false,
146
+ "debug": null,
147
+ "dataloader_drop_last": false,
148
+ "eval_steps": 20.0,
149
+ "dataloader_num_workers": 48,
150
+ "dataloader_prefetch_factor": null,
151
+ "past_index": -1,
152
+ "run_name": "/data/minimax-dialogue/users/ruobai/swift/model/qwen2p5_7B_ba_1012/v0-20251012-211828",
153
+ "disable_tqdm": null,
154
+ "label_names": null,
155
+ "load_best_model_at_end": false,
156
+ "metric_for_best_model": "loss",
157
+ "greater_is_better": false,
158
+ "ignore_data_skip": false,
159
+ "fsdp": "",
160
+ "fsdp_min_num_params": 0,
161
+ "fsdp_config": null,
162
+ "tp_size": 0,
163
+ "fsdp_transformer_layer_cls_to_wrap": null,
164
+ "accelerator_config": {
165
+ "dispatch_batches": false
166
+ },
167
+ "deepspeed": {
168
+ "fp16": {
169
+ "enabled": "auto",
170
+ "loss_scale": 0,
171
+ "loss_scale_window": 1000,
172
+ "initial_scale_power": 16,
173
+ "hysteresis": 2,
174
+ "min_loss_scale": 1
175
+ },
176
+ "bf16": {
177
+ "enabled": "auto"
178
+ },
179
+ "zero_optimization": {
180
+ "stage": 3,
181
+ "offload_optimizer": {
182
+ "device": "none",
183
+ "pin_memory": true
184
+ },
185
+ "offload_param": {
186
+ "device": "none",
187
+ "pin_memory": true
188
+ },
189
+ "overlap_comm": false,
190
+ "contiguous_gradients": true,
191
+ "sub_group_size": 1000000000.0,
192
+ "reduce_bucket_size": "auto",
193
+ "zero_quantized_weights": false,
194
+ "zero_quantized_gradients": false,
195
+ "stage3_prefetch_bucket_size": "auto",
196
+ "stage3_param_persistence_threshold": "auto",
197
+ "stage3_max_live_parameters": 1000000000.0,
198
+ "stage3_max_reuse_distance": 1000000000.0,
199
+ "stage3_gather_16bit_weights_on_model_save": true
200
+ },
201
+ "gradient_accumulation_steps": "auto",
202
+ "gradient_clipping": "auto",
203
+ "steps_per_print": 2000,
204
+ "train_batch_size": "auto",
205
+ "train_micro_batch_size_per_gpu": "auto",
206
+ "wall_clock_breakdown": false
207
+ },
208
+ "label_smoothing_factor": 0.0,
209
+ "optim": "adamw_torch",
210
+ "optim_args": null,
211
+ "adafactor": false,
212
+ "group_by_length": false,
213
+ "length_column_name": "length",
214
+ "report_to": [
215
+ "wandb"
216
+ ],
217
+ "ddp_find_unused_parameters": null,
218
+ "ddp_bucket_cap_mb": null,
219
+ "ddp_broadcast_buffers": null,
220
+ "dataloader_pin_memory": true,
221
+ "dataloader_persistent_workers": false,
222
+ "skip_memory_metrics": true,
223
+ "use_legacy_prediction_loop": false,
224
+ "push_to_hub": false,
225
+ "resume_from_checkpoint": null,
226
+ "hub_model_id": null,
227
+ "hub_strategy": "every_save",
228
+ "hub_private_repo": null,
229
+ "hub_always_push": false,
230
+ "gradient_checkpointing": true,
231
+ "gradient_checkpointing_kwargs": null,
232
+ "include_inputs_for_metrics": false,
233
+ "include_for_metrics": [],
234
+ "eval_do_concat_batches": true,
235
+ "fp16_backend": "auto",
236
+ "push_to_hub_model_id": null,
237
+ "push_to_hub_organization": null,
238
+ "push_to_hub_token": null,
239
+ "mp_parameters": "",
240
+ "auto_find_batch_size": false,
241
+ "full_determinism": false,
242
+ "torchdynamo": null,
243
+ "ray_scope": "last",
244
+ "torch_compile": false,
245
+ "torch_compile_backend": null,
246
+ "torch_compile_mode": null,
247
+ "include_tokens_per_second": false,
248
+ "include_num_input_tokens_seen": false,
249
+ "neftune_noise_alpha": null,
250
+ "optim_target_modules": null,
251
+ "batch_eval_metrics": false,
252
+ "eval_on_start": false,
253
+ "use_liger_kernel": false,
254
+ "eval_use_gather_object": false,
255
+ "average_tokens_across_devices": false,
256
+ "sortish_sampler": false,
257
+ "predict_with_generate": false,
258
+ "generation_max_length": null,
259
+ "generation_num_beams": null,
260
+ "generation_config": null,
261
+ "check_model": true,
262
+ "acc_strategy": "token",
263
+ "train_dataloader_shuffle": true,
264
+ "max_epochs": null,
265
+ "metric_warmup_step": 0,
266
+ "fsdp_num": 1,
267
+ "acc_steps": 1,
268
+ "eval_use_evalscope": false,
269
+ "eval_datasets": [],
270
+ "eval_limit": null,
271
+ "eval_datasets_args": null,
272
+ "eval_generation_config": null,
273
+ "freeze_parameters": [],
274
+ "freeze_parameters_regex": null,
275
+ "freeze_parameters_ratio": 0.0,
276
+ "trainable_parameters": [],
277
+ "trainable_parameters_regex": null,
278
+ "freeze_llm": false,
279
+ "freeze_vit": true,
280
+ "freeze_aligner": false,
281
+ "target_modules": [
282
+ "all-linear"
283
+ ],
284
+ "target_regex": null,
285
+ "modules_to_save": [],
286
+ "lora_rank": 8,
287
+ "lora_alpha": 32,
288
+ "lora_dropout": 0.05,
289
+ "lora_bias": "none",
290
+ "lora_dtype": null,
291
+ "lorap_lr_ratio": null,
292
+ "use_rslora": false,
293
+ "use_dora": false,
294
+ "lora_ga_batch_size": 2,
295
+ "lora_ga_iters": 2,
296
+ "lora_ga_max_length": 1024,
297
+ "lora_ga_direction": "ArB2r",
298
+ "lora_ga_scale": "stable",
299
+ "lora_ga_stable_gamma": 16,
300
+ "init_weights": true,
301
+ "fourier_n_frequency": 2000,
302
+ "fourier_scaling": 300.0,
303
+ "boft_block_size": 4,
304
+ "boft_block_num": 0,
305
+ "boft_n_butterfly_factor": 1,
306
+ "boft_dropout": 0.0,
307
+ "vera_rank": 256,
308
+ "vera_projection_prng_key": 0,
309
+ "vera_dropout": 0.0,
310
+ "vera_d_initial": 0.1,
311
+ "adapter_act": "gelu",
312
+ "adapter_length": 128,
313
+ "use_galore": false,
314
+ "galore_target_modules": null,
315
+ "galore_rank": 128,
316
+ "galore_update_proj_gap": 50,
317
+ "galore_scale": 1.0,
318
+ "galore_proj_type": "std",
319
+ "galore_optim_per_parameter": false,
320
+ "galore_with_embedding": false,
321
+ "galore_quantization": false,
322
+ "galore_proj_quant": false,
323
+ "galore_proj_bits": 4,
324
+ "galore_proj_group_size": 256,
325
+ "galore_cos_threshold": 0.4,
326
+ "galore_gamma_proj": 2,
327
+ "galore_queue_size": 5,
328
+ "adalora_target_r": 8,
329
+ "adalora_init_r": 12,
330
+ "adalora_tinit": 0,
331
+ "adalora_tfinal": 0,
332
+ "adalora_deltaT": 1,
333
+ "adalora_beta1": 0.85,
334
+ "adalora_beta2": 0.85,
335
+ "adalora_orth_reg_weight": 0.5,
336
+ "llamapro_num_new_blocks": 4,
337
+ "llamapro_num_groups": null,
338
+ "lisa_activated_layers": 0,
339
+ "lisa_step_interval": 20,
340
+ "reft_layer_key": null,
341
+ "reft_layers": null,
342
+ "reft_rank": 4,
343
+ "reft_intervention_type": "LoreftIntervention",
344
+ "reft_args": null,
345
+ "swanlab_token": null,
346
+ "swanlab_project": null,
347
+ "swanlab_workspace": null,
348
+ "swanlab_exp_name": null,
349
+ "swanlab_mode": "cloud",
350
+ "add_version": true,
351
+ "resume_only_model": false,
352
+ "create_checkpoint_symlink": false,
353
+ "packing": true,
354
+ "lazy_tokenize": false,
355
+ "loss_type": null,
356
+ "optimizer": null,
357
+ "metric": null,
358
+ "zero_hpz_partition_size": null,
359
+ "rank": 0,
360
+ "global_world_size": 8,
361
+ "local_world_size": 8,
362
+ "model_suffix": "Qwen2.5-7B",
363
+ "model_info": "ModelInfo(model_type='qwen2_5', model_dir='/data/minimax-dialogue/users/ruobai/cogito/base_model/Qwen2.5-7B', torch_dtype=torch.bfloat16, max_model_len=131072, quant_method=None, quant_bits=None, rope_scaling=None, config=None, task_type='causal_lm', num_labels=None)",
364
+ "model_meta": "ModelMeta(model_type='qwen2_5', model_groups=[ModelGroup(models=[Model(ms_model_id='Qwen/Qwen2.5-0.5B-Instruct', hf_model_id='Qwen/Qwen2.5-0.5B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-1.5B-Instruct', hf_model_id='Qwen/Qwen2.5-1.5B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-3B-Instruct', hf_model_id='Qwen/Qwen2.5-3B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-7B-Instruct', hf_model_id='Qwen/Qwen2.5-7B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-14B-Instruct', hf_model_id='Qwen/Qwen2.5-14B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-32B-Instruct', hf_model_id='Qwen/Qwen2.5-32B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-72B-Instruct', hf_model_id='Qwen/Qwen2.5-72B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-0.5B', hf_model_id='Qwen/Qwen2.5-0.5B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-1.5B', hf_model_id='Qwen/Qwen2.5-1.5B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-3B', hf_model_id='Qwen/Qwen2.5-3B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-7B', hf_model_id='Qwen/Qwen2.5-7B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-14B', hf_model_id='Qwen/Qwen2.5-14B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-32B', hf_model_id='Qwen/Qwen2.5-32B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-72B', hf_model_id='Qwen/Qwen2.5-72B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-0.5B-Instruct-GPTQ-Int4', hf_model_id='Qwen/Qwen2.5-0.5B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-1.5B-Instruct-GPTQ-Int4', hf_model_id='Qwen/Qwen2.5-1.5B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-3B-Instruct-GPTQ-Int4', hf_model_id='Qwen/Qwen2.5-3B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-7B-Instruct-GPTQ-Int4', hf_model_id='Qwen/Qwen2.5-7B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-14B-Instruct-GPTQ-Int4', hf_model_id='Qwen/Qwen2.5-14B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-32B-Instruct-GPTQ-Int4', hf_model_id='Qwen/Qwen2.5-32B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-72B-Instruct-GPTQ-Int4', hf_model_id='Qwen/Qwen2.5-72B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-0.5B-Instruct-GPTQ-Int8', hf_model_id='Qwen/Qwen2.5-0.5B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-1.5B-Instruct-GPTQ-Int8', hf_model_id='Qwen/Qwen2.5-1.5B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-3B-Instruct-GPTQ-Int8', hf_model_id='Qwen/Qwen2.5-3B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-7B-Instruct-GPTQ-Int8', hf_model_id='Qwen/Qwen2.5-7B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-14B-Instruct-GPTQ-Int8', hf_model_id='Qwen/Qwen2.5-14B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-32B-Instruct-GPTQ-Int8', hf_model_id='Qwen/Qwen2.5-32B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-72B-Instruct-GPTQ-Int8', hf_model_id='Qwen/Qwen2.5-72B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-0.5B-Instruct-AWQ', hf_model_id='Qwen/Qwen2.5-0.5B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-1.5B-Instruct-AWQ', hf_model_id='Qwen/Qwen2.5-1.5B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-3B-Instruct-AWQ', hf_model_id='Qwen/Qwen2.5-3B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-7B-Instruct-AWQ', hf_model_id='Qwen/Qwen2.5-7B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-14B-Instruct-AWQ', hf_model_id='Qwen/Qwen2.5-14B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-32B-Instruct-AWQ', hf_model_id='Qwen/Qwen2.5-32B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-72B-Instruct-AWQ', hf_model_id='Qwen/Qwen2.5-72B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None)], ignore_patterns=None, requires=None, tags=[]), ModelGroup(models=[Model(ms_model_id='Qwen/Qwen2.5-Coder-0.5B-Instruct', hf_model_id='Qwen/Qwen2.5-Coder-0.5B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-1.5B-Instruct', hf_model_id='Qwen/Qwen2.5-Coder-1.5B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-3B-Instruct', hf_model_id='Qwen/Qwen2.5-Coder-3B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-7B-Instruct', hf_model_id='Qwen/Qwen2.5-Coder-7B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-14B-Instruct', hf_model_id='Qwen/Qwen2.5-Coder-14B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-32B-Instruct', hf_model_id='Qwen/Qwen2.5-Coder-32B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-0.5B', hf_model_id='Qwen/Qwen2.5-Coder-0.5B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-1.5B', hf_model_id='Qwen/Qwen2.5-Coder-1.5B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-3B', hf_model_id='Qwen/Qwen2.5-Coder-3B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-7B', hf_model_id='Qwen/Qwen2.5-Coder-7B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-14B', hf_model_id='Qwen/Qwen2.5-Coder-14B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-32B', hf_model_id='Qwen/Qwen2.5-Coder-32B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-0.5B-Instruct-AWQ', hf_model_id='Qwen/Qwen2.5-Coder-0.5B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-1.5B-Instruct-AWQ', hf_model_id='Qwen/Qwen2.5-Coder-1.5B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-3B-Instruct-AWQ', hf_model_id='Qwen/Qwen2.5-Coder-3B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-7B-Instruct-AWQ', hf_model_id='Qwen/Qwen2.5-Coder-7B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-14B-Instruct-AWQ', hf_model_id='Qwen/Qwen2.5-Coder-14B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-32B-Instruct-AWQ', hf_model_id='Qwen/Qwen2.5-Coder-32B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-0.5B-Instruct-GPTQ-Int4', hf_model_id='Qwen/Qwen2.5-Coder-0.5B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-0.5B-Instruct-GPTQ-Int8', hf_model_id='Qwen/Qwen2.5-Coder-0.5B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-1.5B-Instruct-GPTQ-Int4', hf_model_id='Qwen/Qwen2.5-Coder-1.5B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-1.5B-Instruct-GPTQ-Int8', hf_model_id='Qwen/Qwen2.5-Coder-1.5B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-3B-Instruct-GPTQ-Int4', hf_model_id='Qwen/Qwen2.5-Coder-3B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-3B-Instruct-GPTQ-Int8', hf_model_id='Qwen/Qwen2.5-Coder-3B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-7B-Instruct-GPTQ-Int4', hf_model_id='Qwen/Qwen2.5-Coder-7B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-7B-Instruct-GPTQ-Int8', hf_model_id='Qwen/Qwen2.5-Coder-7B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-14B-Instruct-GPTQ-Int4', hf_model_id='Qwen/Qwen2.5-Coder-14B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-14B-Instruct-GPTQ-Int8', hf_model_id='Qwen/Qwen2.5-Coder-14B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-32B-Instruct-GPTQ-Int4', hf_model_id='Qwen/Qwen2.5-Coder-32B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-32B-Instruct-GPTQ-Int8', hf_model_id='Qwen/Qwen2.5-Coder-32B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None)], ignore_patterns=None, requires=None, tags=['coding'])], template='qwen2_5', get_function=<function get_model_tokenizer_with_flash_attn at 0x7f3ff78a7920>, model_arch='llama', architectures=['Qwen2ForCausalLM'], additional_saved_files=[], torch_dtype=None, is_multimodal=False, is_reward=False, task_type=None, ignore_patterns=None, requires=['transformers>=4.37'], tags=[])",
365
+ "model_dir": "/data/minimax-dialogue/users/ruobai/cogito/base_model/Qwen2.5-7B",
366
+ "hub": "<class 'swift.hub.hub.HFHub'>",
367
+ "evaluation_strategy": "epoch",
368
+ "training_args": "Seq2SeqTrainingArguments(output_dir='/data/minimax-dialogue/users/ruobai/swift/model/qwen2p5_7B_ba_1012/v0-20251012-211828', overwrite_output_dir=False, do_train=False, do_eval=True, do_predict=False, eval_strategy=<IntervalStrategy.EPOCH: 'epoch'>, prediction_loss_only=False, per_device_train_batch_size=1, per_device_eval_batch_size=1, per_gpu_train_batch_size=None, per_gpu_eval_batch_size=None, gradient_accumulation_steps=2, eval_accumulation_steps=None, eval_delay=0, torch_empty_cache_steps=None, learning_rate=1e-05, weight_decay=0.05, adam_beta1=0.9, adam_beta2=0.95, adam_epsilon=1e-08, max_grad_norm=1.0, num_train_epochs=4.0, max_steps=-1, lr_scheduler_type=<SchedulerType.COSINE: 'cosine'>, lr_scheduler_kwargs=None, warmup_ratio=0.05, warmup_steps=0, log_level='passive', log_level_replica='warning', log_on_each_node=True, logging_dir='/data/minimax-dialogue/users/ruobai/swift/model/qwen2p5_7B_ba_1012/v0-20251012-211828/runs', logging_strategy=<IntervalStrategy.STEPS: 'steps'>, logging_first_step=True, logging_steps=1, logging_nan_inf_filter=True, save_strategy=<SaveStrategy.EPOCH: 'epoch'>, save_steps=500, save_total_limit=None, save_safetensors=True, save_on_each_node=False, save_only_model=False, restore_callback_states_from_checkpoint=False, no_cuda=False, use_cpu=False, use_mps_device=False, seed=42, data_seed=42, jit_mode_eval=False, use_ipex=False, bf16=True, fp16=False, fp16_opt_level='O1', half_precision_backend='auto', bf16_full_eval=False, fp16_full_eval=False, tf32=None, local_rank=0, ddp_backend='nccl', tpu_num_cores=None, tpu_metrics_debug=False, debug=[], dataloader_drop_last=False, eval_steps=20.0, dataloader_num_workers=48, dataloader_prefetch_factor=10, past_index=-1, run_name='/data/minimax-dialogue/users/ruobai/swift/model/qwen2p5_7B_ba_1012/v0-20251012-211828', disable_tqdm=False, remove_unused_columns=False, label_names=None, load_best_model_at_end=False, metric_for_best_model='loss', greater_is_better=False, ignore_data_skip=False, fsdp=[], fsdp_min_num_params=0, fsdp_config={'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}, tp_size=0, fsdp_transformer_layer_cls_to_wrap=None, accelerator_config=AcceleratorConfig(split_batches=False, dispatch_batches=False, even_batches=True, use_seedable_sampler=True, non_blocking=False, gradient_accumulation_kwargs=None, use_configured_state=False), deepspeed={'fp16': {'enabled': 'auto', 'loss_scale': 0, 'loss_scale_window': 1000, 'initial_scale_power': 16, 'hysteresis': 2, 'min_loss_scale': 1}, 'bf16': {'enabled': 'auto'}, 'zero_optimization': {'stage': 3, 'offload_optimizer': {'device': 'none', 'pin_memory': True}, 'offload_param': {'device': 'none', 'pin_memory': True}, 'overlap_comm': False, 'contiguous_gradients': True, 'sub_group_size': 1000000000.0, 'reduce_bucket_size': 'auto', 'zero_quantized_weights': False, 'zero_quantized_gradients': False, 'stage3_prefetch_bucket_size': 'auto', 'stage3_param_persistence_threshold': 'auto', 'stage3_max_live_parameters': 1000000000.0, 'stage3_max_reuse_distance': 1000000000.0, 'stage3_gather_16bit_weights_on_model_save': True}, 'gradient_accumulation_steps': 'auto', 'gradient_clipping': 'auto', 'steps_per_print': 2000, 'train_batch_size': 'auto', 'train_micro_batch_size_per_gpu': 'auto', 'wall_clock_breakdown': False}, label_smoothing_factor=0.0, optim=<OptimizerNames.ADAMW_TORCH: 'adamw_torch'>, optim_args=None, adafactor=False, group_by_length=False, length_column_name='length', report_to=['wandb'], ddp_find_unused_parameters=None, ddp_bucket_cap_mb=None, ddp_broadcast_buffers=None, dataloader_pin_memory=True, dataloader_persistent_workers=False, skip_memory_metrics=True, use_legacy_prediction_loop=False, push_to_hub=False, resume_from_checkpoint=None, hub_model_id=None, hub_strategy=<HubStrategy.EVERY_SAVE: 'every_save'>, hub_token=None, hub_private_repo=None, hub_always_push=False, gradient_checkpointing=True, gradient_checkpointing_kwargs=None, include_inputs_for_metrics=False, include_for_metrics=[], eval_do_concat_batches=True, fp16_backend='auto', push_to_hub_model_id=None, push_to_hub_organization=None, push_to_hub_token=None, mp_parameters='', auto_find_batch_size=False, full_determinism=False, torchdynamo=None, ray_scope='last', ddp_timeout=1800, torch_compile=False, torch_compile_backend=None, torch_compile_mode=None, include_tokens_per_second=None, include_num_input_tokens_seen=None, neftune_noise_alpha=None, optim_target_modules=None, batch_eval_metrics=False, eval_on_start=False, use_liger_kernel=False, eval_use_gather_object=False, average_tokens_across_devices=None, sortish_sampler=False, predict_with_generate=False, generation_max_length=None, generation_num_beams=None, generation_config=None, check_model=True, acc_strategy='token', train_dataloader_shuffle=True, max_epochs=None, metric_warmup_step=0, fsdp_num=1, acc_steps=1, eval_use_evalscope=False, eval_datasets=[], eval_limit=None, eval_datasets_args=None, eval_generation_config=None, train_type='full', optimizer=None, local_repo_path=None, galore_config=None)"
369
+ }
v0-20251012-211828/checkpoint-224/added_tokens.json ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "</tool_call>": 151658,
3
+ "<tool_call>": 151657,
4
+ "<|box_end|>": 151649,
5
+ "<|box_start|>": 151648,
6
+ "<|endoftext|>": 151643,
7
+ "<|file_sep|>": 151664,
8
+ "<|fim_middle|>": 151660,
9
+ "<|fim_pad|>": 151662,
10
+ "<|fim_prefix|>": 151659,
11
+ "<|fim_suffix|>": 151661,
12
+ "<|im_end|>": 151645,
13
+ "<|im_start|>": 151644,
14
+ "<|image_pad|>": 151655,
15
+ "<|object_ref_end|>": 151647,
16
+ "<|object_ref_start|>": 151646,
17
+ "<|quad_end|>": 151651,
18
+ "<|quad_start|>": 151650,
19
+ "<|repo_name|>": 151663,
20
+ "<|video_pad|>": 151656,
21
+ "<|vision_end|>": 151653,
22
+ "<|vision_pad|>": 151654,
23
+ "<|vision_start|>": 151652
24
+ }
v0-20251012-211828/checkpoint-224/args.json ADDED
@@ -0,0 +1,369 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model": "/data/minimax-dialogue/users/ruobai/cogito/base_model/Qwen2.5-7B",
3
+ "model_type": "qwen2_5",
4
+ "model_revision": null,
5
+ "task_type": "causal_lm",
6
+ "torch_dtype": "bfloat16",
7
+ "attn_impl": "flash_attn",
8
+ "num_labels": null,
9
+ "problem_type": null,
10
+ "rope_scaling": null,
11
+ "device_map": null,
12
+ "max_memory": {},
13
+ "local_repo_path": null,
14
+ "init_strategy": null,
15
+ "template": "qwen2_5",
16
+ "system": null,
17
+ "max_length": 16240,
18
+ "truncation_strategy": "delete",
19
+ "max_pixels": null,
20
+ "agent_template": null,
21
+ "norm_bbox": null,
22
+ "response_prefix": null,
23
+ "padding_side": "right",
24
+ "loss_scale": "default",
25
+ "sequence_parallel_size": 1,
26
+ "use_chat_template": true,
27
+ "template_backend": "swift",
28
+ "dataset": [
29
+ "/data/minimax-dialogue/users/ruobai/cogito/base_dataset/broswer_training_data/merged_data.jsonl"
30
+ ],
31
+ "val_dataset": [],
32
+ "split_dataset_ratio": 0.001,
33
+ "data_seed": 42,
34
+ "dataset_num_proc": 100,
35
+ "load_from_cache_file": true,
36
+ "dataset_shuffle": true,
37
+ "val_dataset_shuffle": false,
38
+ "streaming": false,
39
+ "interleave_prob": null,
40
+ "stopping_strategy": "first_exhausted",
41
+ "shuffle_buffer_size": 1000,
42
+ "download_mode": "reuse_dataset_if_exists",
43
+ "columns": {},
44
+ "strict": false,
45
+ "remove_unused_columns": false,
46
+ "model_name": [
47
+ null,
48
+ null
49
+ ],
50
+ "model_author": [
51
+ null,
52
+ null
53
+ ],
54
+ "custom_dataset_info": [],
55
+ "quant_method": null,
56
+ "quant_bits": null,
57
+ "hqq_axis": null,
58
+ "bnb_4bit_compute_dtype": "bfloat16",
59
+ "bnb_4bit_quant_type": "nf4",
60
+ "bnb_4bit_use_double_quant": true,
61
+ "bnb_4bit_quant_storage": null,
62
+ "max_new_tokens": 64,
63
+ "temperature": 0.0,
64
+ "top_k": null,
65
+ "top_p": null,
66
+ "repetition_penalty": null,
67
+ "num_beams": 1,
68
+ "stream": false,
69
+ "stop_words": [],
70
+ "logprobs": false,
71
+ "top_logprobs": null,
72
+ "ckpt_dir": null,
73
+ "lora_modules": [],
74
+ "tuner_backend": "peft",
75
+ "train_type": "full",
76
+ "adapters": [],
77
+ "external_plugins": [],
78
+ "seed": 42,
79
+ "model_kwargs": {},
80
+ "load_args": false,
81
+ "load_data_args": false,
82
+ "use_hf": true,
83
+ "hub_token": null,
84
+ "custom_register_path": [],
85
+ "ddp_timeout": 1800,
86
+ "ddp_backend": "nccl",
87
+ "ignore_args_error": false,
88
+ "use_swift_lora": false,
89
+ "output_dir": "/data/minimax-dialogue/users/ruobai/swift/model/qwen2p5_7B_ba_1012/v0-20251012-211828",
90
+ "overwrite_output_dir": false,
91
+ "do_train": false,
92
+ "do_eval": false,
93
+ "do_predict": false,
94
+ "eval_strategy": "epoch",
95
+ "prediction_loss_only": false,
96
+ "per_device_train_batch_size": 1,
97
+ "per_device_eval_batch_size": 1,
98
+ "per_gpu_train_batch_size": null,
99
+ "per_gpu_eval_batch_size": null,
100
+ "gradient_accumulation_steps": 2,
101
+ "eval_accumulation_steps": null,
102
+ "eval_delay": 0,
103
+ "torch_empty_cache_steps": null,
104
+ "learning_rate": 1e-05,
105
+ "weight_decay": 0.05,
106
+ "adam_beta1": 0.9,
107
+ "adam_beta2": 0.95,
108
+ "adam_epsilon": 1e-08,
109
+ "max_grad_norm": 1.0,
110
+ "num_train_epochs": 4.0,
111
+ "max_steps": -1,
112
+ "lr_scheduler_type": "cosine",
113
+ "lr_scheduler_kwargs": null,
114
+ "warmup_ratio": 0.05,
115
+ "warmup_steps": 0,
116
+ "log_level": "passive",
117
+ "log_level_replica": "warning",
118
+ "log_on_each_node": true,
119
+ "logging_dir": "/data/minimax-dialogue/users/ruobai/swift/model/qwen2p5_7B_ba_1012/v0-20251012-211828/runs",
120
+ "logging_strategy": "steps",
121
+ "logging_first_step": true,
122
+ "logging_steps": 1,
123
+ "logging_nan_inf_filter": true,
124
+ "save_strategy": "epoch",
125
+ "save_steps": 500,
126
+ "save_total_limit": null,
127
+ "save_safetensors": true,
128
+ "save_on_each_node": false,
129
+ "save_only_model": false,
130
+ "restore_callback_states_from_checkpoint": false,
131
+ "no_cuda": false,
132
+ "use_cpu": false,
133
+ "use_mps_device": false,
134
+ "jit_mode_eval": false,
135
+ "use_ipex": false,
136
+ "bf16": true,
137
+ "fp16": false,
138
+ "fp16_opt_level": "O1",
139
+ "half_precision_backend": "auto",
140
+ "bf16_full_eval": false,
141
+ "fp16_full_eval": false,
142
+ "tf32": null,
143
+ "local_rank": 0,
144
+ "tpu_num_cores": null,
145
+ "tpu_metrics_debug": false,
146
+ "debug": null,
147
+ "dataloader_drop_last": false,
148
+ "eval_steps": 20.0,
149
+ "dataloader_num_workers": 48,
150
+ "dataloader_prefetch_factor": null,
151
+ "past_index": -1,
152
+ "run_name": "/data/minimax-dialogue/users/ruobai/swift/model/qwen2p5_7B_ba_1012/v0-20251012-211828",
153
+ "disable_tqdm": null,
154
+ "label_names": null,
155
+ "load_best_model_at_end": false,
156
+ "metric_for_best_model": "loss",
157
+ "greater_is_better": false,
158
+ "ignore_data_skip": false,
159
+ "fsdp": "",
160
+ "fsdp_min_num_params": 0,
161
+ "fsdp_config": null,
162
+ "tp_size": 0,
163
+ "fsdp_transformer_layer_cls_to_wrap": null,
164
+ "accelerator_config": {
165
+ "dispatch_batches": false
166
+ },
167
+ "deepspeed": {
168
+ "fp16": {
169
+ "enabled": "auto",
170
+ "loss_scale": 0,
171
+ "loss_scale_window": 1000,
172
+ "initial_scale_power": 16,
173
+ "hysteresis": 2,
174
+ "min_loss_scale": 1
175
+ },
176
+ "bf16": {
177
+ "enabled": "auto"
178
+ },
179
+ "zero_optimization": {
180
+ "stage": 3,
181
+ "offload_optimizer": {
182
+ "device": "none",
183
+ "pin_memory": true
184
+ },
185
+ "offload_param": {
186
+ "device": "none",
187
+ "pin_memory": true
188
+ },
189
+ "overlap_comm": false,
190
+ "contiguous_gradients": true,
191
+ "sub_group_size": 1000000000.0,
192
+ "reduce_bucket_size": "auto",
193
+ "zero_quantized_weights": false,
194
+ "zero_quantized_gradients": false,
195
+ "stage3_prefetch_bucket_size": "auto",
196
+ "stage3_param_persistence_threshold": "auto",
197
+ "stage3_max_live_parameters": 1000000000.0,
198
+ "stage3_max_reuse_distance": 1000000000.0,
199
+ "stage3_gather_16bit_weights_on_model_save": true
200
+ },
201
+ "gradient_accumulation_steps": "auto",
202
+ "gradient_clipping": "auto",
203
+ "steps_per_print": 2000,
204
+ "train_batch_size": "auto",
205
+ "train_micro_batch_size_per_gpu": "auto",
206
+ "wall_clock_breakdown": false
207
+ },
208
+ "label_smoothing_factor": 0.0,
209
+ "optim": "adamw_torch",
210
+ "optim_args": null,
211
+ "adafactor": false,
212
+ "group_by_length": false,
213
+ "length_column_name": "length",
214
+ "report_to": [
215
+ "wandb"
216
+ ],
217
+ "ddp_find_unused_parameters": null,
218
+ "ddp_bucket_cap_mb": null,
219
+ "ddp_broadcast_buffers": null,
220
+ "dataloader_pin_memory": true,
221
+ "dataloader_persistent_workers": false,
222
+ "skip_memory_metrics": true,
223
+ "use_legacy_prediction_loop": false,
224
+ "push_to_hub": false,
225
+ "resume_from_checkpoint": null,
226
+ "hub_model_id": null,
227
+ "hub_strategy": "every_save",
228
+ "hub_private_repo": null,
229
+ "hub_always_push": false,
230
+ "gradient_checkpointing": true,
231
+ "gradient_checkpointing_kwargs": null,
232
+ "include_inputs_for_metrics": false,
233
+ "include_for_metrics": [],
234
+ "eval_do_concat_batches": true,
235
+ "fp16_backend": "auto",
236
+ "push_to_hub_model_id": null,
237
+ "push_to_hub_organization": null,
238
+ "push_to_hub_token": null,
239
+ "mp_parameters": "",
240
+ "auto_find_batch_size": false,
241
+ "full_determinism": false,
242
+ "torchdynamo": null,
243
+ "ray_scope": "last",
244
+ "torch_compile": false,
245
+ "torch_compile_backend": null,
246
+ "torch_compile_mode": null,
247
+ "include_tokens_per_second": false,
248
+ "include_num_input_tokens_seen": false,
249
+ "neftune_noise_alpha": null,
250
+ "optim_target_modules": null,
251
+ "batch_eval_metrics": false,
252
+ "eval_on_start": false,
253
+ "use_liger_kernel": false,
254
+ "eval_use_gather_object": false,
255
+ "average_tokens_across_devices": false,
256
+ "sortish_sampler": false,
257
+ "predict_with_generate": false,
258
+ "generation_max_length": null,
259
+ "generation_num_beams": null,
260
+ "generation_config": null,
261
+ "check_model": true,
262
+ "acc_strategy": "token",
263
+ "train_dataloader_shuffle": true,
264
+ "max_epochs": null,
265
+ "metric_warmup_step": 0,
266
+ "fsdp_num": 1,
267
+ "acc_steps": 1,
268
+ "eval_use_evalscope": false,
269
+ "eval_datasets": [],
270
+ "eval_limit": null,
271
+ "eval_datasets_args": null,
272
+ "eval_generation_config": null,
273
+ "freeze_parameters": [],
274
+ "freeze_parameters_regex": null,
275
+ "freeze_parameters_ratio": 0.0,
276
+ "trainable_parameters": [],
277
+ "trainable_parameters_regex": null,
278
+ "freeze_llm": false,
279
+ "freeze_vit": true,
280
+ "freeze_aligner": false,
281
+ "target_modules": [
282
+ "all-linear"
283
+ ],
284
+ "target_regex": null,
285
+ "modules_to_save": [],
286
+ "lora_rank": 8,
287
+ "lora_alpha": 32,
288
+ "lora_dropout": 0.05,
289
+ "lora_bias": "none",
290
+ "lora_dtype": null,
291
+ "lorap_lr_ratio": null,
292
+ "use_rslora": false,
293
+ "use_dora": false,
294
+ "lora_ga_batch_size": 2,
295
+ "lora_ga_iters": 2,
296
+ "lora_ga_max_length": 1024,
297
+ "lora_ga_direction": "ArB2r",
298
+ "lora_ga_scale": "stable",
299
+ "lora_ga_stable_gamma": 16,
300
+ "init_weights": true,
301
+ "fourier_n_frequency": 2000,
302
+ "fourier_scaling": 300.0,
303
+ "boft_block_size": 4,
304
+ "boft_block_num": 0,
305
+ "boft_n_butterfly_factor": 1,
306
+ "boft_dropout": 0.0,
307
+ "vera_rank": 256,
308
+ "vera_projection_prng_key": 0,
309
+ "vera_dropout": 0.0,
310
+ "vera_d_initial": 0.1,
311
+ "adapter_act": "gelu",
312
+ "adapter_length": 128,
313
+ "use_galore": false,
314
+ "galore_target_modules": null,
315
+ "galore_rank": 128,
316
+ "galore_update_proj_gap": 50,
317
+ "galore_scale": 1.0,
318
+ "galore_proj_type": "std",
319
+ "galore_optim_per_parameter": false,
320
+ "galore_with_embedding": false,
321
+ "galore_quantization": false,
322
+ "galore_proj_quant": false,
323
+ "galore_proj_bits": 4,
324
+ "galore_proj_group_size": 256,
325
+ "galore_cos_threshold": 0.4,
326
+ "galore_gamma_proj": 2,
327
+ "galore_queue_size": 5,
328
+ "adalora_target_r": 8,
329
+ "adalora_init_r": 12,
330
+ "adalora_tinit": 0,
331
+ "adalora_tfinal": 0,
332
+ "adalora_deltaT": 1,
333
+ "adalora_beta1": 0.85,
334
+ "adalora_beta2": 0.85,
335
+ "adalora_orth_reg_weight": 0.5,
336
+ "llamapro_num_new_blocks": 4,
337
+ "llamapro_num_groups": null,
338
+ "lisa_activated_layers": 0,
339
+ "lisa_step_interval": 20,
340
+ "reft_layer_key": null,
341
+ "reft_layers": null,
342
+ "reft_rank": 4,
343
+ "reft_intervention_type": "LoreftIntervention",
344
+ "reft_args": null,
345
+ "swanlab_token": null,
346
+ "swanlab_project": null,
347
+ "swanlab_workspace": null,
348
+ "swanlab_exp_name": null,
349
+ "swanlab_mode": "cloud",
350
+ "add_version": true,
351
+ "resume_only_model": false,
352
+ "create_checkpoint_symlink": false,
353
+ "packing": true,
354
+ "lazy_tokenize": false,
355
+ "loss_type": null,
356
+ "optimizer": null,
357
+ "metric": null,
358
+ "zero_hpz_partition_size": null,
359
+ "rank": 0,
360
+ "global_world_size": 8,
361
+ "local_world_size": 8,
362
+ "model_suffix": "Qwen2.5-7B",
363
+ "model_info": "ModelInfo(model_type='qwen2_5', model_dir='/data/minimax-dialogue/users/ruobai/cogito/base_model/Qwen2.5-7B', torch_dtype=torch.bfloat16, max_model_len=131072, quant_method=None, quant_bits=None, rope_scaling=None, config=None, task_type='causal_lm', num_labels=None)",
364
+ "model_meta": "ModelMeta(model_type='qwen2_5', model_groups=[ModelGroup(models=[Model(ms_model_id='Qwen/Qwen2.5-0.5B-Instruct', hf_model_id='Qwen/Qwen2.5-0.5B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-1.5B-Instruct', hf_model_id='Qwen/Qwen2.5-1.5B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-3B-Instruct', hf_model_id='Qwen/Qwen2.5-3B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-7B-Instruct', hf_model_id='Qwen/Qwen2.5-7B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-14B-Instruct', hf_model_id='Qwen/Qwen2.5-14B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-32B-Instruct', hf_model_id='Qwen/Qwen2.5-32B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-72B-Instruct', hf_model_id='Qwen/Qwen2.5-72B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-0.5B', hf_model_id='Qwen/Qwen2.5-0.5B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-1.5B', hf_model_id='Qwen/Qwen2.5-1.5B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-3B', hf_model_id='Qwen/Qwen2.5-3B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-7B', hf_model_id='Qwen/Qwen2.5-7B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-14B', hf_model_id='Qwen/Qwen2.5-14B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-32B', hf_model_id='Qwen/Qwen2.5-32B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-72B', hf_model_id='Qwen/Qwen2.5-72B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-0.5B-Instruct-GPTQ-Int4', hf_model_id='Qwen/Qwen2.5-0.5B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-1.5B-Instruct-GPTQ-Int4', hf_model_id='Qwen/Qwen2.5-1.5B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-3B-Instruct-GPTQ-Int4', hf_model_id='Qwen/Qwen2.5-3B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-7B-Instruct-GPTQ-Int4', hf_model_id='Qwen/Qwen2.5-7B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-14B-Instruct-GPTQ-Int4', hf_model_id='Qwen/Qwen2.5-14B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-32B-Instruct-GPTQ-Int4', hf_model_id='Qwen/Qwen2.5-32B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-72B-Instruct-GPTQ-Int4', hf_model_id='Qwen/Qwen2.5-72B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-0.5B-Instruct-GPTQ-Int8', hf_model_id='Qwen/Qwen2.5-0.5B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-1.5B-Instruct-GPTQ-Int8', hf_model_id='Qwen/Qwen2.5-1.5B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-3B-Instruct-GPTQ-Int8', hf_model_id='Qwen/Qwen2.5-3B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-7B-Instruct-GPTQ-Int8', hf_model_id='Qwen/Qwen2.5-7B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-14B-Instruct-GPTQ-Int8', hf_model_id='Qwen/Qwen2.5-14B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-32B-Instruct-GPTQ-Int8', hf_model_id='Qwen/Qwen2.5-32B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-72B-Instruct-GPTQ-Int8', hf_model_id='Qwen/Qwen2.5-72B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-0.5B-Instruct-AWQ', hf_model_id='Qwen/Qwen2.5-0.5B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-1.5B-Instruct-AWQ', hf_model_id='Qwen/Qwen2.5-1.5B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-3B-Instruct-AWQ', hf_model_id='Qwen/Qwen2.5-3B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-7B-Instruct-AWQ', hf_model_id='Qwen/Qwen2.5-7B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-14B-Instruct-AWQ', hf_model_id='Qwen/Qwen2.5-14B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-32B-Instruct-AWQ', hf_model_id='Qwen/Qwen2.5-32B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-72B-Instruct-AWQ', hf_model_id='Qwen/Qwen2.5-72B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None)], ignore_patterns=None, requires=None, tags=[]), ModelGroup(models=[Model(ms_model_id='Qwen/Qwen2.5-Coder-0.5B-Instruct', hf_model_id='Qwen/Qwen2.5-Coder-0.5B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-1.5B-Instruct', hf_model_id='Qwen/Qwen2.5-Coder-1.5B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-3B-Instruct', hf_model_id='Qwen/Qwen2.5-Coder-3B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-7B-Instruct', hf_model_id='Qwen/Qwen2.5-Coder-7B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-14B-Instruct', hf_model_id='Qwen/Qwen2.5-Coder-14B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-32B-Instruct', hf_model_id='Qwen/Qwen2.5-Coder-32B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-0.5B', hf_model_id='Qwen/Qwen2.5-Coder-0.5B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-1.5B', hf_model_id='Qwen/Qwen2.5-Coder-1.5B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-3B', hf_model_id='Qwen/Qwen2.5-Coder-3B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-7B', hf_model_id='Qwen/Qwen2.5-Coder-7B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-14B', hf_model_id='Qwen/Qwen2.5-Coder-14B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-32B', hf_model_id='Qwen/Qwen2.5-Coder-32B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-0.5B-Instruct-AWQ', hf_model_id='Qwen/Qwen2.5-Coder-0.5B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-1.5B-Instruct-AWQ', hf_model_id='Qwen/Qwen2.5-Coder-1.5B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-3B-Instruct-AWQ', hf_model_id='Qwen/Qwen2.5-Coder-3B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-7B-Instruct-AWQ', hf_model_id='Qwen/Qwen2.5-Coder-7B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-14B-Instruct-AWQ', hf_model_id='Qwen/Qwen2.5-Coder-14B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-32B-Instruct-AWQ', hf_model_id='Qwen/Qwen2.5-Coder-32B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-0.5B-Instruct-GPTQ-Int4', hf_model_id='Qwen/Qwen2.5-Coder-0.5B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-0.5B-Instruct-GPTQ-Int8', hf_model_id='Qwen/Qwen2.5-Coder-0.5B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-1.5B-Instruct-GPTQ-Int4', hf_model_id='Qwen/Qwen2.5-Coder-1.5B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-1.5B-Instruct-GPTQ-Int8', hf_model_id='Qwen/Qwen2.5-Coder-1.5B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-3B-Instruct-GPTQ-Int4', hf_model_id='Qwen/Qwen2.5-Coder-3B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-3B-Instruct-GPTQ-Int8', hf_model_id='Qwen/Qwen2.5-Coder-3B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-7B-Instruct-GPTQ-Int4', hf_model_id='Qwen/Qwen2.5-Coder-7B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-7B-Instruct-GPTQ-Int8', hf_model_id='Qwen/Qwen2.5-Coder-7B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-14B-Instruct-GPTQ-Int4', hf_model_id='Qwen/Qwen2.5-Coder-14B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-14B-Instruct-GPTQ-Int8', hf_model_id='Qwen/Qwen2.5-Coder-14B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-32B-Instruct-GPTQ-Int4', hf_model_id='Qwen/Qwen2.5-Coder-32B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-32B-Instruct-GPTQ-Int8', hf_model_id='Qwen/Qwen2.5-Coder-32B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None)], ignore_patterns=None, requires=None, tags=['coding'])], template='qwen2_5', get_function=<function get_model_tokenizer_with_flash_attn at 0x7f3ff78a7920>, model_arch='llama', architectures=['Qwen2ForCausalLM'], additional_saved_files=[], torch_dtype=None, is_multimodal=False, is_reward=False, task_type=None, ignore_patterns=None, requires=['transformers>=4.37'], tags=[])",
365
+ "model_dir": "/data/minimax-dialogue/users/ruobai/cogito/base_model/Qwen2.5-7B",
366
+ "hub": "<class 'swift.hub.hub.HFHub'>",
367
+ "evaluation_strategy": "epoch",
368
+ "training_args": "Seq2SeqTrainingArguments(output_dir='/data/minimax-dialogue/users/ruobai/swift/model/qwen2p5_7B_ba_1012/v0-20251012-211828', overwrite_output_dir=False, do_train=False, do_eval=True, do_predict=False, eval_strategy=<IntervalStrategy.EPOCH: 'epoch'>, prediction_loss_only=False, per_device_train_batch_size=1, per_device_eval_batch_size=1, per_gpu_train_batch_size=None, per_gpu_eval_batch_size=None, gradient_accumulation_steps=2, eval_accumulation_steps=None, eval_delay=0, torch_empty_cache_steps=None, learning_rate=1e-05, weight_decay=0.05, adam_beta1=0.9, adam_beta2=0.95, adam_epsilon=1e-08, max_grad_norm=1.0, num_train_epochs=4.0, max_steps=-1, lr_scheduler_type=<SchedulerType.COSINE: 'cosine'>, lr_scheduler_kwargs=None, warmup_ratio=0.05, warmup_steps=0, log_level='passive', log_level_replica='warning', log_on_each_node=True, logging_dir='/data/minimax-dialogue/users/ruobai/swift/model/qwen2p5_7B_ba_1012/v0-20251012-211828/runs', logging_strategy=<IntervalStrategy.STEPS: 'steps'>, logging_first_step=True, logging_steps=1, logging_nan_inf_filter=True, save_strategy=<SaveStrategy.EPOCH: 'epoch'>, save_steps=500, save_total_limit=None, save_safetensors=True, save_on_each_node=False, save_only_model=False, restore_callback_states_from_checkpoint=False, no_cuda=False, use_cpu=False, use_mps_device=False, seed=42, data_seed=42, jit_mode_eval=False, use_ipex=False, bf16=True, fp16=False, fp16_opt_level='O1', half_precision_backend='auto', bf16_full_eval=False, fp16_full_eval=False, tf32=None, local_rank=0, ddp_backend='nccl', tpu_num_cores=None, tpu_metrics_debug=False, debug=[], dataloader_drop_last=False, eval_steps=20.0, dataloader_num_workers=48, dataloader_prefetch_factor=10, past_index=-1, run_name='/data/minimax-dialogue/users/ruobai/swift/model/qwen2p5_7B_ba_1012/v0-20251012-211828', disable_tqdm=False, remove_unused_columns=False, label_names=None, load_best_model_at_end=False, metric_for_best_model='loss', greater_is_better=False, ignore_data_skip=False, fsdp=[], fsdp_min_num_params=0, fsdp_config={'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}, tp_size=0, fsdp_transformer_layer_cls_to_wrap=None, accelerator_config=AcceleratorConfig(split_batches=False, dispatch_batches=False, even_batches=True, use_seedable_sampler=True, non_blocking=False, gradient_accumulation_kwargs=None, use_configured_state=False), deepspeed={'fp16': {'enabled': 'auto', 'loss_scale': 0, 'loss_scale_window': 1000, 'initial_scale_power': 16, 'hysteresis': 2, 'min_loss_scale': 1}, 'bf16': {'enabled': 'auto'}, 'zero_optimization': {'stage': 3, 'offload_optimizer': {'device': 'none', 'pin_memory': True}, 'offload_param': {'device': 'none', 'pin_memory': True}, 'overlap_comm': False, 'contiguous_gradients': True, 'sub_group_size': 1000000000.0, 'reduce_bucket_size': 'auto', 'zero_quantized_weights': False, 'zero_quantized_gradients': False, 'stage3_prefetch_bucket_size': 'auto', 'stage3_param_persistence_threshold': 'auto', 'stage3_max_live_parameters': 1000000000.0, 'stage3_max_reuse_distance': 1000000000.0, 'stage3_gather_16bit_weights_on_model_save': True}, 'gradient_accumulation_steps': 'auto', 'gradient_clipping': 'auto', 'steps_per_print': 2000, 'train_batch_size': 'auto', 'train_micro_batch_size_per_gpu': 'auto', 'wall_clock_breakdown': False}, label_smoothing_factor=0.0, optim=<OptimizerNames.ADAMW_TORCH: 'adamw_torch'>, optim_args=None, adafactor=False, group_by_length=False, length_column_name='length', report_to=['wandb'], ddp_find_unused_parameters=None, ddp_bucket_cap_mb=None, ddp_broadcast_buffers=None, dataloader_pin_memory=True, dataloader_persistent_workers=False, skip_memory_metrics=True, use_legacy_prediction_loop=False, push_to_hub=False, resume_from_checkpoint=None, hub_model_id=None, hub_strategy=<HubStrategy.EVERY_SAVE: 'every_save'>, hub_token=None, hub_private_repo=None, hub_always_push=False, gradient_checkpointing=True, gradient_checkpointing_kwargs=None, include_inputs_for_metrics=False, include_for_metrics=[], eval_do_concat_batches=True, fp16_backend='auto', push_to_hub_model_id=None, push_to_hub_organization=None, push_to_hub_token=None, mp_parameters='', auto_find_batch_size=False, full_determinism=False, torchdynamo=None, ray_scope='last', ddp_timeout=1800, torch_compile=False, torch_compile_backend=None, torch_compile_mode=None, include_tokens_per_second=None, include_num_input_tokens_seen=None, neftune_noise_alpha=None, optim_target_modules=None, batch_eval_metrics=False, eval_on_start=False, use_liger_kernel=False, eval_use_gather_object=False, average_tokens_across_devices=None, sortish_sampler=False, predict_with_generate=False, generation_max_length=None, generation_num_beams=None, generation_config=None, check_model=True, acc_strategy='token', train_dataloader_shuffle=True, max_epochs=None, metric_warmup_step=0, fsdp_num=1, acc_steps=1, eval_use_evalscope=False, eval_datasets=[], eval_limit=None, eval_datasets_args=None, eval_generation_config=None, train_type='full', optimizer=None, local_repo_path=None, galore_config=None)"
369
+ }
v0-20251012-211828/checkpoint-224/config.json ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "Qwen2ForCausalLM"
4
+ ],
5
+ "attention_dropout": 0.0,
6
+ "bos_token_id": 151643,
7
+ "eos_token_id": 151643,
8
+ "hidden_act": "silu",
9
+ "hidden_size": 3584,
10
+ "initializer_range": 0.02,
11
+ "intermediate_size": 18944,
12
+ "max_position_embeddings": 131072,
13
+ "max_window_layers": 28,
14
+ "model_type": "qwen2",
15
+ "num_attention_heads": 28,
16
+ "num_hidden_layers": 28,
17
+ "num_key_value_heads": 4,
18
+ "pad_token_id": 151643,
19
+ "rms_norm_eps": 1e-06,
20
+ "rope_scaling": null,
21
+ "rope_theta": 1000000.0,
22
+ "sliding_window": 131072,
23
+ "tie_word_embeddings": false,
24
+ "torch_dtype": "bfloat16",
25
+ "transformers_version": "4.51.3",
26
+ "use_cache": false,
27
+ "use_mrope": false,
28
+ "use_sliding_window": false,
29
+ "vocab_size": 152064
30
+ }
v0-20251012-211828/checkpoint-224/generation_config.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token_id": 151643,
3
+ "eos_token_id": 151643,
4
+ "max_new_tokens": 2048,
5
+ "transformers_version": "4.51.3"
6
+ }
v0-20251012-211828/checkpoint-224/latest ADDED
@@ -0,0 +1 @@
 
 
1
+ global_step224
v0-20251012-211828/checkpoint-224/merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
v0-20251012-211828/checkpoint-224/model-00001-of-00004.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:47e4a50ad6f22212a03f72a90780581c18fc74c61f7a496b2843fca3bc07980f
3
+ size 4877660776
v0-20251012-211828/checkpoint-224/model-00002-of-00004.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:71449c6a8393462bda8cbb108ef372376ac9732605ef73025444a1152a1ae057
3
+ size 4932751008
v0-20251012-211828/checkpoint-224/model-00003-of-00004.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0967793d81e50b062ebcedbe59362c8c569ea38092559427069ce67f86b415c4
3
+ size 4330865200
v0-20251012-211828/checkpoint-224/model-00004-of-00004.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9faafb4490147e060ccd2871de2a1e2582060fce40dfb9c88329ec6f60348b46
3
+ size 1089994880
v0-20251012-211828/checkpoint-224/model.safetensors.index.json ADDED
@@ -0,0 +1,346 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "metadata": {
3
+ "total_size": 15231233024
4
+ },
5
+ "weight_map": {
6
+ "lm_head.weight": "model-00004-of-00004.safetensors",
7
+ "model.embed_tokens.weight": "model-00001-of-00004.safetensors",
8
+ "model.layers.0.input_layernorm.weight": "model-00001-of-00004.safetensors",
9
+ "model.layers.0.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
10
+ "model.layers.0.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
11
+ "model.layers.0.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
12
+ "model.layers.0.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
13
+ "model.layers.0.self_attn.k_proj.bias": "model-00001-of-00004.safetensors",
14
+ "model.layers.0.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
15
+ "model.layers.0.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
16
+ "model.layers.0.self_attn.q_proj.bias": "model-00001-of-00004.safetensors",
17
+ "model.layers.0.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
18
+ "model.layers.0.self_attn.v_proj.bias": "model-00001-of-00004.safetensors",
19
+ "model.layers.0.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
20
+ "model.layers.1.input_layernorm.weight": "model-00001-of-00004.safetensors",
21
+ "model.layers.1.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
22
+ "model.layers.1.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
23
+ "model.layers.1.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
24
+ "model.layers.1.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
25
+ "model.layers.1.self_attn.k_proj.bias": "model-00001-of-00004.safetensors",
26
+ "model.layers.1.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
27
+ "model.layers.1.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
28
+ "model.layers.1.self_attn.q_proj.bias": "model-00001-of-00004.safetensors",
29
+ "model.layers.1.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
30
+ "model.layers.1.self_attn.v_proj.bias": "model-00001-of-00004.safetensors",
31
+ "model.layers.1.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
32
+ "model.layers.10.input_layernorm.weight": "model-00002-of-00004.safetensors",
33
+ "model.layers.10.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
34
+ "model.layers.10.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
35
+ "model.layers.10.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
36
+ "model.layers.10.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
37
+ "model.layers.10.self_attn.k_proj.bias": "model-00002-of-00004.safetensors",
38
+ "model.layers.10.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
39
+ "model.layers.10.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
40
+ "model.layers.10.self_attn.q_proj.bias": "model-00002-of-00004.safetensors",
41
+ "model.layers.10.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
42
+ "model.layers.10.self_attn.v_proj.bias": "model-00002-of-00004.safetensors",
43
+ "model.layers.10.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
44
+ "model.layers.11.input_layernorm.weight": "model-00002-of-00004.safetensors",
45
+ "model.layers.11.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
46
+ "model.layers.11.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
47
+ "model.layers.11.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
48
+ "model.layers.11.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
49
+ "model.layers.11.self_attn.k_proj.bias": "model-00002-of-00004.safetensors",
50
+ "model.layers.11.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
51
+ "model.layers.11.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
52
+ "model.layers.11.self_attn.q_proj.bias": "model-00002-of-00004.safetensors",
53
+ "model.layers.11.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
54
+ "model.layers.11.self_attn.v_proj.bias": "model-00002-of-00004.safetensors",
55
+ "model.layers.11.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
56
+ "model.layers.12.input_layernorm.weight": "model-00002-of-00004.safetensors",
57
+ "model.layers.12.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
58
+ "model.layers.12.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
59
+ "model.layers.12.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
60
+ "model.layers.12.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
61
+ "model.layers.12.self_attn.k_proj.bias": "model-00002-of-00004.safetensors",
62
+ "model.layers.12.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
63
+ "model.layers.12.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
64
+ "model.layers.12.self_attn.q_proj.bias": "model-00002-of-00004.safetensors",
65
+ "model.layers.12.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
66
+ "model.layers.12.self_attn.v_proj.bias": "model-00002-of-00004.safetensors",
67
+ "model.layers.12.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
68
+ "model.layers.13.input_layernorm.weight": "model-00002-of-00004.safetensors",
69
+ "model.layers.13.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
70
+ "model.layers.13.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
71
+ "model.layers.13.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
72
+ "model.layers.13.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
73
+ "model.layers.13.self_attn.k_proj.bias": "model-00002-of-00004.safetensors",
74
+ "model.layers.13.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
75
+ "model.layers.13.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
76
+ "model.layers.13.self_attn.q_proj.bias": "model-00002-of-00004.safetensors",
77
+ "model.layers.13.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
78
+ "model.layers.13.self_attn.v_proj.bias": "model-00002-of-00004.safetensors",
79
+ "model.layers.13.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
80
+ "model.layers.14.input_layernorm.weight": "model-00002-of-00004.safetensors",
81
+ "model.layers.14.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
82
+ "model.layers.14.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
83
+ "model.layers.14.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
84
+ "model.layers.14.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
85
+ "model.layers.14.self_attn.k_proj.bias": "model-00002-of-00004.safetensors",
86
+ "model.layers.14.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
87
+ "model.layers.14.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
88
+ "model.layers.14.self_attn.q_proj.bias": "model-00002-of-00004.safetensors",
89
+ "model.layers.14.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
90
+ "model.layers.14.self_attn.v_proj.bias": "model-00002-of-00004.safetensors",
91
+ "model.layers.14.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
92
+ "model.layers.15.input_layernorm.weight": "model-00002-of-00004.safetensors",
93
+ "model.layers.15.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
94
+ "model.layers.15.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
95
+ "model.layers.15.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
96
+ "model.layers.15.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
97
+ "model.layers.15.self_attn.k_proj.bias": "model-00002-of-00004.safetensors",
98
+ "model.layers.15.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
99
+ "model.layers.15.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
100
+ "model.layers.15.self_attn.q_proj.bias": "model-00002-of-00004.safetensors",
101
+ "model.layers.15.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
102
+ "model.layers.15.self_attn.v_proj.bias": "model-00002-of-00004.safetensors",
103
+ "model.layers.15.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
104
+ "model.layers.16.input_layernorm.weight": "model-00002-of-00004.safetensors",
105
+ "model.layers.16.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
106
+ "model.layers.16.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
107
+ "model.layers.16.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
108
+ "model.layers.16.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
109
+ "model.layers.16.self_attn.k_proj.bias": "model-00002-of-00004.safetensors",
110
+ "model.layers.16.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
111
+ "model.layers.16.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
112
+ "model.layers.16.self_attn.q_proj.bias": "model-00002-of-00004.safetensors",
113
+ "model.layers.16.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
114
+ "model.layers.16.self_attn.v_proj.bias": "model-00002-of-00004.safetensors",
115
+ "model.layers.16.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
116
+ "model.layers.17.input_layernorm.weight": "model-00002-of-00004.safetensors",
117
+ "model.layers.17.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
118
+ "model.layers.17.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
119
+ "model.layers.17.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
120
+ "model.layers.17.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
121
+ "model.layers.17.self_attn.k_proj.bias": "model-00002-of-00004.safetensors",
122
+ "model.layers.17.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
123
+ "model.layers.17.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
124
+ "model.layers.17.self_attn.q_proj.bias": "model-00002-of-00004.safetensors",
125
+ "model.layers.17.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
126
+ "model.layers.17.self_attn.v_proj.bias": "model-00002-of-00004.safetensors",
127
+ "model.layers.17.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
128
+ "model.layers.18.input_layernorm.weight": "model-00003-of-00004.safetensors",
129
+ "model.layers.18.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
130
+ "model.layers.18.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
131
+ "model.layers.18.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
132
+ "model.layers.18.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
133
+ "model.layers.18.self_attn.k_proj.bias": "model-00002-of-00004.safetensors",
134
+ "model.layers.18.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
135
+ "model.layers.18.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
136
+ "model.layers.18.self_attn.q_proj.bias": "model-00002-of-00004.safetensors",
137
+ "model.layers.18.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
138
+ "model.layers.18.self_attn.v_proj.bias": "model-00002-of-00004.safetensors",
139
+ "model.layers.18.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
140
+ "model.layers.19.input_layernorm.weight": "model-00003-of-00004.safetensors",
141
+ "model.layers.19.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
142
+ "model.layers.19.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
143
+ "model.layers.19.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
144
+ "model.layers.19.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
145
+ "model.layers.19.self_attn.k_proj.bias": "model-00003-of-00004.safetensors",
146
+ "model.layers.19.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
147
+ "model.layers.19.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
148
+ "model.layers.19.self_attn.q_proj.bias": "model-00003-of-00004.safetensors",
149
+ "model.layers.19.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
150
+ "model.layers.19.self_attn.v_proj.bias": "model-00003-of-00004.safetensors",
151
+ "model.layers.19.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
152
+ "model.layers.2.input_layernorm.weight": "model-00001-of-00004.safetensors",
153
+ "model.layers.2.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
154
+ "model.layers.2.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
155
+ "model.layers.2.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
156
+ "model.layers.2.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
157
+ "model.layers.2.self_attn.k_proj.bias": "model-00001-of-00004.safetensors",
158
+ "model.layers.2.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
159
+ "model.layers.2.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
160
+ "model.layers.2.self_attn.q_proj.bias": "model-00001-of-00004.safetensors",
161
+ "model.layers.2.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
162
+ "model.layers.2.self_attn.v_proj.bias": "model-00001-of-00004.safetensors",
163
+ "model.layers.2.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
164
+ "model.layers.20.input_layernorm.weight": "model-00003-of-00004.safetensors",
165
+ "model.layers.20.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
166
+ "model.layers.20.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
167
+ "model.layers.20.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
168
+ "model.layers.20.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
169
+ "model.layers.20.self_attn.k_proj.bias": "model-00003-of-00004.safetensors",
170
+ "model.layers.20.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
171
+ "model.layers.20.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
172
+ "model.layers.20.self_attn.q_proj.bias": "model-00003-of-00004.safetensors",
173
+ "model.layers.20.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
174
+ "model.layers.20.self_attn.v_proj.bias": "model-00003-of-00004.safetensors",
175
+ "model.layers.20.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
176
+ "model.layers.21.input_layernorm.weight": "model-00003-of-00004.safetensors",
177
+ "model.layers.21.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
178
+ "model.layers.21.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
179
+ "model.layers.21.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
180
+ "model.layers.21.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
181
+ "model.layers.21.self_attn.k_proj.bias": "model-00003-of-00004.safetensors",
182
+ "model.layers.21.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
183
+ "model.layers.21.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
184
+ "model.layers.21.self_attn.q_proj.bias": "model-00003-of-00004.safetensors",
185
+ "model.layers.21.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
186
+ "model.layers.21.self_attn.v_proj.bias": "model-00003-of-00004.safetensors",
187
+ "model.layers.21.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
188
+ "model.layers.22.input_layernorm.weight": "model-00003-of-00004.safetensors",
189
+ "model.layers.22.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
190
+ "model.layers.22.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
191
+ "model.layers.22.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
192
+ "model.layers.22.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
193
+ "model.layers.22.self_attn.k_proj.bias": "model-00003-of-00004.safetensors",
194
+ "model.layers.22.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
195
+ "model.layers.22.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
196
+ "model.layers.22.self_attn.q_proj.bias": "model-00003-of-00004.safetensors",
197
+ "model.layers.22.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
198
+ "model.layers.22.self_attn.v_proj.bias": "model-00003-of-00004.safetensors",
199
+ "model.layers.22.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
200
+ "model.layers.23.input_layernorm.weight": "model-00003-of-00004.safetensors",
201
+ "model.layers.23.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
202
+ "model.layers.23.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
203
+ "model.layers.23.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
204
+ "model.layers.23.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
205
+ "model.layers.23.self_attn.k_proj.bias": "model-00003-of-00004.safetensors",
206
+ "model.layers.23.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
207
+ "model.layers.23.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
208
+ "model.layers.23.self_attn.q_proj.bias": "model-00003-of-00004.safetensors",
209
+ "model.layers.23.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
210
+ "model.layers.23.self_attn.v_proj.bias": "model-00003-of-00004.safetensors",
211
+ "model.layers.23.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
212
+ "model.layers.24.input_layernorm.weight": "model-00003-of-00004.safetensors",
213
+ "model.layers.24.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
214
+ "model.layers.24.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
215
+ "model.layers.24.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
216
+ "model.layers.24.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
217
+ "model.layers.24.self_attn.k_proj.bias": "model-00003-of-00004.safetensors",
218
+ "model.layers.24.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
219
+ "model.layers.24.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
220
+ "model.layers.24.self_attn.q_proj.bias": "model-00003-of-00004.safetensors",
221
+ "model.layers.24.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
222
+ "model.layers.24.self_attn.v_proj.bias": "model-00003-of-00004.safetensors",
223
+ "model.layers.24.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
224
+ "model.layers.25.input_layernorm.weight": "model-00003-of-00004.safetensors",
225
+ "model.layers.25.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
226
+ "model.layers.25.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
227
+ "model.layers.25.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
228
+ "model.layers.25.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
229
+ "model.layers.25.self_attn.k_proj.bias": "model-00003-of-00004.safetensors",
230
+ "model.layers.25.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
231
+ "model.layers.25.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
232
+ "model.layers.25.self_attn.q_proj.bias": "model-00003-of-00004.safetensors",
233
+ "model.layers.25.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
234
+ "model.layers.25.self_attn.v_proj.bias": "model-00003-of-00004.safetensors",
235
+ "model.layers.25.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
236
+ "model.layers.26.input_layernorm.weight": "model-00003-of-00004.safetensors",
237
+ "model.layers.26.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
238
+ "model.layers.26.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
239
+ "model.layers.26.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
240
+ "model.layers.26.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
241
+ "model.layers.26.self_attn.k_proj.bias": "model-00003-of-00004.safetensors",
242
+ "model.layers.26.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
243
+ "model.layers.26.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
244
+ "model.layers.26.self_attn.q_proj.bias": "model-00003-of-00004.safetensors",
245
+ "model.layers.26.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
246
+ "model.layers.26.self_attn.v_proj.bias": "model-00003-of-00004.safetensors",
247
+ "model.layers.26.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
248
+ "model.layers.27.input_layernorm.weight": "model-00003-of-00004.safetensors",
249
+ "model.layers.27.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
250
+ "model.layers.27.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
251
+ "model.layers.27.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
252
+ "model.layers.27.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
253
+ "model.layers.27.self_attn.k_proj.bias": "model-00003-of-00004.safetensors",
254
+ "model.layers.27.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
255
+ "model.layers.27.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
256
+ "model.layers.27.self_attn.q_proj.bias": "model-00003-of-00004.safetensors",
257
+ "model.layers.27.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
258
+ "model.layers.27.self_attn.v_proj.bias": "model-00003-of-00004.safetensors",
259
+ "model.layers.27.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
260
+ "model.layers.3.input_layernorm.weight": "model-00001-of-00004.safetensors",
261
+ "model.layers.3.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
262
+ "model.layers.3.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
263
+ "model.layers.3.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
264
+ "model.layers.3.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
265
+ "model.layers.3.self_attn.k_proj.bias": "model-00001-of-00004.safetensors",
266
+ "model.layers.3.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
267
+ "model.layers.3.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
268
+ "model.layers.3.self_attn.q_proj.bias": "model-00001-of-00004.safetensors",
269
+ "model.layers.3.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
270
+ "model.layers.3.self_attn.v_proj.bias": "model-00001-of-00004.safetensors",
271
+ "model.layers.3.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
272
+ "model.layers.4.input_layernorm.weight": "model-00001-of-00004.safetensors",
273
+ "model.layers.4.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
274
+ "model.layers.4.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
275
+ "model.layers.4.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
276
+ "model.layers.4.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
277
+ "model.layers.4.self_attn.k_proj.bias": "model-00001-of-00004.safetensors",
278
+ "model.layers.4.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
279
+ "model.layers.4.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
280
+ "model.layers.4.self_attn.q_proj.bias": "model-00001-of-00004.safetensors",
281
+ "model.layers.4.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
282
+ "model.layers.4.self_attn.v_proj.bias": "model-00001-of-00004.safetensors",
283
+ "model.layers.4.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
284
+ "model.layers.5.input_layernorm.weight": "model-00001-of-00004.safetensors",
285
+ "model.layers.5.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
286
+ "model.layers.5.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
287
+ "model.layers.5.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
288
+ "model.layers.5.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
289
+ "model.layers.5.self_attn.k_proj.bias": "model-00001-of-00004.safetensors",
290
+ "model.layers.5.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
291
+ "model.layers.5.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
292
+ "model.layers.5.self_attn.q_proj.bias": "model-00001-of-00004.safetensors",
293
+ "model.layers.5.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
294
+ "model.layers.5.self_attn.v_proj.bias": "model-00001-of-00004.safetensors",
295
+ "model.layers.5.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
296
+ "model.layers.6.input_layernorm.weight": "model-00001-of-00004.safetensors",
297
+ "model.layers.6.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
298
+ "model.layers.6.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
299
+ "model.layers.6.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
300
+ "model.layers.6.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
301
+ "model.layers.6.self_attn.k_proj.bias": "model-00001-of-00004.safetensors",
302
+ "model.layers.6.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
303
+ "model.layers.6.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
304
+ "model.layers.6.self_attn.q_proj.bias": "model-00001-of-00004.safetensors",
305
+ "model.layers.6.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
306
+ "model.layers.6.self_attn.v_proj.bias": "model-00001-of-00004.safetensors",
307
+ "model.layers.6.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
308
+ "model.layers.7.input_layernorm.weight": "model-00001-of-00004.safetensors",
309
+ "model.layers.7.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
310
+ "model.layers.7.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
311
+ "model.layers.7.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
312
+ "model.layers.7.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
313
+ "model.layers.7.self_attn.k_proj.bias": "model-00001-of-00004.safetensors",
314
+ "model.layers.7.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
315
+ "model.layers.7.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
316
+ "model.layers.7.self_attn.q_proj.bias": "model-00001-of-00004.safetensors",
317
+ "model.layers.7.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
318
+ "model.layers.7.self_attn.v_proj.bias": "model-00001-of-00004.safetensors",
319
+ "model.layers.7.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
320
+ "model.layers.8.input_layernorm.weight": "model-00002-of-00004.safetensors",
321
+ "model.layers.8.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
322
+ "model.layers.8.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
323
+ "model.layers.8.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
324
+ "model.layers.8.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
325
+ "model.layers.8.self_attn.k_proj.bias": "model-00001-of-00004.safetensors",
326
+ "model.layers.8.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
327
+ "model.layers.8.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
328
+ "model.layers.8.self_attn.q_proj.bias": "model-00001-of-00004.safetensors",
329
+ "model.layers.8.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
330
+ "model.layers.8.self_attn.v_proj.bias": "model-00001-of-00004.safetensors",
331
+ "model.layers.8.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
332
+ "model.layers.9.input_layernorm.weight": "model-00002-of-00004.safetensors",
333
+ "model.layers.9.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
334
+ "model.layers.9.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
335
+ "model.layers.9.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
336
+ "model.layers.9.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
337
+ "model.layers.9.self_attn.k_proj.bias": "model-00002-of-00004.safetensors",
338
+ "model.layers.9.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
339
+ "model.layers.9.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
340
+ "model.layers.9.self_attn.q_proj.bias": "model-00002-of-00004.safetensors",
341
+ "model.layers.9.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
342
+ "model.layers.9.self_attn.v_proj.bias": "model-00002-of-00004.safetensors",
343
+ "model.layers.9.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
344
+ "model.norm.weight": "model-00003-of-00004.safetensors"
345
+ }
346
+ }
v0-20251012-211828/checkpoint-224/rng_state_0.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2946430e67d53bbdba29ad82934bbaaedc694d31b613744d3a4e1a1e9a46ba8f
3
+ size 15920
v0-20251012-211828/checkpoint-224/rng_state_1.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d2f8ffeab37684493e90f06d39e88dd23930f6d4b22434e761cf24eb66c33624
3
+ size 15920
v0-20251012-211828/checkpoint-224/rng_state_2.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f3e34f4b25d022859f36dc8565e9864ef89df31dee0b6f28ae330b4505432890
3
+ size 15920
v0-20251012-211828/checkpoint-224/rng_state_3.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2c75713cb224376ed36decc995102191568e3ae75807c8df22de6bdfa763f0e1
3
+ size 15920
v0-20251012-211828/checkpoint-224/rng_state_4.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6a045b81eb8aff0c6f04e5c1c068c22b50a29b0c3cd0ceb0f7b0633e087645e2
3
+ size 15920
v0-20251012-211828/checkpoint-224/rng_state_5.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4fbb9337a1f2ddc7522ae749e74101fdc6408581ff2447c6ce01d1264feaa554
3
+ size 15920
v0-20251012-211828/checkpoint-224/rng_state_6.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bd4f55e2919a65fc83570ea3e656aee5c665014b2d4ebb1e757b6604bfa2f91e
3
+ size 15920
v0-20251012-211828/checkpoint-224/rng_state_7.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e742eccb95f66543b690641ccc536fb0f2c7f6d2ad1dce55de41ec2f9bcac81a
3
+ size 15920
v0-20251012-211828/checkpoint-224/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bbd6c9531096989d9e391789dabe461a76afc83be7dae88308783658168c9414
3
+ size 1064
v0-20251012-211828/checkpoint-224/special_tokens_map.json ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "additional_special_tokens": [
3
+ "<|im_start|>",
4
+ "<|im_end|>",
5
+ "<|object_ref_start|>",
6
+ "<|object_ref_end|>",
7
+ "<|box_start|>",
8
+ "<|box_end|>",
9
+ "<|quad_start|>",
10
+ "<|quad_end|>",
11
+ "<|vision_start|>",
12
+ "<|vision_end|>",
13
+ "<|vision_pad|>",
14
+ "<|image_pad|>",
15
+ "<|video_pad|>"
16
+ ],
17
+ "eos_token": {
18
+ "content": "<|endoftext|>",
19
+ "lstrip": false,
20
+ "normalized": false,
21
+ "rstrip": false,
22
+ "single_word": false
23
+ },
24
+ "pad_token": {
25
+ "content": "<|endoftext|>",
26
+ "lstrip": false,
27
+ "normalized": false,
28
+ "rstrip": false,
29
+ "single_word": false
30
+ }
31
+ }
v0-20251012-211828/checkpoint-224/tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9c5ae00e602b8860cbd784ba82a8aa14e8feecec692e7076590d014d7b7fdafa
3
+ size 11421896
v0-20251012-211828/checkpoint-224/tokenizer_config.json ADDED
@@ -0,0 +1,208 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "add_bos_token": false,
3
+ "add_prefix_space": false,
4
+ "added_tokens_decoder": {
5
+ "151643": {
6
+ "content": "<|endoftext|>",
7
+ "lstrip": false,
8
+ "normalized": false,
9
+ "rstrip": false,
10
+ "single_word": false,
11
+ "special": true
12
+ },
13
+ "151644": {
14
+ "content": "<|im_start|>",
15
+ "lstrip": false,
16
+ "normalized": false,
17
+ "rstrip": false,
18
+ "single_word": false,
19
+ "special": true
20
+ },
21
+ "151645": {
22
+ "content": "<|im_end|>",
23
+ "lstrip": false,
24
+ "normalized": false,
25
+ "rstrip": false,
26
+ "single_word": false,
27
+ "special": true
28
+ },
29
+ "151646": {
30
+ "content": "<|object_ref_start|>",
31
+ "lstrip": false,
32
+ "normalized": false,
33
+ "rstrip": false,
34
+ "single_word": false,
35
+ "special": true
36
+ },
37
+ "151647": {
38
+ "content": "<|object_ref_end|>",
39
+ "lstrip": false,
40
+ "normalized": false,
41
+ "rstrip": false,
42
+ "single_word": false,
43
+ "special": true
44
+ },
45
+ "151648": {
46
+ "content": "<|box_start|>",
47
+ "lstrip": false,
48
+ "normalized": false,
49
+ "rstrip": false,
50
+ "single_word": false,
51
+ "special": true
52
+ },
53
+ "151649": {
54
+ "content": "<|box_end|>",
55
+ "lstrip": false,
56
+ "normalized": false,
57
+ "rstrip": false,
58
+ "single_word": false,
59
+ "special": true
60
+ },
61
+ "151650": {
62
+ "content": "<|quad_start|>",
63
+ "lstrip": false,
64
+ "normalized": false,
65
+ "rstrip": false,
66
+ "single_word": false,
67
+ "special": true
68
+ },
69
+ "151651": {
70
+ "content": "<|quad_end|>",
71
+ "lstrip": false,
72
+ "normalized": false,
73
+ "rstrip": false,
74
+ "single_word": false,
75
+ "special": true
76
+ },
77
+ "151652": {
78
+ "content": "<|vision_start|>",
79
+ "lstrip": false,
80
+ "normalized": false,
81
+ "rstrip": false,
82
+ "single_word": false,
83
+ "special": true
84
+ },
85
+ "151653": {
86
+ "content": "<|vision_end|>",
87
+ "lstrip": false,
88
+ "normalized": false,
89
+ "rstrip": false,
90
+ "single_word": false,
91
+ "special": true
92
+ },
93
+ "151654": {
94
+ "content": "<|vision_pad|>",
95
+ "lstrip": false,
96
+ "normalized": false,
97
+ "rstrip": false,
98
+ "single_word": false,
99
+ "special": true
100
+ },
101
+ "151655": {
102
+ "content": "<|image_pad|>",
103
+ "lstrip": false,
104
+ "normalized": false,
105
+ "rstrip": false,
106
+ "single_word": false,
107
+ "special": true
108
+ },
109
+ "151656": {
110
+ "content": "<|video_pad|>",
111
+ "lstrip": false,
112
+ "normalized": false,
113
+ "rstrip": false,
114
+ "single_word": false,
115
+ "special": true
116
+ },
117
+ "151657": {
118
+ "content": "<tool_call>",
119
+ "lstrip": false,
120
+ "normalized": false,
121
+ "rstrip": false,
122
+ "single_word": false,
123
+ "special": false
124
+ },
125
+ "151658": {
126
+ "content": "</tool_call>",
127
+ "lstrip": false,
128
+ "normalized": false,
129
+ "rstrip": false,
130
+ "single_word": false,
131
+ "special": false
132
+ },
133
+ "151659": {
134
+ "content": "<|fim_prefix|>",
135
+ "lstrip": false,
136
+ "normalized": false,
137
+ "rstrip": false,
138
+ "single_word": false,
139
+ "special": false
140
+ },
141
+ "151660": {
142
+ "content": "<|fim_middle|>",
143
+ "lstrip": false,
144
+ "normalized": false,
145
+ "rstrip": false,
146
+ "single_word": false,
147
+ "special": false
148
+ },
149
+ "151661": {
150
+ "content": "<|fim_suffix|>",
151
+ "lstrip": false,
152
+ "normalized": false,
153
+ "rstrip": false,
154
+ "single_word": false,
155
+ "special": false
156
+ },
157
+ "151662": {
158
+ "content": "<|fim_pad|>",
159
+ "lstrip": false,
160
+ "normalized": false,
161
+ "rstrip": false,
162
+ "single_word": false,
163
+ "special": false
164
+ },
165
+ "151663": {
166
+ "content": "<|repo_name|>",
167
+ "lstrip": false,
168
+ "normalized": false,
169
+ "rstrip": false,
170
+ "single_word": false,
171
+ "special": false
172
+ },
173
+ "151664": {
174
+ "content": "<|file_sep|>",
175
+ "lstrip": false,
176
+ "normalized": false,
177
+ "rstrip": false,
178
+ "single_word": false,
179
+ "special": false
180
+ }
181
+ },
182
+ "additional_special_tokens": [
183
+ "<|im_start|>",
184
+ "<|im_end|>",
185
+ "<|object_ref_start|>",
186
+ "<|object_ref_end|>",
187
+ "<|box_start|>",
188
+ "<|box_end|>",
189
+ "<|quad_start|>",
190
+ "<|quad_end|>",
191
+ "<|vision_start|>",
192
+ "<|vision_end|>",
193
+ "<|vision_pad|>",
194
+ "<|image_pad|>",
195
+ "<|video_pad|>"
196
+ ],
197
+ "bos_token": null,
198
+ "chat_template": "{%- if tools %}\n {{- '<|im_start|>system\\n' }}\n {%- if messages[0]['role'] == 'system' %}\n {{- messages[0]['content'] }}\n {%- else %}\n {{- 'You are a helpful assistant.' }}\n {%- endif %}\n {{- \"\\n\\n# Tools\\n\\nYou may call one or more functions to assist with the user query.\\n\\nYou are provided with function signatures within <tools></tools> XML tags:\\n<tools>\" }}\n {%- for tool in tools %}\n {{- \"\\n\" }}\n {{- tool | tojson }}\n {%- endfor %}\n {{- \"\\n</tools>\\n\\nFor each function call, return a json object with function name and arguments within <tool_call></tool_call> XML tags:\\n<tool_call>\\n{\\\"name\\\": <function-name>, \\\"arguments\\\": <args-json-object>}\\n</tool_call><|im_end|>\\n\" }}\n{%- else %}\n {%- if messages[0]['role'] == 'system' %}\n {{- '<|im_start|>system\\n' + messages[0]['content'] + '<|im_end|>\\n' }}\n {%- else %}\n {{- '<|im_start|>system\\nYou are a helpful assistant.<|im_end|>\\n' }}\n {%- endif %}\n{%- endif %}\n{%- for message in messages %}\n {%- if (message.role == \"user\") or (message.role == \"system\" and not loop.first) or (message.role == \"assistant\" and not message.tool_calls) %}\n {{- '<|im_start|>' + message.role + '\\n' + message.content + '<|im_end|>' + '\\n' }}\n {%- elif message.role == \"assistant\" %}\n {{- '<|im_start|>' + message.role }}\n {%- if message.content %}\n {{- '\\n' + message.content }}\n {%- endif %}\n {%- for tool_call in message.tool_calls %}\n {%- if tool_call.function is defined %}\n {%- set tool_call = tool_call.function %}\n {%- endif %}\n {{- '\\n<tool_call>\\n{\"name\": \"' }}\n {{- tool_call.name }}\n {{- '\", \"arguments\": ' }}\n {{- tool_call.arguments | tojson }}\n {{- '}\\n</tool_call>' }}\n {%- endfor %}\n {{- '<|im_end|>\\n' }}\n {%- elif message.role == \"tool\" %}\n {%- if (loop.index0 == 0) or (messages[loop.index0 - 1].role != \"tool\") %}\n {{- '<|im_start|>user' }}\n {%- endif %}\n {{- '\\n<tool_response>\\n' }}\n {{- message.content }}\n {{- '\\n</tool_response>' }}\n {%- if loop.last or (messages[loop.index0 + 1].role != \"tool\") %}\n {{- '<|im_end|>\\n' }}\n {%- endif %}\n {%- endif %}\n{%- endfor %}\n{%- if add_generation_prompt %}\n {{- '<|im_start|>assistant\\n' }}\n{%- endif %}\n",
199
+ "clean_up_tokenization_spaces": false,
200
+ "eos_token": "<|endoftext|>",
201
+ "errors": "replace",
202
+ "extra_special_tokens": {},
203
+ "model_max_length": 131072,
204
+ "pad_token": "<|endoftext|>",
205
+ "split_special_tokens": false,
206
+ "tokenizer_class": "Qwen2Tokenizer",
207
+ "unk_token": null
208
+ }
v0-20251012-211828/checkpoint-224/trainer_state.json ADDED
@@ -0,0 +1,2283 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": 224,
3
+ "best_metric": 0.36971456,
4
+ "best_model_checkpoint": "/data/minimax-dialogue/users/ruobai/swift/model/qwen2p5_7B_ba_1012/v0-20251012-211828/checkpoint-224",
5
+ "epoch": 1.0,
6
+ "eval_steps": 20.0,
7
+ "global_step": 224,
8
+ "is_hyper_param_search": false,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "epoch": 0.004464285714285714,
14
+ "grad_norm": 607.0161258405667,
15
+ "learning_rate": 2.2222222222222224e-07,
16
+ "loss": 1.2842669486999512,
17
+ "memory(GiB)": 73.22,
18
+ "step": 1,
19
+ "token_acc": 0.7433547204399633,
20
+ "train_speed(iter/s)": 0.074483
21
+ },
22
+ {
23
+ "epoch": 0.008928571428571428,
24
+ "grad_norm": 416.31840864440954,
25
+ "learning_rate": 4.444444444444445e-07,
26
+ "loss": 1.3240495920181274,
27
+ "memory(GiB)": 73.22,
28
+ "step": 2,
29
+ "token_acc": 0.6602722772277227,
30
+ "train_speed(iter/s)": 0.112796
31
+ },
32
+ {
33
+ "epoch": 0.013392857142857142,
34
+ "grad_norm": 160.17360140132547,
35
+ "learning_rate": 6.666666666666667e-07,
36
+ "loss": 1.2871371507644653,
37
+ "memory(GiB)": 73.22,
38
+ "step": 3,
39
+ "token_acc": 0.6800364630811303,
40
+ "train_speed(iter/s)": 0.136364
41
+ },
42
+ {
43
+ "epoch": 0.017857142857142856,
44
+ "grad_norm": 450.37534576513326,
45
+ "learning_rate": 8.88888888888889e-07,
46
+ "loss": 1.2711200714111328,
47
+ "memory(GiB)": 100.82,
48
+ "step": 4,
49
+ "token_acc": 0.7110352673492605,
50
+ "train_speed(iter/s)": 0.15202
51
+ },
52
+ {
53
+ "epoch": 0.022321428571428572,
54
+ "grad_norm": 442.68256941397544,
55
+ "learning_rate": 1.111111111111111e-06,
56
+ "loss": 1.2603538036346436,
57
+ "memory(GiB)": 100.82,
58
+ "step": 5,
59
+ "token_acc": 0.7108695652173913,
60
+ "train_speed(iter/s)": 0.161769
61
+ },
62
+ {
63
+ "epoch": 0.026785714285714284,
64
+ "grad_norm": 290.49695787962474,
65
+ "learning_rate": 1.3333333333333334e-06,
66
+ "loss": 1.2085299491882324,
67
+ "memory(GiB)": 100.82,
68
+ "step": 6,
69
+ "token_acc": 0.7053445850914205,
70
+ "train_speed(iter/s)": 0.170606
71
+ },
72
+ {
73
+ "epoch": 0.03125,
74
+ "grad_norm": 122.03536728572118,
75
+ "learning_rate": 1.5555555555555558e-06,
76
+ "loss": 1.1946239471435547,
77
+ "memory(GiB)": 100.82,
78
+ "step": 7,
79
+ "token_acc": 0.7184265010351967,
80
+ "train_speed(iter/s)": 0.177314
81
+ },
82
+ {
83
+ "epoch": 0.03571428571428571,
84
+ "grad_norm": 48.184047801314975,
85
+ "learning_rate": 1.777777777777778e-06,
86
+ "loss": 1.1202031373977661,
87
+ "memory(GiB)": 100.82,
88
+ "step": 8,
89
+ "token_acc": 0.692090395480226,
90
+ "train_speed(iter/s)": 0.182678
91
+ },
92
+ {
93
+ "epoch": 0.04017857142857143,
94
+ "grad_norm": 64.05149006299645,
95
+ "learning_rate": 2.0000000000000003e-06,
96
+ "loss": 1.058239459991455,
97
+ "memory(GiB)": 100.82,
98
+ "step": 9,
99
+ "token_acc": 0.735609756097561,
100
+ "train_speed(iter/s)": 0.186756
101
+ },
102
+ {
103
+ "epoch": 0.044642857142857144,
104
+ "grad_norm": 32.35823954189375,
105
+ "learning_rate": 2.222222222222222e-06,
106
+ "loss": 1.0513522624969482,
107
+ "memory(GiB)": 128.42,
108
+ "step": 10,
109
+ "token_acc": 0.7422945205479452,
110
+ "train_speed(iter/s)": 0.190564
111
+ },
112
+ {
113
+ "epoch": 0.049107142857142856,
114
+ "grad_norm": 33.455251112438354,
115
+ "learning_rate": 2.4444444444444447e-06,
116
+ "loss": 0.8908174633979797,
117
+ "memory(GiB)": 128.42,
118
+ "step": 11,
119
+ "token_acc": 0.7462585034013606,
120
+ "train_speed(iter/s)": 0.19371
121
+ },
122
+ {
123
+ "epoch": 0.05357142857142857,
124
+ "grad_norm": 20.237589968447807,
125
+ "learning_rate": 2.666666666666667e-06,
126
+ "loss": 0.9060617089271545,
127
+ "memory(GiB)": 128.42,
128
+ "step": 12,
129
+ "token_acc": 0.7580266249021144,
130
+ "train_speed(iter/s)": 0.196325
131
+ },
132
+ {
133
+ "epoch": 0.05803571428571429,
134
+ "grad_norm": 19.104830741854588,
135
+ "learning_rate": 2.888888888888889e-06,
136
+ "loss": 0.8370632529258728,
137
+ "memory(GiB)": 128.42,
138
+ "step": 13,
139
+ "token_acc": 0.7583454281567489,
140
+ "train_speed(iter/s)": 0.198756
141
+ },
142
+ {
143
+ "epoch": 0.0625,
144
+ "grad_norm": 8.07005348419444,
145
+ "learning_rate": 3.1111111111111116e-06,
146
+ "loss": 0.7916728258132935,
147
+ "memory(GiB)": 128.42,
148
+ "step": 14,
149
+ "token_acc": 0.7912087912087912,
150
+ "train_speed(iter/s)": 0.200858
151
+ },
152
+ {
153
+ "epoch": 0.06696428571428571,
154
+ "grad_norm": 9.110633598653314,
155
+ "learning_rate": 3.3333333333333333e-06,
156
+ "loss": 0.7699524164199829,
157
+ "memory(GiB)": 128.42,
158
+ "step": 15,
159
+ "token_acc": 0.7777777777777778,
160
+ "train_speed(iter/s)": 0.202712
161
+ },
162
+ {
163
+ "epoch": 0.07142857142857142,
164
+ "grad_norm": 5.340593386565067,
165
+ "learning_rate": 3.555555555555556e-06,
166
+ "loss": 0.7363409996032715,
167
+ "memory(GiB)": 128.42,
168
+ "step": 16,
169
+ "token_acc": 0.7858439201451906,
170
+ "train_speed(iter/s)": 0.204218
171
+ },
172
+ {
173
+ "epoch": 0.07589285714285714,
174
+ "grad_norm": 9.109649268071344,
175
+ "learning_rate": 3.777777777777778e-06,
176
+ "loss": 0.7368918657302856,
177
+ "memory(GiB)": 128.42,
178
+ "step": 17,
179
+ "token_acc": 0.7816216216216216,
180
+ "train_speed(iter/s)": 0.205652
181
+ },
182
+ {
183
+ "epoch": 0.08035714285714286,
184
+ "grad_norm": 11.131867979943577,
185
+ "learning_rate": 4.000000000000001e-06,
186
+ "loss": 0.6874697804450989,
187
+ "memory(GiB)": 128.42,
188
+ "step": 18,
189
+ "token_acc": 0.7817299919159256,
190
+ "train_speed(iter/s)": 0.20693
191
+ },
192
+ {
193
+ "epoch": 0.08482142857142858,
194
+ "grad_norm": 4.574035974729758,
195
+ "learning_rate": 4.222222222222223e-06,
196
+ "loss": 0.6693994998931885,
197
+ "memory(GiB)": 128.42,
198
+ "step": 19,
199
+ "token_acc": 0.8025659301496793,
200
+ "train_speed(iter/s)": 0.208049
201
+ },
202
+ {
203
+ "epoch": 0.08928571428571429,
204
+ "grad_norm": 4.387355536169233,
205
+ "learning_rate": 4.444444444444444e-06,
206
+ "loss": 0.6153415441513062,
207
+ "memory(GiB)": 128.42,
208
+ "step": 20,
209
+ "token_acc": 0.8143491124260355,
210
+ "train_speed(iter/s)": 0.20903
211
+ },
212
+ {
213
+ "epoch": 0.09375,
214
+ "grad_norm": 5.155423473676148,
215
+ "learning_rate": 4.666666666666667e-06,
216
+ "loss": 0.6413918137550354,
217
+ "memory(GiB)": 128.42,
218
+ "step": 21,
219
+ "token_acc": 0.8059581320450886,
220
+ "train_speed(iter/s)": 0.209872
221
+ },
222
+ {
223
+ "epoch": 0.09821428571428571,
224
+ "grad_norm": 4.52090458658663,
225
+ "learning_rate": 4.888888888888889e-06,
226
+ "loss": 0.6642692685127258,
227
+ "memory(GiB)": 128.42,
228
+ "step": 22,
229
+ "token_acc": 0.8458274398868458,
230
+ "train_speed(iter/s)": 0.210769
231
+ },
232
+ {
233
+ "epoch": 0.10267857142857142,
234
+ "grad_norm": 4.0676391904396105,
235
+ "learning_rate": 5.1111111111111115e-06,
236
+ "loss": 0.612501859664917,
237
+ "memory(GiB)": 128.42,
238
+ "step": 23,
239
+ "token_acc": 0.8329896907216495,
240
+ "train_speed(iter/s)": 0.211563
241
+ },
242
+ {
243
+ "epoch": 0.10714285714285714,
244
+ "grad_norm": 9.05605916321185,
245
+ "learning_rate": 5.333333333333334e-06,
246
+ "loss": 0.5991834402084351,
247
+ "memory(GiB)": 128.42,
248
+ "step": 24,
249
+ "token_acc": 0.820136852394917,
250
+ "train_speed(iter/s)": 0.212389
251
+ },
252
+ {
253
+ "epoch": 0.11160714285714286,
254
+ "grad_norm": 4.0651164644201945,
255
+ "learning_rate": 5.555555555555557e-06,
256
+ "loss": 0.5781806707382202,
257
+ "memory(GiB)": 128.42,
258
+ "step": 25,
259
+ "token_acc": 0.8080684596577017,
260
+ "train_speed(iter/s)": 0.213071
261
+ },
262
+ {
263
+ "epoch": 0.11607142857142858,
264
+ "grad_norm": 8.002751969235684,
265
+ "learning_rate": 5.777777777777778e-06,
266
+ "loss": 0.572704553604126,
267
+ "memory(GiB)": 128.42,
268
+ "step": 26,
269
+ "token_acc": 0.7769130998702983,
270
+ "train_speed(iter/s)": 0.213503
271
+ },
272
+ {
273
+ "epoch": 0.12053571428571429,
274
+ "grad_norm": 4.262323152935433,
275
+ "learning_rate": 6e-06,
276
+ "loss": 0.5535041689872742,
277
+ "memory(GiB)": 128.42,
278
+ "step": 27,
279
+ "token_acc": 0.8503253796095445,
280
+ "train_speed(iter/s)": 0.214104
281
+ },
282
+ {
283
+ "epoch": 0.125,
284
+ "grad_norm": 3.966702275668715,
285
+ "learning_rate": 6.222222222222223e-06,
286
+ "loss": 0.6064645051956177,
287
+ "memory(GiB)": 128.42,
288
+ "step": 28,
289
+ "token_acc": 0.8506348020911128,
290
+ "train_speed(iter/s)": 0.214658
291
+ },
292
+ {
293
+ "epoch": 0.12946428571428573,
294
+ "grad_norm": 3.8567123289596172,
295
+ "learning_rate": 6.444444444444445e-06,
296
+ "loss": 0.5491228103637695,
297
+ "memory(GiB)": 128.42,
298
+ "step": 29,
299
+ "token_acc": 0.8545454545454545,
300
+ "train_speed(iter/s)": 0.215116
301
+ },
302
+ {
303
+ "epoch": 0.13392857142857142,
304
+ "grad_norm": 3.710395065072247,
305
+ "learning_rate": 6.666666666666667e-06,
306
+ "loss": 0.5767688751220703,
307
+ "memory(GiB)": 128.42,
308
+ "step": 30,
309
+ "token_acc": 0.84375,
310
+ "train_speed(iter/s)": 0.215602
311
+ },
312
+ {
313
+ "epoch": 0.13839285714285715,
314
+ "grad_norm": 4.281979952710429,
315
+ "learning_rate": 6.88888888888889e-06,
316
+ "loss": 0.5191075205802917,
317
+ "memory(GiB)": 128.42,
318
+ "step": 31,
319
+ "token_acc": 0.8571428571428571,
320
+ "train_speed(iter/s)": 0.216142
321
+ },
322
+ {
323
+ "epoch": 0.14285714285714285,
324
+ "grad_norm": 3.7571532580897102,
325
+ "learning_rate": 7.111111111111112e-06,
326
+ "loss": 0.5514931082725525,
327
+ "memory(GiB)": 128.42,
328
+ "step": 32,
329
+ "token_acc": 0.8145224940805051,
330
+ "train_speed(iter/s)": 0.216598
331
+ },
332
+ {
333
+ "epoch": 0.14732142857142858,
334
+ "grad_norm": 3.597514204800206,
335
+ "learning_rate": 7.333333333333333e-06,
336
+ "loss": 0.5634205341339111,
337
+ "memory(GiB)": 128.42,
338
+ "step": 33,
339
+ "token_acc": 0.8357963875205254,
340
+ "train_speed(iter/s)": 0.217011
341
+ },
342
+ {
343
+ "epoch": 0.15178571428571427,
344
+ "grad_norm": 3.511624738132458,
345
+ "learning_rate": 7.555555555555556e-06,
346
+ "loss": 0.5661548376083374,
347
+ "memory(GiB)": 128.42,
348
+ "step": 34,
349
+ "token_acc": 0.8533635676492819,
350
+ "train_speed(iter/s)": 0.217375
351
+ },
352
+ {
353
+ "epoch": 0.15625,
354
+ "grad_norm": 3.5093325994826454,
355
+ "learning_rate": 7.77777777777778e-06,
356
+ "loss": 0.5768500566482544,
357
+ "memory(GiB)": 128.42,
358
+ "step": 35,
359
+ "token_acc": 0.8479952124476361,
360
+ "train_speed(iter/s)": 0.217665
361
+ },
362
+ {
363
+ "epoch": 0.16071428571428573,
364
+ "grad_norm": 3.7955307572836268,
365
+ "learning_rate": 8.000000000000001e-06,
366
+ "loss": 0.5004974007606506,
367
+ "memory(GiB)": 128.42,
368
+ "step": 36,
369
+ "token_acc": 0.8147471054235222,
370
+ "train_speed(iter/s)": 0.217992
371
+ },
372
+ {
373
+ "epoch": 0.16517857142857142,
374
+ "grad_norm": 3.4615768583761954,
375
+ "learning_rate": 8.222222222222222e-06,
376
+ "loss": 0.5323991775512695,
377
+ "memory(GiB)": 128.42,
378
+ "step": 37,
379
+ "token_acc": 0.8320715760495526,
380
+ "train_speed(iter/s)": 0.218325
381
+ },
382
+ {
383
+ "epoch": 0.16964285714285715,
384
+ "grad_norm": 3.3517149428315505,
385
+ "learning_rate": 8.444444444444446e-06,
386
+ "loss": 0.5705447196960449,
387
+ "memory(GiB)": 128.42,
388
+ "step": 38,
389
+ "token_acc": 0.8195197521301317,
390
+ "train_speed(iter/s)": 0.21859
391
+ },
392
+ {
393
+ "epoch": 0.17410714285714285,
394
+ "grad_norm": 3.579209602352775,
395
+ "learning_rate": 8.666666666666668e-06,
396
+ "loss": 0.5391894578933716,
397
+ "memory(GiB)": 128.42,
398
+ "step": 39,
399
+ "token_acc": 0.8760736196319019,
400
+ "train_speed(iter/s)": 0.218783
401
+ },
402
+ {
403
+ "epoch": 0.17857142857142858,
404
+ "grad_norm": 3.4898869370386243,
405
+ "learning_rate": 8.888888888888888e-06,
406
+ "loss": 0.49445533752441406,
407
+ "memory(GiB)": 128.42,
408
+ "step": 40,
409
+ "token_acc": 0.883495145631068,
410
+ "train_speed(iter/s)": 0.21908
411
+ },
412
+ {
413
+ "epoch": 0.18303571428571427,
414
+ "grad_norm": 8.989941754990463,
415
+ "learning_rate": 9.111111111111112e-06,
416
+ "loss": 0.49502032995224,
417
+ "memory(GiB)": 128.42,
418
+ "step": 41,
419
+ "token_acc": 0.8318452380952381,
420
+ "train_speed(iter/s)": 0.219326
421
+ },
422
+ {
423
+ "epoch": 0.1875,
424
+ "grad_norm": 3.592536913293317,
425
+ "learning_rate": 9.333333333333334e-06,
426
+ "loss": 0.5180964469909668,
427
+ "memory(GiB)": 128.42,
428
+ "step": 42,
429
+ "token_acc": 0.8248995983935743,
430
+ "train_speed(iter/s)": 0.21958
431
+ },
432
+ {
433
+ "epoch": 0.19196428571428573,
434
+ "grad_norm": 3.3595948687960897,
435
+ "learning_rate": 9.555555555555556e-06,
436
+ "loss": 0.48923468589782715,
437
+ "memory(GiB)": 128.42,
438
+ "step": 43,
439
+ "token_acc": 0.8549222797927462,
440
+ "train_speed(iter/s)": 0.219825
441
+ },
442
+ {
443
+ "epoch": 0.19642857142857142,
444
+ "grad_norm": 3.254464883159724,
445
+ "learning_rate": 9.777777777777779e-06,
446
+ "loss": 0.4752197861671448,
447
+ "memory(GiB)": 128.42,
448
+ "step": 44,
449
+ "token_acc": 0.8691341705221415,
450
+ "train_speed(iter/s)": 0.220006
451
+ },
452
+ {
453
+ "epoch": 0.20089285714285715,
454
+ "grad_norm": 3.4895034998885666,
455
+ "learning_rate": 1e-05,
456
+ "loss": 0.46786195039749146,
457
+ "memory(GiB)": 128.42,
458
+ "step": 45,
459
+ "token_acc": 0.8913894324853229,
460
+ "train_speed(iter/s)": 0.220218
461
+ },
462
+ {
463
+ "epoch": 0.20535714285714285,
464
+ "grad_norm": 3.536205201848946,
465
+ "learning_rate": 9.999965929371846e-06,
466
+ "loss": 0.490744411945343,
467
+ "memory(GiB)": 128.42,
468
+ "step": 46,
469
+ "token_acc": 0.889745566692367,
470
+ "train_speed(iter/s)": 0.220409
471
+ },
472
+ {
473
+ "epoch": 0.20982142857142858,
474
+ "grad_norm": 3.4060931659065465,
475
+ "learning_rate": 9.999863717951704e-06,
476
+ "loss": 0.4735819101333618,
477
+ "memory(GiB)": 128.42,
478
+ "step": 47,
479
+ "token_acc": 0.839968152866242,
480
+ "train_speed(iter/s)": 0.220607
481
+ },
482
+ {
483
+ "epoch": 0.21428571428571427,
484
+ "grad_norm": 3.5259252637192753,
485
+ "learning_rate": 9.999693367132534e-06,
486
+ "loss": 0.4596032202243805,
487
+ "memory(GiB)": 128.42,
488
+ "step": 48,
489
+ "token_acc": 0.8322520852641334,
490
+ "train_speed(iter/s)": 0.220805
491
+ },
492
+ {
493
+ "epoch": 0.21875,
494
+ "grad_norm": 3.342810881695556,
495
+ "learning_rate": 9.999454879235926e-06,
496
+ "loss": 0.4986742436885834,
497
+ "memory(GiB)": 128.42,
498
+ "step": 49,
499
+ "token_acc": 0.8661538461538462,
500
+ "train_speed(iter/s)": 0.221017
501
+ },
502
+ {
503
+ "epoch": 0.22321428571428573,
504
+ "grad_norm": 3.4094938255821097,
505
+ "learning_rate": 9.99914825751205e-06,
506
+ "loss": 0.45326530933380127,
507
+ "memory(GiB)": 128.42,
508
+ "step": 50,
509
+ "token_acc": 0.8617771509167842,
510
+ "train_speed(iter/s)": 0.221194
511
+ },
512
+ {
513
+ "epoch": 0.22767857142857142,
514
+ "grad_norm": 3.366606187311854,
515
+ "learning_rate": 9.998773506139622e-06,
516
+ "loss": 0.49953100085258484,
517
+ "memory(GiB)": 128.42,
518
+ "step": 51,
519
+ "token_acc": 0.8632010943912448,
520
+ "train_speed(iter/s)": 0.221244
521
+ },
522
+ {
523
+ "epoch": 0.23214285714285715,
524
+ "grad_norm": 3.342945040893082,
525
+ "learning_rate": 9.998330630225853e-06,
526
+ "loss": 0.4754818081855774,
527
+ "memory(GiB)": 128.42,
528
+ "step": 52,
529
+ "token_acc": 0.8538116591928251,
530
+ "train_speed(iter/s)": 0.221426
531
+ },
532
+ {
533
+ "epoch": 0.23660714285714285,
534
+ "grad_norm": 3.2984275728119874,
535
+ "learning_rate": 9.99781963580636e-06,
536
+ "loss": 0.4748860001564026,
537
+ "memory(GiB)": 128.42,
538
+ "step": 53,
539
+ "token_acc": 0.8231116849580374,
540
+ "train_speed(iter/s)": 0.221635
541
+ },
542
+ {
543
+ "epoch": 0.24107142857142858,
544
+ "grad_norm": 3.1307130443222406,
545
+ "learning_rate": 9.997240529845112e-06,
546
+ "loss": 0.42877909541130066,
547
+ "memory(GiB)": 128.42,
548
+ "step": 54,
549
+ "token_acc": 0.8777173913043478,
550
+ "train_speed(iter/s)": 0.221813
551
+ },
552
+ {
553
+ "epoch": 0.24553571428571427,
554
+ "grad_norm": 3.1327655583720126,
555
+ "learning_rate": 9.996593320234302e-06,
556
+ "loss": 0.4814930856227875,
557
+ "memory(GiB)": 128.42,
558
+ "step": 55,
559
+ "token_acc": 0.865686274509804,
560
+ "train_speed(iter/s)": 0.221971
561
+ },
562
+ {
563
+ "epoch": 0.25,
564
+ "grad_norm": 3.0672140778239676,
565
+ "learning_rate": 9.995878015794271e-06,
566
+ "loss": 0.4632089138031006,
567
+ "memory(GiB)": 128.42,
568
+ "step": 56,
569
+ "token_acc": 0.8444940476190477,
570
+ "train_speed(iter/s)": 0.222117
571
+ },
572
+ {
573
+ "epoch": 0.2544642857142857,
574
+ "grad_norm": 3.4023664393185724,
575
+ "learning_rate": 9.995094626273365e-06,
576
+ "loss": 0.501305103302002,
577
+ "memory(GiB)": 128.42,
578
+ "step": 57,
579
+ "token_acc": 0.8551660516605166,
580
+ "train_speed(iter/s)": 0.222231
581
+ },
582
+ {
583
+ "epoch": 0.25892857142857145,
584
+ "grad_norm": 2.908553060683868,
585
+ "learning_rate": 9.994243162347814e-06,
586
+ "loss": 0.41847342252731323,
587
+ "memory(GiB)": 128.42,
588
+ "step": 58,
589
+ "token_acc": 0.8760233918128655,
590
+ "train_speed(iter/s)": 0.222366
591
+ },
592
+ {
593
+ "epoch": 0.26339285714285715,
594
+ "grad_norm": 2.808351908407882,
595
+ "learning_rate": 9.993323635621582e-06,
596
+ "loss": 0.47732818126678467,
597
+ "memory(GiB)": 128.42,
598
+ "step": 59,
599
+ "token_acc": 0.8490945674044266,
600
+ "train_speed(iter/s)": 0.222388
601
+ },
602
+ {
603
+ "epoch": 0.26785714285714285,
604
+ "grad_norm": 2.7565786085479274,
605
+ "learning_rate": 9.992336058626212e-06,
606
+ "loss": 0.4374464452266693,
607
+ "memory(GiB)": 128.42,
608
+ "step": 60,
609
+ "token_acc": 0.8333333333333334,
610
+ "train_speed(iter/s)": 0.222532
611
+ },
612
+ {
613
+ "epoch": 0.27232142857142855,
614
+ "grad_norm": 2.500488866173154,
615
+ "learning_rate": 9.991280444820647e-06,
616
+ "loss": 0.4108882546424866,
617
+ "memory(GiB)": 128.42,
618
+ "step": 61,
619
+ "token_acc": 0.8650176678445229,
620
+ "train_speed(iter/s)": 0.222682
621
+ },
622
+ {
623
+ "epoch": 0.2767857142857143,
624
+ "grad_norm": 2.3349742419431676,
625
+ "learning_rate": 9.990156808591064e-06,
626
+ "loss": 0.43476420640945435,
627
+ "memory(GiB)": 128.42,
628
+ "step": 62,
629
+ "token_acc": 0.8691099476439791,
630
+ "train_speed(iter/s)": 0.222807
631
+ },
632
+ {
633
+ "epoch": 0.28125,
634
+ "grad_norm": 2.2421342964515407,
635
+ "learning_rate": 9.988965165250654e-06,
636
+ "loss": 0.4363088011741638,
637
+ "memory(GiB)": 128.42,
638
+ "step": 63,
639
+ "token_acc": 0.8770161290322581,
640
+ "train_speed(iter/s)": 0.222953
641
+ },
642
+ {
643
+ "epoch": 0.2857142857142857,
644
+ "grad_norm": 2.314693697676626,
645
+ "learning_rate": 9.987705531039434e-06,
646
+ "loss": 0.48462244868278503,
647
+ "memory(GiB)": 128.42,
648
+ "step": 64,
649
+ "token_acc": 0.8437990580847724,
650
+ "train_speed(iter/s)": 0.223033
651
+ },
652
+ {
653
+ "epoch": 0.29017857142857145,
654
+ "grad_norm": 2.404628201722386,
655
+ "learning_rate": 9.986377923124016e-06,
656
+ "loss": 0.46896690130233765,
657
+ "memory(GiB)": 128.42,
658
+ "step": 65,
659
+ "token_acc": 0.8426601784266018,
660
+ "train_speed(iter/s)": 0.223118
661
+ },
662
+ {
663
+ "epoch": 0.29464285714285715,
664
+ "grad_norm": 2.4623641287897224,
665
+ "learning_rate": 9.984982359597373e-06,
666
+ "loss": 0.43249690532684326,
667
+ "memory(GiB)": 128.42,
668
+ "step": 66,
669
+ "token_acc": 0.8550592525068368,
670
+ "train_speed(iter/s)": 0.223248
671
+ },
672
+ {
673
+ "epoch": 0.29910714285714285,
674
+ "grad_norm": 2.227239059437109,
675
+ "learning_rate": 9.983518859478597e-06,
676
+ "loss": 0.42663565278053284,
677
+ "memory(GiB)": 128.42,
678
+ "step": 67,
679
+ "token_acc": 0.8267526188557615,
680
+ "train_speed(iter/s)": 0.223364
681
+ },
682
+ {
683
+ "epoch": 0.30357142857142855,
684
+ "grad_norm": 2.152719299561224,
685
+ "learning_rate": 9.981987442712634e-06,
686
+ "loss": 0.42308685183525085,
687
+ "memory(GiB)": 128.42,
688
+ "step": 68,
689
+ "token_acc": 0.8523769808173478,
690
+ "train_speed(iter/s)": 0.223446
691
+ },
692
+ {
693
+ "epoch": 0.3080357142857143,
694
+ "grad_norm": 2.128383011723548,
695
+ "learning_rate": 9.980388130170015e-06,
696
+ "loss": 0.4897618293762207,
697
+ "memory(GiB)": 128.42,
698
+ "step": 69,
699
+ "token_acc": 0.8330404217926186,
700
+ "train_speed(iter/s)": 0.223543
701
+ },
702
+ {
703
+ "epoch": 0.3125,
704
+ "grad_norm": 2.058851793564121,
705
+ "learning_rate": 9.978720943646578e-06,
706
+ "loss": 0.4273832440376282,
707
+ "memory(GiB)": 128.42,
708
+ "step": 70,
709
+ "token_acc": 0.8670212765957447,
710
+ "train_speed(iter/s)": 0.223628
711
+ },
712
+ {
713
+ "epoch": 0.3169642857142857,
714
+ "grad_norm": 2.152936819995939,
715
+ "learning_rate": 9.976985905863155e-06,
716
+ "loss": 0.4190593361854553,
717
+ "memory(GiB)": 128.42,
718
+ "step": 71,
719
+ "token_acc": 0.8797145769622834,
720
+ "train_speed(iter/s)": 0.223737
721
+ },
722
+ {
723
+ "epoch": 0.32142857142857145,
724
+ "grad_norm": 2.021590225323182,
725
+ "learning_rate": 9.975183040465279e-06,
726
+ "loss": 0.3998218774795532,
727
+ "memory(GiB)": 128.42,
728
+ "step": 72,
729
+ "token_acc": 0.8594507269789984,
730
+ "train_speed(iter/s)": 0.223782
731
+ },
732
+ {
733
+ "epoch": 0.32589285714285715,
734
+ "grad_norm": 2.3074914591810103,
735
+ "learning_rate": 9.973312372022853e-06,
736
+ "loss": 0.44243016839027405,
737
+ "memory(GiB)": 128.42,
738
+ "step": 73,
739
+ "token_acc": 0.8616207951070336,
740
+ "train_speed(iter/s)": 0.223836
741
+ },
742
+ {
743
+ "epoch": 0.33035714285714285,
744
+ "grad_norm": 2.334012104081163,
745
+ "learning_rate": 9.971373926029815e-06,
746
+ "loss": 0.3950923681259155,
747
+ "memory(GiB)": 128.42,
748
+ "step": 74,
749
+ "token_acc": 0.868,
750
+ "train_speed(iter/s)": 0.22386
751
+ },
752
+ {
753
+ "epoch": 0.33482142857142855,
754
+ "grad_norm": 2.122646560469079,
755
+ "learning_rate": 9.969367728903796e-06,
756
+ "loss": 0.4495946168899536,
757
+ "memory(GiB)": 128.42,
758
+ "step": 75,
759
+ "token_acc": 0.8410689170182841,
760
+ "train_speed(iter/s)": 0.223964
761
+ },
762
+ {
763
+ "epoch": 0.3392857142857143,
764
+ "grad_norm": 1.9740961761354934,
765
+ "learning_rate": 9.967293807985754e-06,
766
+ "loss": 0.43762198090553284,
767
+ "memory(GiB)": 128.42,
768
+ "step": 76,
769
+ "token_acc": 0.8256802721088435,
770
+ "train_speed(iter/s)": 0.224053
771
+ },
772
+ {
773
+ "epoch": 0.34375,
774
+ "grad_norm": 2.142762356279292,
775
+ "learning_rate": 9.965152191539602e-06,
776
+ "loss": 0.4208489656448364,
777
+ "memory(GiB)": 128.42,
778
+ "step": 77,
779
+ "token_acc": 0.8441158348736907,
780
+ "train_speed(iter/s)": 0.224065
781
+ },
782
+ {
783
+ "epoch": 0.3482142857142857,
784
+ "grad_norm": 2.170173305950217,
785
+ "learning_rate": 9.96294290875183e-06,
786
+ "loss": 0.44035041332244873,
787
+ "memory(GiB)": 128.42,
788
+ "step": 78,
789
+ "token_acc": 0.8557422969187675,
790
+ "train_speed(iter/s)": 0.224153
791
+ },
792
+ {
793
+ "epoch": 0.35267857142857145,
794
+ "grad_norm": 2.0939957380279877,
795
+ "learning_rate": 9.960665989731099e-06,
796
+ "loss": 0.455330491065979,
797
+ "memory(GiB)": 128.42,
798
+ "step": 79,
799
+ "token_acc": 0.8592162554426706,
800
+ "train_speed(iter/s)": 0.224244
801
+ },
802
+ {
803
+ "epoch": 0.35714285714285715,
804
+ "grad_norm": 1.8572342974477305,
805
+ "learning_rate": 9.95832146550783e-06,
806
+ "loss": 0.42493194341659546,
807
+ "memory(GiB)": 128.42,
808
+ "step": 80,
809
+ "token_acc": 0.8617121354656632,
810
+ "train_speed(iter/s)": 0.224289
811
+ },
812
+ {
813
+ "epoch": 0.36160714285714285,
814
+ "grad_norm": 2.0697549040221017,
815
+ "learning_rate": 9.955909368033793e-06,
816
+ "loss": 0.38421130180358887,
817
+ "memory(GiB)": 128.42,
818
+ "step": 81,
819
+ "token_acc": 0.8648,
820
+ "train_speed(iter/s)": 0.224369
821
+ },
822
+ {
823
+ "epoch": 0.36607142857142855,
824
+ "grad_norm": 2.219370159090403,
825
+ "learning_rate": 9.953429730181653e-06,
826
+ "loss": 0.42679229378700256,
827
+ "memory(GiB)": 128.42,
828
+ "step": 82,
829
+ "token_acc": 0.8590038314176245,
830
+ "train_speed(iter/s)": 0.224448
831
+ },
832
+ {
833
+ "epoch": 0.3705357142857143,
834
+ "grad_norm": 2.0309694729211976,
835
+ "learning_rate": 9.950882585744544e-06,
836
+ "loss": 0.42348772287368774,
837
+ "memory(GiB)": 128.42,
838
+ "step": 83,
839
+ "token_acc": 0.8685220729366603,
840
+ "train_speed(iter/s)": 0.224528
841
+ },
842
+ {
843
+ "epoch": 0.375,
844
+ "grad_norm": 1.9953416437787699,
845
+ "learning_rate": 9.948267969435585e-06,
846
+ "loss": 0.4190869927406311,
847
+ "memory(GiB)": 128.42,
848
+ "step": 84,
849
+ "token_acc": 0.8526682134570766,
850
+ "train_speed(iter/s)": 0.22461
851
+ },
852
+ {
853
+ "epoch": 0.3794642857142857,
854
+ "grad_norm": 2.066239379356436,
855
+ "learning_rate": 9.945585916887428e-06,
856
+ "loss": 0.4057341516017914,
857
+ "memory(GiB)": 128.42,
858
+ "step": 85,
859
+ "token_acc": 0.8572555205047319,
860
+ "train_speed(iter/s)": 0.224673
861
+ },
862
+ {
863
+ "epoch": 0.38392857142857145,
864
+ "grad_norm": 2.1617655431273186,
865
+ "learning_rate": 9.942836464651755e-06,
866
+ "loss": 0.4227410554885864,
867
+ "memory(GiB)": 128.42,
868
+ "step": 86,
869
+ "token_acc": 0.8300704776820673,
870
+ "train_speed(iter/s)": 0.224755
871
+ },
872
+ {
873
+ "epoch": 0.38839285714285715,
874
+ "grad_norm": 2.021096735054414,
875
+ "learning_rate": 9.940019650198796e-06,
876
+ "loss": 0.4023309051990509,
877
+ "memory(GiB)": 128.42,
878
+ "step": 87,
879
+ "token_acc": 0.8765765765765766,
880
+ "train_speed(iter/s)": 0.224785
881
+ },
882
+ {
883
+ "epoch": 0.39285714285714285,
884
+ "grad_norm": 1.84828349187564,
885
+ "learning_rate": 9.937135511916802e-06,
886
+ "loss": 0.4349609613418579,
887
+ "memory(GiB)": 128.42,
888
+ "step": 88,
889
+ "token_acc": 0.8710155670867309,
890
+ "train_speed(iter/s)": 0.224842
891
+ },
892
+ {
893
+ "epoch": 0.39732142857142855,
894
+ "grad_norm": 1.8764163466409773,
895
+ "learning_rate": 9.934184089111539e-06,
896
+ "loss": 0.4185750186443329,
897
+ "memory(GiB)": 128.42,
898
+ "step": 89,
899
+ "token_acc": 0.8533333333333334,
900
+ "train_speed(iter/s)": 0.224865
901
+ },
902
+ {
903
+ "epoch": 0.4017857142857143,
904
+ "grad_norm": 1.8632398207712257,
905
+ "learning_rate": 9.931165422005734e-06,
906
+ "loss": 0.438082754611969,
907
+ "memory(GiB)": 128.42,
908
+ "step": 90,
909
+ "token_acc": 0.8347417840375587,
910
+ "train_speed(iter/s)": 0.224921
911
+ },
912
+ {
913
+ "epoch": 0.40625,
914
+ "grad_norm": 2.2173701934991477,
915
+ "learning_rate": 9.928079551738542e-06,
916
+ "loss": 0.43483027815818787,
917
+ "memory(GiB)": 128.42,
918
+ "step": 91,
919
+ "token_acc": 0.8567454798331016,
920
+ "train_speed(iter/s)": 0.224959
921
+ },
922
+ {
923
+ "epoch": 0.4107142857142857,
924
+ "grad_norm": 2.016109177102097,
925
+ "learning_rate": 9.92492652036498e-06,
926
+ "loss": 0.4152158498764038,
927
+ "memory(GiB)": 128.42,
928
+ "step": 92,
929
+ "token_acc": 0.8462462462462462,
930
+ "train_speed(iter/s)": 0.225036
931
+ },
932
+ {
933
+ "epoch": 0.41517857142857145,
934
+ "grad_norm": 2.0334417717146978,
935
+ "learning_rate": 9.921706370855353e-06,
936
+ "loss": 0.3815462589263916,
937
+ "memory(GiB)": 128.42,
938
+ "step": 93,
939
+ "token_acc": 0.8731203007518797,
940
+ "train_speed(iter/s)": 0.225079
941
+ },
942
+ {
943
+ "epoch": 0.41964285714285715,
944
+ "grad_norm": 2.0732022690706198,
945
+ "learning_rate": 9.918419147094663e-06,
946
+ "loss": 0.40599995851516724,
947
+ "memory(GiB)": 128.42,
948
+ "step": 94,
949
+ "token_acc": 0.8433382137628112,
950
+ "train_speed(iter/s)": 0.225114
951
+ },
952
+ {
953
+ "epoch": 0.42410714285714285,
954
+ "grad_norm": 2.0919041720198797,
955
+ "learning_rate": 9.915064893882025e-06,
956
+ "loss": 0.4103803038597107,
957
+ "memory(GiB)": 128.42,
958
+ "step": 95,
959
+ "token_acc": 0.8777555110220441,
960
+ "train_speed(iter/s)": 0.225148
961
+ },
962
+ {
963
+ "epoch": 0.42857142857142855,
964
+ "grad_norm": 3.0894564290504496,
965
+ "learning_rate": 9.911643656930044e-06,
966
+ "loss": 0.441211998462677,
967
+ "memory(GiB)": 128.42,
968
+ "step": 96,
969
+ "token_acc": 0.8468013468013468,
970
+ "train_speed(iter/s)": 0.225197
971
+ },
972
+ {
973
+ "epoch": 0.4330357142857143,
974
+ "grad_norm": 1.9988119785462557,
975
+ "learning_rate": 9.908155482864194e-06,
976
+ "loss": 0.45225849747657776,
977
+ "memory(GiB)": 128.42,
978
+ "step": 97,
979
+ "token_acc": 0.8370786516853933,
980
+ "train_speed(iter/s)": 0.225271
981
+ },
982
+ {
983
+ "epoch": 0.4375,
984
+ "grad_norm": 2.342392393484222,
985
+ "learning_rate": 9.904600419222191e-06,
986
+ "loss": 0.38248753547668457,
987
+ "memory(GiB)": 128.42,
988
+ "step": 98,
989
+ "token_acc": 0.9082652134423251,
990
+ "train_speed(iter/s)": 0.225303
991
+ },
992
+ {
993
+ "epoch": 0.4419642857142857,
994
+ "grad_norm": 2.709700792154672,
995
+ "learning_rate": 9.900978514453335e-06,
996
+ "loss": 0.43812692165374756,
997
+ "memory(GiB)": 128.42,
998
+ "step": 99,
999
+ "token_acc": 0.8883374689826302,
1000
+ "train_speed(iter/s)": 0.225353
1001
+ },
1002
+ {
1003
+ "epoch": 0.44642857142857145,
1004
+ "grad_norm": 2.450946542331287,
1005
+ "learning_rate": 9.897289817917852e-06,
1006
+ "loss": 0.41455474495887756,
1007
+ "memory(GiB)": 128.42,
1008
+ "step": 100,
1009
+ "token_acc": 0.8553672316384181,
1010
+ "train_speed(iter/s)": 0.225398
1011
+ },
1012
+ {
1013
+ "epoch": 0.45089285714285715,
1014
+ "grad_norm": 1.9224997517284426,
1015
+ "learning_rate": 9.893534379886229e-06,
1016
+ "loss": 0.3818276822566986,
1017
+ "memory(GiB)": 128.42,
1018
+ "step": 101,
1019
+ "token_acc": 0.8751062022090059,
1020
+ "train_speed(iter/s)": 0.225453
1021
+ },
1022
+ {
1023
+ "epoch": 0.45535714285714285,
1024
+ "grad_norm": 1.9760417729359396,
1025
+ "learning_rate": 9.889712251538516e-06,
1026
+ "loss": 0.4196687936782837,
1027
+ "memory(GiB)": 128.42,
1028
+ "step": 102,
1029
+ "token_acc": 0.8859784283513097,
1030
+ "train_speed(iter/s)": 0.225506
1031
+ },
1032
+ {
1033
+ "epoch": 0.45982142857142855,
1034
+ "grad_norm": 1.9615231678461282,
1035
+ "learning_rate": 9.88582348496364e-06,
1036
+ "loss": 0.41014599800109863,
1037
+ "memory(GiB)": 128.42,
1038
+ "step": 103,
1039
+ "token_acc": 0.8596377749029754,
1040
+ "train_speed(iter/s)": 0.225567
1041
+ },
1042
+ {
1043
+ "epoch": 0.4642857142857143,
1044
+ "grad_norm": 2.053532677771882,
1045
+ "learning_rate": 9.881868133158687e-06,
1046
+ "loss": 0.3833034634590149,
1047
+ "memory(GiB)": 128.42,
1048
+ "step": 104,
1049
+ "token_acc": 0.8368932038834952,
1050
+ "train_speed(iter/s)": 0.225632
1051
+ },
1052
+ {
1053
+ "epoch": 0.46875,
1054
+ "grad_norm": 2.1912755777585526,
1055
+ "learning_rate": 9.877846250028188e-06,
1056
+ "loss": 0.40928328037261963,
1057
+ "memory(GiB)": 128.42,
1058
+ "step": 105,
1059
+ "token_acc": 0.8811881188118812,
1060
+ "train_speed(iter/s)": 0.225663
1061
+ },
1062
+ {
1063
+ "epoch": 0.4732142857142857,
1064
+ "grad_norm": 1.9575903894574092,
1065
+ "learning_rate": 9.873757890383374e-06,
1066
+ "loss": 0.41512739658355713,
1067
+ "memory(GiB)": 128.42,
1068
+ "step": 106,
1069
+ "token_acc": 0.8568298027757487,
1070
+ "train_speed(iter/s)": 0.225715
1071
+ },
1072
+ {
1073
+ "epoch": 0.47767857142857145,
1074
+ "grad_norm": 2.2673539760779136,
1075
+ "learning_rate": 9.869603109941442e-06,
1076
+ "loss": 0.41499513387680054,
1077
+ "memory(GiB)": 128.42,
1078
+ "step": 107,
1079
+ "token_acc": 0.8683220466516177,
1080
+ "train_speed(iter/s)": 0.225753
1081
+ },
1082
+ {
1083
+ "epoch": 0.48214285714285715,
1084
+ "grad_norm": 5.359243918313353,
1085
+ "learning_rate": 9.865381965324778e-06,
1086
+ "loss": 0.40060681104660034,
1087
+ "memory(GiB)": 128.42,
1088
+ "step": 108,
1089
+ "token_acc": 0.876412429378531,
1090
+ "train_speed(iter/s)": 0.22579
1091
+ },
1092
+ {
1093
+ "epoch": 0.48660714285714285,
1094
+ "grad_norm": 2.1052724949805257,
1095
+ "learning_rate": 9.861094514060206e-06,
1096
+ "loss": 0.45214220881462097,
1097
+ "memory(GiB)": 128.42,
1098
+ "step": 109,
1099
+ "token_acc": 0.8455079764903443,
1100
+ "train_speed(iter/s)": 0.225835
1101
+ },
1102
+ {
1103
+ "epoch": 0.49107142857142855,
1104
+ "grad_norm": 3.125050220505369,
1105
+ "learning_rate": 9.856740814578186e-06,
1106
+ "loss": 0.3993504047393799,
1107
+ "memory(GiB)": 128.42,
1108
+ "step": 110,
1109
+ "token_acc": 0.8351555929352397,
1110
+ "train_speed(iter/s)": 0.22589
1111
+ },
1112
+ {
1113
+ "epoch": 0.4955357142857143,
1114
+ "grad_norm": 1.8525790808401106,
1115
+ "learning_rate": 9.85232092621203e-06,
1116
+ "loss": 0.39937469363212585,
1117
+ "memory(GiB)": 128.42,
1118
+ "step": 111,
1119
+ "token_acc": 0.8457446808510638,
1120
+ "train_speed(iter/s)": 0.225948
1121
+ },
1122
+ {
1123
+ "epoch": 0.5,
1124
+ "grad_norm": 1.9089549875623408,
1125
+ "learning_rate": 9.847834909197089e-06,
1126
+ "loss": 0.40335044264793396,
1127
+ "memory(GiB)": 128.42,
1128
+ "step": 112,
1129
+ "token_acc": 0.8628019323671497,
1130
+ "train_speed(iter/s)": 0.226007
1131
+ },
1132
+ {
1133
+ "epoch": 0.5044642857142857,
1134
+ "grad_norm": 1.9749493510077332,
1135
+ "learning_rate": 9.843282824669926e-06,
1136
+ "loss": 0.42468175292015076,
1137
+ "memory(GiB)": 128.42,
1138
+ "step": 113,
1139
+ "token_acc": 0.8340108401084011,
1140
+ "train_speed(iter/s)": 0.226034
1141
+ },
1142
+ {
1143
+ "epoch": 0.5089285714285714,
1144
+ "grad_norm": 1.9937044644554462,
1145
+ "learning_rate": 9.838664734667496e-06,
1146
+ "loss": 0.38329237699508667,
1147
+ "memory(GiB)": 128.42,
1148
+ "step": 114,
1149
+ "token_acc": 0.8425443169968717,
1150
+ "train_speed(iter/s)": 0.226058
1151
+ },
1152
+ {
1153
+ "epoch": 0.5133928571428571,
1154
+ "grad_norm": 1.560721136274801,
1155
+ "learning_rate": 9.833980702126287e-06,
1156
+ "loss": 0.3472307324409485,
1157
+ "memory(GiB)": 128.42,
1158
+ "step": 115,
1159
+ "token_acc": 0.8923444976076556,
1160
+ "train_speed(iter/s)": 0.226091
1161
+ },
1162
+ {
1163
+ "epoch": 0.5178571428571429,
1164
+ "grad_norm": 1.7863431958110831,
1165
+ "learning_rate": 9.829230790881474e-06,
1166
+ "loss": 0.4139295518398285,
1167
+ "memory(GiB)": 128.42,
1168
+ "step": 116,
1169
+ "token_acc": 0.8657142857142858,
1170
+ "train_speed(iter/s)": 0.226136
1171
+ },
1172
+ {
1173
+ "epoch": 0.5223214285714286,
1174
+ "grad_norm": 2.0227555853718804,
1175
+ "learning_rate": 9.82441506566604e-06,
1176
+ "loss": 0.4211953282356262,
1177
+ "memory(GiB)": 128.42,
1178
+ "step": 117,
1179
+ "token_acc": 0.8500986193293886,
1180
+ "train_speed(iter/s)": 0.226158
1181
+ },
1182
+ {
1183
+ "epoch": 0.5267857142857143,
1184
+ "grad_norm": 1.892221973169317,
1185
+ "learning_rate": 9.819533592109898e-06,
1186
+ "loss": 0.3711167573928833,
1187
+ "memory(GiB)": 128.42,
1188
+ "step": 118,
1189
+ "token_acc": 0.8703849175176748,
1190
+ "train_speed(iter/s)": 0.226201
1191
+ },
1192
+ {
1193
+ "epoch": 0.53125,
1194
+ "grad_norm": 1.8769689856557228,
1195
+ "learning_rate": 9.814586436738998e-06,
1196
+ "loss": 0.3753877282142639,
1197
+ "memory(GiB)": 128.42,
1198
+ "step": 119,
1199
+ "token_acc": 0.8629655657062544,
1200
+ "train_speed(iter/s)": 0.226223
1201
+ },
1202
+ {
1203
+ "epoch": 0.5357142857142857,
1204
+ "grad_norm": 1.9077137109160918,
1205
+ "learning_rate": 9.809573666974413e-06,
1206
+ "loss": 0.3845405876636505,
1207
+ "memory(GiB)": 128.42,
1208
+ "step": 120,
1209
+ "token_acc": 0.8865248226950354,
1210
+ "train_speed(iter/s)": 0.226266
1211
+ },
1212
+ {
1213
+ "epoch": 0.5401785714285714,
1214
+ "grad_norm": 1.8483195634970053,
1215
+ "learning_rate": 9.80449535113143e-06,
1216
+ "loss": 0.3865225315093994,
1217
+ "memory(GiB)": 128.42,
1218
+ "step": 121,
1219
+ "token_acc": 0.8959227467811158,
1220
+ "train_speed(iter/s)": 0.226304
1221
+ },
1222
+ {
1223
+ "epoch": 0.5446428571428571,
1224
+ "grad_norm": 2.015815422114459,
1225
+ "learning_rate": 9.799351558418616e-06,
1226
+ "loss": 0.4358481764793396,
1227
+ "memory(GiB)": 128.42,
1228
+ "step": 122,
1229
+ "token_acc": 0.8597063621533442,
1230
+ "train_speed(iter/s)": 0.226338
1231
+ },
1232
+ {
1233
+ "epoch": 0.5491071428571429,
1234
+ "grad_norm": 1.8365267081758667,
1235
+ "learning_rate": 9.794142358936869e-06,
1236
+ "loss": 0.4205068051815033,
1237
+ "memory(GiB)": 128.42,
1238
+ "step": 123,
1239
+ "token_acc": 0.8647194465795542,
1240
+ "train_speed(iter/s)": 0.226383
1241
+ },
1242
+ {
1243
+ "epoch": 0.5535714285714286,
1244
+ "grad_norm": 2.0725669814164362,
1245
+ "learning_rate": 9.788867823678465e-06,
1246
+ "loss": 0.42481255531311035,
1247
+ "memory(GiB)": 128.42,
1248
+ "step": 124,
1249
+ "token_acc": 0.8483412322274881,
1250
+ "train_speed(iter/s)": 0.226421
1251
+ },
1252
+ {
1253
+ "epoch": 0.5580357142857143,
1254
+ "grad_norm": 1.914935421654447,
1255
+ "learning_rate": 9.7835280245261e-06,
1256
+ "loss": 0.39662468433380127,
1257
+ "memory(GiB)": 128.42,
1258
+ "step": 125,
1259
+ "token_acc": 0.8517520215633423,
1260
+ "train_speed(iter/s)": 0.226451
1261
+ },
1262
+ {
1263
+ "epoch": 0.5625,
1264
+ "grad_norm": 1.8558613972881188,
1265
+ "learning_rate": 9.778123034251898e-06,
1266
+ "loss": 0.40216946601867676,
1267
+ "memory(GiB)": 128.42,
1268
+ "step": 126,
1269
+ "token_acc": 0.8529411764705882,
1270
+ "train_speed(iter/s)": 0.226489
1271
+ },
1272
+ {
1273
+ "epoch": 0.5669642857142857,
1274
+ "grad_norm": 2.073519670141887,
1275
+ "learning_rate": 9.772652926516424e-06,
1276
+ "loss": 0.4289711117744446,
1277
+ "memory(GiB)": 128.42,
1278
+ "step": 127,
1279
+ "token_acc": 0.8672032193158954,
1280
+ "train_speed(iter/s)": 0.226539
1281
+ },
1282
+ {
1283
+ "epoch": 0.5714285714285714,
1284
+ "grad_norm": 1.9393590937147855,
1285
+ "learning_rate": 9.76711777586768e-06,
1286
+ "loss": 0.411954402923584,
1287
+ "memory(GiB)": 128.42,
1288
+ "step": 128,
1289
+ "token_acc": 0.8615771812080537,
1290
+ "train_speed(iter/s)": 0.226549
1291
+ },
1292
+ {
1293
+ "epoch": 0.5758928571428571,
1294
+ "grad_norm": 1.700134216785422,
1295
+ "learning_rate": 9.76151765774009e-06,
1296
+ "loss": 0.38732337951660156,
1297
+ "memory(GiB)": 128.42,
1298
+ "step": 129,
1299
+ "token_acc": 0.8750952018278751,
1300
+ "train_speed(iter/s)": 0.226588
1301
+ },
1302
+ {
1303
+ "epoch": 0.5803571428571429,
1304
+ "grad_norm": 1.8772660862292765,
1305
+ "learning_rate": 9.755852648453472e-06,
1306
+ "loss": 0.4092137813568115,
1307
+ "memory(GiB)": 128.42,
1308
+ "step": 130,
1309
+ "token_acc": 0.8649969456322542,
1310
+ "train_speed(iter/s)": 0.226612
1311
+ },
1312
+ {
1313
+ "epoch": 0.5848214285714286,
1314
+ "grad_norm": 1.84397328756411,
1315
+ "learning_rate": 9.750122825211995e-06,
1316
+ "loss": 0.3621349334716797,
1317
+ "memory(GiB)": 128.42,
1318
+ "step": 131,
1319
+ "token_acc": 0.9,
1320
+ "train_speed(iter/s)": 0.226653
1321
+ },
1322
+ {
1323
+ "epoch": 0.5892857142857143,
1324
+ "grad_norm": 1.8410084418847592,
1325
+ "learning_rate": 9.74432826610313e-06,
1326
+ "loss": 0.3831413984298706,
1327
+ "memory(GiB)": 128.42,
1328
+ "step": 132,
1329
+ "token_acc": 0.8408723747980614,
1330
+ "train_speed(iter/s)": 0.226689
1331
+ },
1332
+ {
1333
+ "epoch": 0.59375,
1334
+ "grad_norm": 1.8350848291819015,
1335
+ "learning_rate": 9.738469050096583e-06,
1336
+ "loss": 0.4100695252418518,
1337
+ "memory(GiB)": 128.42,
1338
+ "step": 133,
1339
+ "token_acc": 0.8534258456201215,
1340
+ "train_speed(iter/s)": 0.226704
1341
+ },
1342
+ {
1343
+ "epoch": 0.5982142857142857,
1344
+ "grad_norm": 1.8565849327061523,
1345
+ "learning_rate": 9.732545257043224e-06,
1346
+ "loss": 0.42670905590057373,
1347
+ "memory(GiB)": 128.42,
1348
+ "step": 134,
1349
+ "token_acc": 0.8765670202507232,
1350
+ "train_speed(iter/s)": 0.226619
1351
+ },
1352
+ {
1353
+ "epoch": 0.6026785714285714,
1354
+ "grad_norm": 1.8345425226029803,
1355
+ "learning_rate": 9.726556967673992e-06,
1356
+ "loss": 0.4107423424720764,
1357
+ "memory(GiB)": 128.42,
1358
+ "step": 135,
1359
+ "token_acc": 0.8629764065335753,
1360
+ "train_speed(iter/s)": 0.226641
1361
+ },
1362
+ {
1363
+ "epoch": 0.6071428571428571,
1364
+ "grad_norm": 1.8197069879254395,
1365
+ "learning_rate": 9.720504263598802e-06,
1366
+ "loss": 0.3949946463108063,
1367
+ "memory(GiB)": 128.42,
1368
+ "step": 136,
1369
+ "token_acc": 0.8668706962509564,
1370
+ "train_speed(iter/s)": 0.226655
1371
+ },
1372
+ {
1373
+ "epoch": 0.6116071428571429,
1374
+ "grad_norm": 1.8605070813229492,
1375
+ "learning_rate": 9.714387227305422e-06,
1376
+ "loss": 0.3933284878730774,
1377
+ "memory(GiB)": 128.42,
1378
+ "step": 137,
1379
+ "token_acc": 0.8596247394023627,
1380
+ "train_speed(iter/s)": 0.226682
1381
+ },
1382
+ {
1383
+ "epoch": 0.6160714285714286,
1384
+ "grad_norm": 1.9607719811040827,
1385
+ "learning_rate": 9.708205942158362e-06,
1386
+ "loss": 0.37001386284828186,
1387
+ "memory(GiB)": 128.42,
1388
+ "step": 138,
1389
+ "token_acc": 0.8576923076923076,
1390
+ "train_speed(iter/s)": 0.226702
1391
+ },
1392
+ {
1393
+ "epoch": 0.6205357142857143,
1394
+ "grad_norm": 1.7714294063492095,
1395
+ "learning_rate": 9.701960492397727e-06,
1396
+ "loss": 0.40147697925567627,
1397
+ "memory(GiB)": 128.42,
1398
+ "step": 139,
1399
+ "token_acc": 0.8800834202294057,
1400
+ "train_speed(iter/s)": 0.226715
1401
+ },
1402
+ {
1403
+ "epoch": 0.625,
1404
+ "grad_norm": 1.9248044276022995,
1405
+ "learning_rate": 9.695650963138079e-06,
1406
+ "loss": 0.4316914677619934,
1407
+ "memory(GiB)": 128.42,
1408
+ "step": 140,
1409
+ "token_acc": 0.8928994082840237,
1410
+ "train_speed(iter/s)": 0.226745
1411
+ },
1412
+ {
1413
+ "epoch": 0.6294642857142857,
1414
+ "grad_norm": 1.9422362404479026,
1415
+ "learning_rate": 9.689277440367263e-06,
1416
+ "loss": 0.43073832988739014,
1417
+ "memory(GiB)": 128.42,
1418
+ "step": 141,
1419
+ "token_acc": 0.8460329929300864,
1420
+ "train_speed(iter/s)": 0.226779
1421
+ },
1422
+ {
1423
+ "epoch": 0.6339285714285714,
1424
+ "grad_norm": 1.8593524333480311,
1425
+ "learning_rate": 9.682840010945254e-06,
1426
+ "loss": 0.392534077167511,
1427
+ "memory(GiB)": 128.42,
1428
+ "step": 142,
1429
+ "token_acc": 0.9042076991942704,
1430
+ "train_speed(iter/s)": 0.226798
1431
+ },
1432
+ {
1433
+ "epoch": 0.6383928571428571,
1434
+ "grad_norm": 1.924142438232916,
1435
+ "learning_rate": 9.676338762602954e-06,
1436
+ "loss": 0.3955026865005493,
1437
+ "memory(GiB)": 128.42,
1438
+ "step": 143,
1439
+ "token_acc": 0.8868398994132439,
1440
+ "train_speed(iter/s)": 0.226803
1441
+ },
1442
+ {
1443
+ "epoch": 0.6428571428571429,
1444
+ "grad_norm": 1.803030550107758,
1445
+ "learning_rate": 9.669773783941012e-06,
1446
+ "loss": 0.38179874420166016,
1447
+ "memory(GiB)": 128.42,
1448
+ "step": 144,
1449
+ "token_acc": 0.853448275862069,
1450
+ "train_speed(iter/s)": 0.226841
1451
+ },
1452
+ {
1453
+ "epoch": 0.6473214285714286,
1454
+ "grad_norm": 2.136428096666585,
1455
+ "learning_rate": 9.663145164428606e-06,
1456
+ "loss": 0.43818652629852295,
1457
+ "memory(GiB)": 128.42,
1458
+ "step": 145,
1459
+ "token_acc": 0.8600973236009732,
1460
+ "train_speed(iter/s)": 0.226848
1461
+ },
1462
+ {
1463
+ "epoch": 0.6517857142857143,
1464
+ "grad_norm": 1.9680755115856965,
1465
+ "learning_rate": 9.656452994402225e-06,
1466
+ "loss": 0.38914287090301514,
1467
+ "memory(GiB)": 128.42,
1468
+ "step": 146,
1469
+ "token_acc": 0.8546511627906976,
1470
+ "train_speed(iter/s)": 0.226865
1471
+ },
1472
+ {
1473
+ "epoch": 0.65625,
1474
+ "grad_norm": 1.8644180163525486,
1475
+ "learning_rate": 9.649697365064448e-06,
1476
+ "loss": 0.3467503786087036,
1477
+ "memory(GiB)": 128.42,
1478
+ "step": 147,
1479
+ "token_acc": 0.8939580764488286,
1480
+ "train_speed(iter/s)": 0.226873
1481
+ },
1482
+ {
1483
+ "epoch": 0.6607142857142857,
1484
+ "grad_norm": 1.7646489123753146,
1485
+ "learning_rate": 9.642878368482685e-06,
1486
+ "loss": 0.40651845932006836,
1487
+ "memory(GiB)": 128.42,
1488
+ "step": 148,
1489
+ "token_acc": 0.8590006844626967,
1490
+ "train_speed(iter/s)": 0.226905
1491
+ },
1492
+ {
1493
+ "epoch": 0.6651785714285714,
1494
+ "grad_norm": 1.7110984383437826,
1495
+ "learning_rate": 9.635996097587938e-06,
1496
+ "loss": 0.3564274311065674,
1497
+ "memory(GiB)": 128.42,
1498
+ "step": 149,
1499
+ "token_acc": 0.8814642649622313,
1500
+ "train_speed(iter/s)": 0.226926
1501
+ },
1502
+ {
1503
+ "epoch": 0.6696428571428571,
1504
+ "grad_norm": 1.8656412445393,
1505
+ "learning_rate": 9.629050646173521e-06,
1506
+ "loss": 0.4157032370567322,
1507
+ "memory(GiB)": 128.42,
1508
+ "step": 150,
1509
+ "token_acc": 0.8456883509833586,
1510
+ "train_speed(iter/s)": 0.22696
1511
+ },
1512
+ {
1513
+ "epoch": 0.6741071428571429,
1514
+ "grad_norm": 1.7627569933191594,
1515
+ "learning_rate": 9.622042108893794e-06,
1516
+ "loss": 0.38666343688964844,
1517
+ "memory(GiB)": 128.42,
1518
+ "step": 151,
1519
+ "token_acc": 0.8736532810969637,
1520
+ "train_speed(iter/s)": 0.226987
1521
+ },
1522
+ {
1523
+ "epoch": 0.6785714285714286,
1524
+ "grad_norm": 1.8386703451357664,
1525
+ "learning_rate": 9.614970581262863e-06,
1526
+ "loss": 0.4122501611709595,
1527
+ "memory(GiB)": 128.42,
1528
+ "step": 152,
1529
+ "token_acc": 0.8647302904564316,
1530
+ "train_speed(iter/s)": 0.227012
1531
+ },
1532
+ {
1533
+ "epoch": 0.6830357142857143,
1534
+ "grad_norm": 1.7836526981478449,
1535
+ "learning_rate": 9.607836159653283e-06,
1536
+ "loss": 0.3835086226463318,
1537
+ "memory(GiB)": 128.42,
1538
+ "step": 153,
1539
+ "token_acc": 0.8738277919863597,
1540
+ "train_speed(iter/s)": 0.227049
1541
+ },
1542
+ {
1543
+ "epoch": 0.6875,
1544
+ "grad_norm": 1.821283805318599,
1545
+ "learning_rate": 9.600638941294744e-06,
1546
+ "loss": 0.38780418038368225,
1547
+ "memory(GiB)": 128.42,
1548
+ "step": 154,
1549
+ "token_acc": 0.8396739130434783,
1550
+ "train_speed(iter/s)": 0.227066
1551
+ },
1552
+ {
1553
+ "epoch": 0.6919642857142857,
1554
+ "grad_norm": 1.86521272428445,
1555
+ "learning_rate": 9.593379024272748e-06,
1556
+ "loss": 0.44408848881721497,
1557
+ "memory(GiB)": 128.42,
1558
+ "step": 155,
1559
+ "token_acc": 0.810251798561151,
1560
+ "train_speed(iter/s)": 0.227065
1561
+ },
1562
+ {
1563
+ "epoch": 0.6964285714285714,
1564
+ "grad_norm": 1.6172624083533582,
1565
+ "learning_rate": 9.586056507527266e-06,
1566
+ "loss": 0.38970640301704407,
1567
+ "memory(GiB)": 128.42,
1568
+ "step": 156,
1569
+ "token_acc": 0.8582302568981922,
1570
+ "train_speed(iter/s)": 0.22708
1571
+ },
1572
+ {
1573
+ "epoch": 0.7008928571428571,
1574
+ "grad_norm": 1.7015443365556533,
1575
+ "learning_rate": 9.578671490851396e-06,
1576
+ "loss": 0.37300947308540344,
1577
+ "memory(GiB)": 128.42,
1578
+ "step": 157,
1579
+ "token_acc": 0.8557409224730128,
1580
+ "train_speed(iter/s)": 0.227102
1581
+ },
1582
+ {
1583
+ "epoch": 0.7053571428571429,
1584
+ "grad_norm": 1.9126023576603783,
1585
+ "learning_rate": 9.571224074890004e-06,
1586
+ "loss": 0.40674978494644165,
1587
+ "memory(GiB)": 128.42,
1588
+ "step": 158,
1589
+ "token_acc": 0.8636613523914238,
1590
+ "train_speed(iter/s)": 0.227116
1591
+ },
1592
+ {
1593
+ "epoch": 0.7098214285714286,
1594
+ "grad_norm": 1.8236785039444972,
1595
+ "learning_rate": 9.563714361138343e-06,
1596
+ "loss": 0.34416210651397705,
1597
+ "memory(GiB)": 128.42,
1598
+ "step": 159,
1599
+ "token_acc": 0.8637236084452975,
1600
+ "train_speed(iter/s)": 0.227134
1601
+ },
1602
+ {
1603
+ "epoch": 0.7142857142857143,
1604
+ "grad_norm": 1.737598943732038,
1605
+ "learning_rate": 9.55614245194068e-06,
1606
+ "loss": 0.40488338470458984,
1607
+ "memory(GiB)": 128.42,
1608
+ "step": 160,
1609
+ "token_acc": 0.856110776640578,
1610
+ "train_speed(iter/s)": 0.227158
1611
+ },
1612
+ {
1613
+ "epoch": 0.71875,
1614
+ "grad_norm": 1.6120450500571684,
1615
+ "learning_rate": 9.548508450488895e-06,
1616
+ "loss": 0.3595935106277466,
1617
+ "memory(GiB)": 128.42,
1618
+ "step": 161,
1619
+ "token_acc": 0.8699132755170114,
1620
+ "train_speed(iter/s)": 0.227195
1621
+ },
1622
+ {
1623
+ "epoch": 0.7232142857142857,
1624
+ "grad_norm": 1.693568310349621,
1625
+ "learning_rate": 9.540812460821078e-06,
1626
+ "loss": 0.352266401052475,
1627
+ "memory(GiB)": 128.42,
1628
+ "step": 162,
1629
+ "token_acc": 0.8873072360616845,
1630
+ "train_speed(iter/s)": 0.227208
1631
+ },
1632
+ {
1633
+ "epoch": 0.7276785714285714,
1634
+ "grad_norm": 1.5757683818467814,
1635
+ "learning_rate": 9.533054587820112e-06,
1636
+ "loss": 0.369427353143692,
1637
+ "memory(GiB)": 128.42,
1638
+ "step": 163,
1639
+ "token_acc": 0.8559322033898306,
1640
+ "train_speed(iter/s)": 0.227247
1641
+ },
1642
+ {
1643
+ "epoch": 0.7321428571428571,
1644
+ "grad_norm": 1.6076615751659218,
1645
+ "learning_rate": 9.525234937212238e-06,
1646
+ "loss": 0.3551783859729767,
1647
+ "memory(GiB)": 128.42,
1648
+ "step": 164,
1649
+ "token_acc": 0.8765217391304347,
1650
+ "train_speed(iter/s)": 0.227269
1651
+ },
1652
+ {
1653
+ "epoch": 0.7366071428571429,
1654
+ "grad_norm": 1.8347392969943477,
1655
+ "learning_rate": 9.51735361556562e-06,
1656
+ "loss": 0.38330310583114624,
1657
+ "memory(GiB)": 128.42,
1658
+ "step": 165,
1659
+ "token_acc": 0.8470474346563408,
1660
+ "train_speed(iter/s)": 0.227248
1661
+ },
1662
+ {
1663
+ "epoch": 0.7410714285714286,
1664
+ "grad_norm": 1.7630173128684303,
1665
+ "learning_rate": 9.509410730288888e-06,
1666
+ "loss": 0.37735095620155334,
1667
+ "memory(GiB)": 128.42,
1668
+ "step": 166,
1669
+ "token_acc": 0.8636363636363636,
1670
+ "train_speed(iter/s)": 0.227248
1671
+ },
1672
+ {
1673
+ "epoch": 0.7455357142857143,
1674
+ "grad_norm": 1.782005009352597,
1675
+ "learning_rate": 9.501406389629682e-06,
1676
+ "loss": 0.3654155135154724,
1677
+ "memory(GiB)": 128.42,
1678
+ "step": 167,
1679
+ "token_acc": 0.8790035587188612,
1680
+ "train_speed(iter/s)": 0.227271
1681
+ },
1682
+ {
1683
+ "epoch": 0.75,
1684
+ "grad_norm": 1.7574512911157494,
1685
+ "learning_rate": 9.493340702673162e-06,
1686
+ "loss": 0.411441445350647,
1687
+ "memory(GiB)": 128.42,
1688
+ "step": 168,
1689
+ "token_acc": 0.8481375358166189,
1690
+ "train_speed(iter/s)": 0.227286
1691
+ },
1692
+ {
1693
+ "epoch": 0.7544642857142857,
1694
+ "grad_norm": 1.7576357797026396,
1695
+ "learning_rate": 9.485213779340541e-06,
1696
+ "loss": 0.4257476329803467,
1697
+ "memory(GiB)": 128.42,
1698
+ "step": 169,
1699
+ "token_acc": 0.8625083836351441,
1700
+ "train_speed(iter/s)": 0.22732
1701
+ },
1702
+ {
1703
+ "epoch": 0.7589285714285714,
1704
+ "grad_norm": 1.7527537543212042,
1705
+ "learning_rate": 9.47702573038757e-06,
1706
+ "loss": 0.39181047677993774,
1707
+ "memory(GiB)": 128.42,
1708
+ "step": 170,
1709
+ "token_acc": 0.8573643410852713,
1710
+ "train_speed(iter/s)": 0.227307
1711
+ },
1712
+ {
1713
+ "epoch": 0.7633928571428571,
1714
+ "grad_norm": 1.7629371359176924,
1715
+ "learning_rate": 9.468776667403039e-06,
1716
+ "loss": 0.4168310761451721,
1717
+ "memory(GiB)": 128.42,
1718
+ "step": 171,
1719
+ "token_acc": 0.8672046955245781,
1720
+ "train_speed(iter/s)": 0.227328
1721
+ },
1722
+ {
1723
+ "epoch": 0.7678571428571429,
1724
+ "grad_norm": 1.8049015584821484,
1725
+ "learning_rate": 9.460466702807249e-06,
1726
+ "loss": 0.4058079123497009,
1727
+ "memory(GiB)": 128.42,
1728
+ "step": 172,
1729
+ "token_acc": 0.8550583657587548,
1730
+ "train_speed(iter/s)": 0.227352
1731
+ },
1732
+ {
1733
+ "epoch": 0.7723214285714286,
1734
+ "grad_norm": 1.7714028210772665,
1735
+ "learning_rate": 9.452095949850489e-06,
1736
+ "loss": 0.3684278130531311,
1737
+ "memory(GiB)": 128.42,
1738
+ "step": 173,
1739
+ "token_acc": 0.8781074578989575,
1740
+ "train_speed(iter/s)": 0.22737
1741
+ },
1742
+ {
1743
+ "epoch": 0.7767857142857143,
1744
+ "grad_norm": 1.7353068281562303,
1745
+ "learning_rate": 9.44366452261148e-06,
1746
+ "loss": 0.37866702675819397,
1747
+ "memory(GiB)": 128.42,
1748
+ "step": 174,
1749
+ "token_acc": 0.8828058169375534,
1750
+ "train_speed(iter/s)": 0.227381
1751
+ },
1752
+ {
1753
+ "epoch": 0.78125,
1754
+ "grad_norm": 1.7300961263809764,
1755
+ "learning_rate": 9.435172535995831e-06,
1756
+ "loss": 0.38601797819137573,
1757
+ "memory(GiB)": 128.42,
1758
+ "step": 175,
1759
+ "token_acc": 0.8662420382165605,
1760
+ "train_speed(iter/s)": 0.227398
1761
+ },
1762
+ {
1763
+ "epoch": 0.7857142857142857,
1764
+ "grad_norm": 1.741272487827446,
1765
+ "learning_rate": 9.426620105734473e-06,
1766
+ "loss": 0.3633964955806732,
1767
+ "memory(GiB)": 128.42,
1768
+ "step": 176,
1769
+ "token_acc": 0.9051339285714286,
1770
+ "train_speed(iter/s)": 0.227429
1771
+ },
1772
+ {
1773
+ "epoch": 0.7901785714285714,
1774
+ "grad_norm": 1.6875950307747867,
1775
+ "learning_rate": 9.41800734838207e-06,
1776
+ "loss": 0.4060158431529999,
1777
+ "memory(GiB)": 128.42,
1778
+ "step": 177,
1779
+ "token_acc": 0.8517985611510791,
1780
+ "train_speed(iter/s)": 0.227457
1781
+ },
1782
+ {
1783
+ "epoch": 0.7946428571428571,
1784
+ "grad_norm": 1.6697656183460996,
1785
+ "learning_rate": 9.409334381315445e-06,
1786
+ "loss": 0.37892186641693115,
1787
+ "memory(GiB)": 128.42,
1788
+ "step": 178,
1789
+ "token_acc": 0.876099706744868,
1790
+ "train_speed(iter/s)": 0.227485
1791
+ },
1792
+ {
1793
+ "epoch": 0.7991071428571429,
1794
+ "grad_norm": 1.6519426865105165,
1795
+ "learning_rate": 9.400601322731974e-06,
1796
+ "loss": 0.3567098379135132,
1797
+ "memory(GiB)": 128.42,
1798
+ "step": 179,
1799
+ "token_acc": 0.8501170960187353,
1800
+ "train_speed(iter/s)": 0.227509
1801
+ },
1802
+ {
1803
+ "epoch": 0.8035714285714286,
1804
+ "grad_norm": 1.7786028639956994,
1805
+ "learning_rate": 9.391808291647971e-06,
1806
+ "loss": 0.37524616718292236,
1807
+ "memory(GiB)": 128.42,
1808
+ "step": 180,
1809
+ "token_acc": 0.8761421319796955,
1810
+ "train_speed(iter/s)": 0.227536
1811
+ },
1812
+ {
1813
+ "epoch": 0.8080357142857143,
1814
+ "grad_norm": 1.7489126178483658,
1815
+ "learning_rate": 9.382955407897076e-06,
1816
+ "loss": 0.41019266843795776,
1817
+ "memory(GiB)": 128.42,
1818
+ "step": 181,
1819
+ "token_acc": 0.8840182648401826,
1820
+ "train_speed(iter/s)": 0.227569
1821
+ },
1822
+ {
1823
+ "epoch": 0.8125,
1824
+ "grad_norm": 1.987286094728141,
1825
+ "learning_rate": 9.37404279212861e-06,
1826
+ "loss": 0.4129825830459595,
1827
+ "memory(GiB)": 128.42,
1828
+ "step": 182,
1829
+ "token_acc": 0.8421052631578947,
1830
+ "train_speed(iter/s)": 0.227601
1831
+ },
1832
+ {
1833
+ "epoch": 0.8169642857142857,
1834
+ "grad_norm": 1.9033194956801682,
1835
+ "learning_rate": 9.365070565805941e-06,
1836
+ "loss": 0.4232962131500244,
1837
+ "memory(GiB)": 128.42,
1838
+ "step": 183,
1839
+ "token_acc": 0.8359788359788359,
1840
+ "train_speed(iter/s)": 0.227614
1841
+ },
1842
+ {
1843
+ "epoch": 0.8214285714285714,
1844
+ "grad_norm": 1.8779587414450334,
1845
+ "learning_rate": 9.356038851204826e-06,
1846
+ "loss": 0.4073084592819214,
1847
+ "memory(GiB)": 128.42,
1848
+ "step": 184,
1849
+ "token_acc": 0.8229548229548229,
1850
+ "train_speed(iter/s)": 0.227615
1851
+ },
1852
+ {
1853
+ "epoch": 0.8258928571428571,
1854
+ "grad_norm": 1.9063659856430246,
1855
+ "learning_rate": 9.346947771411738e-06,
1856
+ "loss": 0.4089735746383667,
1857
+ "memory(GiB)": 128.42,
1858
+ "step": 185,
1859
+ "token_acc": 0.8871391076115486,
1860
+ "train_speed(iter/s)": 0.22762
1861
+ },
1862
+ {
1863
+ "epoch": 0.8303571428571429,
1864
+ "grad_norm": 1.4836242663153665,
1865
+ "learning_rate": 9.337797450322197e-06,
1866
+ "loss": 0.3512466549873352,
1867
+ "memory(GiB)": 128.42,
1868
+ "step": 186,
1869
+ "token_acc": 0.872382851445663,
1870
+ "train_speed(iter/s)": 0.227641
1871
+ },
1872
+ {
1873
+ "epoch": 0.8348214285714286,
1874
+ "grad_norm": 1.555500407882726,
1875
+ "learning_rate": 9.328588012639081e-06,
1876
+ "loss": 0.3591383099555969,
1877
+ "memory(GiB)": 128.42,
1878
+ "step": 187,
1879
+ "token_acc": 0.8741721854304636,
1880
+ "train_speed(iter/s)": 0.227662
1881
+ },
1882
+ {
1883
+ "epoch": 0.8392857142857143,
1884
+ "grad_norm": 1.7087345811385168,
1885
+ "learning_rate": 9.319319583870918e-06,
1886
+ "loss": 0.37500661611557007,
1887
+ "memory(GiB)": 128.42,
1888
+ "step": 188,
1889
+ "token_acc": 0.8708685626441199,
1890
+ "train_speed(iter/s)": 0.22768
1891
+ },
1892
+ {
1893
+ "epoch": 0.84375,
1894
+ "grad_norm": 1.7815220397240672,
1895
+ "learning_rate": 9.309992290330182e-06,
1896
+ "loss": 0.40438687801361084,
1897
+ "memory(GiB)": 128.42,
1898
+ "step": 189,
1899
+ "token_acc": 0.8631436314363143,
1900
+ "train_speed(iter/s)": 0.227689
1901
+ },
1902
+ {
1903
+ "epoch": 0.8482142857142857,
1904
+ "grad_norm": 1.5997044351272065,
1905
+ "learning_rate": 9.300606259131578e-06,
1906
+ "loss": 0.37673109769821167,
1907
+ "memory(GiB)": 128.42,
1908
+ "step": 190,
1909
+ "token_acc": 0.8944281524926686,
1910
+ "train_speed(iter/s)": 0.227709
1911
+ },
1912
+ {
1913
+ "epoch": 0.8526785714285714,
1914
+ "grad_norm": 1.7618126169744526,
1915
+ "learning_rate": 9.291161618190295e-06,
1916
+ "loss": 0.41258758306503296,
1917
+ "memory(GiB)": 128.42,
1918
+ "step": 191,
1919
+ "token_acc": 0.8530646515533166,
1920
+ "train_speed(iter/s)": 0.227723
1921
+ },
1922
+ {
1923
+ "epoch": 0.8571428571428571,
1924
+ "grad_norm": 1.5709357605507697,
1925
+ "learning_rate": 9.281658496220272e-06,
1926
+ "loss": 0.3878024220466614,
1927
+ "memory(GiB)": 128.42,
1928
+ "step": 192,
1929
+ "token_acc": 0.8767885532591415,
1930
+ "train_speed(iter/s)": 0.227734
1931
+ },
1932
+ {
1933
+ "epoch": 0.8616071428571429,
1934
+ "grad_norm": 1.5758516203486068,
1935
+ "learning_rate": 9.272097022732444e-06,
1936
+ "loss": 0.3386090099811554,
1937
+ "memory(GiB)": 128.42,
1938
+ "step": 193,
1939
+ "token_acc": 0.8722772277227723,
1940
+ "train_speed(iter/s)": 0.22775
1941
+ },
1942
+ {
1943
+ "epoch": 0.8660714285714286,
1944
+ "grad_norm": 1.7875878983881768,
1945
+ "learning_rate": 9.262477328032972e-06,
1946
+ "loss": 0.36395904421806335,
1947
+ "memory(GiB)": 128.42,
1948
+ "step": 194,
1949
+ "token_acc": 0.8647058823529412,
1950
+ "train_speed(iter/s)": 0.227755
1951
+ },
1952
+ {
1953
+ "epoch": 0.8705357142857143,
1954
+ "grad_norm": 1.7206122039653617,
1955
+ "learning_rate": 9.252799543221478e-06,
1956
+ "loss": 0.3222510814666748,
1957
+ "memory(GiB)": 128.42,
1958
+ "step": 195,
1959
+ "token_acc": 0.8872691933916423,
1960
+ "train_speed(iter/s)": 0.227771
1961
+ },
1962
+ {
1963
+ "epoch": 0.875,
1964
+ "grad_norm": 1.6063432393863786,
1965
+ "learning_rate": 9.24306380018924e-06,
1966
+ "loss": 0.366471529006958,
1967
+ "memory(GiB)": 128.42,
1968
+ "step": 196,
1969
+ "token_acc": 0.8529680365296803,
1970
+ "train_speed(iter/s)": 0.227755
1971
+ },
1972
+ {
1973
+ "epoch": 0.8794642857142857,
1974
+ "grad_norm": 1.5262897164552471,
1975
+ "learning_rate": 9.23327023161741e-06,
1976
+ "loss": 0.35482266545295715,
1977
+ "memory(GiB)": 128.42,
1978
+ "step": 197,
1979
+ "token_acc": 0.8913502109704642,
1980
+ "train_speed(iter/s)": 0.227775
1981
+ },
1982
+ {
1983
+ "epoch": 0.8839285714285714,
1984
+ "grad_norm": 1.5528440615685493,
1985
+ "learning_rate": 9.223418970975206e-06,
1986
+ "loss": 0.34359997510910034,
1987
+ "memory(GiB)": 128.42,
1988
+ "step": 198,
1989
+ "token_acc": 0.8822222222222222,
1990
+ "train_speed(iter/s)": 0.227794
1991
+ },
1992
+ {
1993
+ "epoch": 0.8883928571428571,
1994
+ "grad_norm": 1.6542393490950473,
1995
+ "learning_rate": 9.213510152518079e-06,
1996
+ "loss": 0.35520508885383606,
1997
+ "memory(GiB)": 128.42,
1998
+ "step": 199,
1999
+ "token_acc": 0.8734087694483734,
2000
+ "train_speed(iter/s)": 0.227806
2001
+ },
2002
+ {
2003
+ "epoch": 0.8928571428571429,
2004
+ "grad_norm": 1.6709892296463946,
2005
+ "learning_rate": 9.203543911285897e-06,
2006
+ "loss": 0.41599422693252563,
2007
+ "memory(GiB)": 128.42,
2008
+ "step": 200,
2009
+ "token_acc": 0.852233676975945,
2010
+ "train_speed(iter/s)": 0.227806
2011
+ },
2012
+ {
2013
+ "epoch": 0.8973214285714286,
2014
+ "grad_norm": 1.671640006565075,
2015
+ "learning_rate": 9.1935203831011e-06,
2016
+ "loss": 0.35883253812789917,
2017
+ "memory(GiB)": 128.42,
2018
+ "step": 201,
2019
+ "token_acc": 0.8736122971818958,
2020
+ "train_speed(iter/s)": 0.22781
2021
+ },
2022
+ {
2023
+ "epoch": 0.9017857142857143,
2024
+ "grad_norm": 1.6477395677283788,
2025
+ "learning_rate": 9.183439704566851e-06,
2026
+ "loss": 0.37051939964294434,
2027
+ "memory(GiB)": 128.42,
2028
+ "step": 202,
2029
+ "token_acc": 0.8587786259541985,
2030
+ "train_speed(iter/s)": 0.227826
2031
+ },
2032
+ {
2033
+ "epoch": 0.90625,
2034
+ "grad_norm": 1.853741854901314,
2035
+ "learning_rate": 9.173302013065168e-06,
2036
+ "loss": 0.4104780852794647,
2037
+ "memory(GiB)": 128.42,
2038
+ "step": 203,
2039
+ "token_acc": 0.8921810699588477,
2040
+ "train_speed(iter/s)": 0.227845
2041
+ },
2042
+ {
2043
+ "epoch": 0.9107142857142857,
2044
+ "grad_norm": 1.5402394296023612,
2045
+ "learning_rate": 9.163107446755058e-06,
2046
+ "loss": 0.3700962960720062,
2047
+ "memory(GiB)": 128.42,
2048
+ "step": 204,
2049
+ "token_acc": 0.8727615457115928,
2050
+ "train_speed(iter/s)": 0.227855
2051
+ },
2052
+ {
2053
+ "epoch": 0.9151785714285714,
2054
+ "grad_norm": 1.7657060068556503,
2055
+ "learning_rate": 9.152856144570633e-06,
2056
+ "loss": 0.40010184049606323,
2057
+ "memory(GiB)": 128.42,
2058
+ "step": 205,
2059
+ "token_acc": 0.8792415169660679,
2060
+ "train_speed(iter/s)": 0.227847
2061
+ },
2062
+ {
2063
+ "epoch": 0.9196428571428571,
2064
+ "grad_norm": 1.9095214293712117,
2065
+ "learning_rate": 9.142548246219212e-06,
2066
+ "loss": 0.41824835538864136,
2067
+ "memory(GiB)": 128.42,
2068
+ "step": 206,
2069
+ "token_acc": 0.8832159624413145,
2070
+ "train_speed(iter/s)": 0.227836
2071
+ },
2072
+ {
2073
+ "epoch": 0.9241071428571429,
2074
+ "grad_norm": 1.7097757171587031,
2075
+ "learning_rate": 9.132183892179428e-06,
2076
+ "loss": 0.3989163935184479,
2077
+ "memory(GiB)": 128.42,
2078
+ "step": 207,
2079
+ "token_acc": 0.8699690402476781,
2080
+ "train_speed(iter/s)": 0.227851
2081
+ },
2082
+ {
2083
+ "epoch": 0.9285714285714286,
2084
+ "grad_norm": 1.7525425330091031,
2085
+ "learning_rate": 9.121763223699297e-06,
2086
+ "loss": 0.384308785200119,
2087
+ "memory(GiB)": 128.42,
2088
+ "step": 208,
2089
+ "token_acc": 0.8719211822660099,
2090
+ "train_speed(iter/s)": 0.227863
2091
+ },
2092
+ {
2093
+ "epoch": 0.9330357142857143,
2094
+ "grad_norm": 1.687286542996637,
2095
+ "learning_rate": 9.111286382794313e-06,
2096
+ "loss": 0.39490386843681335,
2097
+ "memory(GiB)": 128.42,
2098
+ "step": 209,
2099
+ "token_acc": 0.8538461538461538,
2100
+ "train_speed(iter/s)": 0.227879
2101
+ },
2102
+ {
2103
+ "epoch": 0.9375,
2104
+ "grad_norm": 1.6728271677101256,
2105
+ "learning_rate": 9.100753512245493e-06,
2106
+ "loss": 0.40762174129486084,
2107
+ "memory(GiB)": 128.42,
2108
+ "step": 210,
2109
+ "token_acc": 0.8591439688715953,
2110
+ "train_speed(iter/s)": 0.227893
2111
+ },
2112
+ {
2113
+ "epoch": 0.9419642857142857,
2114
+ "grad_norm": 1.5297271530329113,
2115
+ "learning_rate": 9.090164755597444e-06,
2116
+ "loss": 0.33648815751075745,
2117
+ "memory(GiB)": 128.42,
2118
+ "step": 211,
2119
+ "token_acc": 0.8709454796411318,
2120
+ "train_speed(iter/s)": 0.227904
2121
+ },
2122
+ {
2123
+ "epoch": 0.9464285714285714,
2124
+ "grad_norm": 1.6329682526256801,
2125
+ "learning_rate": 9.079520257156402e-06,
2126
+ "loss": 0.3839446008205414,
2127
+ "memory(GiB)": 128.42,
2128
+ "step": 212,
2129
+ "token_acc": 0.8720073664825047,
2130
+ "train_speed(iter/s)": 0.227907
2131
+ },
2132
+ {
2133
+ "epoch": 0.9508928571428571,
2134
+ "grad_norm": 1.6995622246856035,
2135
+ "learning_rate": 9.068820161988268e-06,
2136
+ "loss": 0.37411996722221375,
2137
+ "memory(GiB)": 128.42,
2138
+ "step": 213,
2139
+ "token_acc": 0.8675246025738077,
2140
+ "train_speed(iter/s)": 0.22791
2141
+ },
2142
+ {
2143
+ "epoch": 0.9553571428571429,
2144
+ "grad_norm": 1.644210658432108,
2145
+ "learning_rate": 9.058064615916623e-06,
2146
+ "loss": 0.40447884798049927,
2147
+ "memory(GiB)": 128.42,
2148
+ "step": 214,
2149
+ "token_acc": 0.8810900082576383,
2150
+ "train_speed(iter/s)": 0.227909
2151
+ },
2152
+ {
2153
+ "epoch": 0.9598214285714286,
2154
+ "grad_norm": 1.73954688053968,
2155
+ "learning_rate": 9.047253765520758e-06,
2156
+ "loss": 0.3625633716583252,
2157
+ "memory(GiB)": 128.42,
2158
+ "step": 215,
2159
+ "token_acc": 0.8854718981972428,
2160
+ "train_speed(iter/s)": 0.227913
2161
+ },
2162
+ {
2163
+ "epoch": 0.9642857142857143,
2164
+ "grad_norm": 1.5014266406948364,
2165
+ "learning_rate": 9.036387758133653e-06,
2166
+ "loss": 0.35661768913269043,
2167
+ "memory(GiB)": 128.42,
2168
+ "step": 216,
2169
+ "token_acc": 0.8856275303643725,
2170
+ "train_speed(iter/s)": 0.227927
2171
+ },
2172
+ {
2173
+ "epoch": 0.96875,
2174
+ "grad_norm": 1.4981188941125985,
2175
+ "learning_rate": 9.025466741839987e-06,
2176
+ "loss": 0.3550759553909302,
2177
+ "memory(GiB)": 128.42,
2178
+ "step": 217,
2179
+ "token_acc": 0.9011786038077969,
2180
+ "train_speed(iter/s)": 0.227946
2181
+ },
2182
+ {
2183
+ "epoch": 0.9732142857142857,
2184
+ "grad_norm": 1.5386161305286659,
2185
+ "learning_rate": 9.014490865474119e-06,
2186
+ "loss": 0.36008650064468384,
2187
+ "memory(GiB)": 128.42,
2188
+ "step": 218,
2189
+ "token_acc": 0.8419429452582884,
2190
+ "train_speed(iter/s)": 0.227967
2191
+ },
2192
+ {
2193
+ "epoch": 0.9776785714285714,
2194
+ "grad_norm": 1.612365460152363,
2195
+ "learning_rate": 9.003460278618046e-06,
2196
+ "loss": 0.3536672592163086,
2197
+ "memory(GiB)": 128.42,
2198
+ "step": 219,
2199
+ "token_acc": 0.8841059602649006,
2200
+ "train_speed(iter/s)": 0.227984
2201
+ },
2202
+ {
2203
+ "epoch": 0.9821428571428571,
2204
+ "grad_norm": 1.5788262432034628,
2205
+ "learning_rate": 8.992375131599376e-06,
2206
+ "loss": 0.3935646712779999,
2207
+ "memory(GiB)": 128.42,
2208
+ "step": 220,
2209
+ "token_acc": 0.8902691511387164,
2210
+ "train_speed(iter/s)": 0.227957
2211
+ },
2212
+ {
2213
+ "epoch": 0.9866071428571429,
2214
+ "grad_norm": 1.642418411618006,
2215
+ "learning_rate": 8.981235575489282e-06,
2216
+ "loss": 0.36323970556259155,
2217
+ "memory(GiB)": 128.42,
2218
+ "step": 221,
2219
+ "token_acc": 0.8784956605593057,
2220
+ "train_speed(iter/s)": 0.227976
2221
+ },
2222
+ {
2223
+ "epoch": 0.9910714285714286,
2224
+ "grad_norm": 1.7315886339339615,
2225
+ "learning_rate": 8.97004176210043e-06,
2226
+ "loss": 0.3866032063961029,
2227
+ "memory(GiB)": 128.42,
2228
+ "step": 222,
2229
+ "token_acc": 0.8881188118811881,
2230
+ "train_speed(iter/s)": 0.22799
2231
+ },
2232
+ {
2233
+ "epoch": 0.9955357142857143,
2234
+ "grad_norm": 1.5614784501414387,
2235
+ "learning_rate": 8.958793843984923e-06,
2236
+ "loss": 0.3766946792602539,
2237
+ "memory(GiB)": 128.42,
2238
+ "step": 223,
2239
+ "token_acc": 0.8715124816446402,
2240
+ "train_speed(iter/s)": 0.228007
2241
+ },
2242
+ {
2243
+ "epoch": 1.0,
2244
+ "grad_norm": 1.5980875576332965,
2245
+ "learning_rate": 8.947491974432216e-06,
2246
+ "loss": 0.3782101273536682,
2247
+ "memory(GiB)": 128.42,
2248
+ "step": 224,
2249
+ "token_acc": 0.9090909090909091,
2250
+ "train_speed(iter/s)": 0.228013
2251
+ },
2252
+ {
2253
+ "epoch": 1.0,
2254
+ "eval_loss": 0.36971455812454224,
2255
+ "eval_runtime": 1.9584,
2256
+ "eval_samples_per_second": 2.042,
2257
+ "eval_steps_per_second": 0.511,
2258
+ "eval_token_acc": 0.8700199203187251,
2259
+ "step": 224
2260
+ }
2261
+ ],
2262
+ "logging_steps": 1,
2263
+ "max_steps": 896,
2264
+ "num_input_tokens_seen": 0,
2265
+ "num_train_epochs": 4,
2266
+ "save_steps": 500,
2267
+ "stateful_callbacks": {
2268
+ "TrainerControl": {
2269
+ "args": {
2270
+ "should_epoch_stop": false,
2271
+ "should_evaluate": false,
2272
+ "should_log": false,
2273
+ "should_save": true,
2274
+ "should_training_stop": false
2275
+ },
2276
+ "attributes": {}
2277
+ }
2278
+ },
2279
+ "total_flos": 114578270584832.0,
2280
+ "train_batch_size": 1,
2281
+ "trial_name": null,
2282
+ "trial_params": null
2283
+ }
v0-20251012-211828/checkpoint-224/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:88b1c4a5028e83060e213b58a43474a92ff6a3042e8f34900d9a7ab9f5639e64
3
+ size 8312
v0-20251012-211828/checkpoint-224/vocab.json ADDED
The diff for this file is too large to render. See raw diff
 
v0-20251012-211828/checkpoint-224/zero_to_fp32.py ADDED
@@ -0,0 +1,604 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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: python zero_to_fp32.py . pytorch_model.bin
14
+
15
+ import argparse
16
+ import torch
17
+ import glob
18
+ import math
19
+ import os
20
+ import re
21
+ from collections import OrderedDict
22
+ from dataclasses import dataclass
23
+
24
+ # while this script doesn't use deepspeed to recover data, since the checkpoints are pickled with
25
+ # DeepSpeed data structures it has to be available in the current python environment.
26
+ from deepspeed.utils import logger
27
+ from deepspeed.checkpoint.constants import (DS_VERSION, OPTIMIZER_STATE_DICT, SINGLE_PARTITION_OF_FP32_GROUPS,
28
+ FP32_FLAT_GROUPS, ZERO_STAGE, PARTITION_COUNT, PARAM_SHAPES, BUFFER_NAMES,
29
+ FROZEN_PARAM_SHAPES, FROZEN_PARAM_FRAGMENTS)
30
+
31
+
32
+ @dataclass
33
+ class zero_model_state:
34
+ buffers: dict()
35
+ param_shapes: dict()
36
+ shared_params: list
37
+ ds_version: int
38
+ frozen_param_shapes: dict()
39
+ frozen_param_fragments: dict()
40
+
41
+
42
+ debug = 0
43
+
44
+ # load to cpu
45
+ device = torch.device('cpu')
46
+
47
+
48
+ def atoi(text):
49
+ return int(text) if text.isdigit() else text
50
+
51
+
52
+ def natural_keys(text):
53
+ '''
54
+ alist.sort(key=natural_keys) sorts in human order
55
+ http://nedbatchelder.com/blog/200712/human_sorting.html
56
+ (See Toothy's implementation in the comments)
57
+ '''
58
+ return [atoi(c) for c in re.split(r'(\d+)', text)]
59
+
60
+
61
+ def get_model_state_file(checkpoint_dir, zero_stage):
62
+ if not os.path.isdir(checkpoint_dir):
63
+ raise FileNotFoundError(f"Directory '{checkpoint_dir}' doesn't exist")
64
+
65
+ # there should be only one file
66
+ if zero_stage <= 2:
67
+ file = os.path.join(checkpoint_dir, "mp_rank_00_model_states.pt")
68
+ elif zero_stage == 3:
69
+ file = os.path.join(checkpoint_dir, "zero_pp_rank_0_mp_rank_00_model_states.pt")
70
+
71
+ if not os.path.exists(file):
72
+ raise FileNotFoundError(f"can't find model states file at '{file}'")
73
+
74
+ return file
75
+
76
+
77
+ def get_checkpoint_files(checkpoint_dir, glob_pattern):
78
+ # XXX: need to test that this simple glob rule works for multi-node setup too
79
+ ckpt_files = sorted(glob.glob(os.path.join(checkpoint_dir, glob_pattern)), key=natural_keys)
80
+
81
+ if len(ckpt_files) == 0:
82
+ raise FileNotFoundError(f"can't find {glob_pattern} files in directory '{checkpoint_dir}'")
83
+
84
+ return ckpt_files
85
+
86
+
87
+ def get_optim_files(checkpoint_dir):
88
+ return get_checkpoint_files(checkpoint_dir, "*_optim_states.pt")
89
+
90
+
91
+ def get_model_state_files(checkpoint_dir):
92
+ return get_checkpoint_files(checkpoint_dir, "*_model_states.pt")
93
+
94
+
95
+ def parse_model_states(files):
96
+ zero_model_states = []
97
+ for file in files:
98
+ state_dict = torch.load(file, map_location=device)
99
+
100
+ if BUFFER_NAMES not in state_dict:
101
+ raise ValueError(f"{file} is not a model state checkpoint")
102
+ buffer_names = state_dict[BUFFER_NAMES]
103
+ if debug:
104
+ print("Found buffers:", buffer_names)
105
+
106
+ # recover just the buffers while restoring them to fp32 if they were saved in fp16
107
+ buffers = {k: v.float() for k, v in state_dict["module"].items() if k in buffer_names}
108
+ param_shapes = state_dict[PARAM_SHAPES]
109
+
110
+ # collect parameters that are included in param_shapes
111
+ param_names = []
112
+ for s in param_shapes:
113
+ for name in s.keys():
114
+ param_names.append(name)
115
+
116
+ # update with frozen parameters
117
+ frozen_param_shapes = state_dict.get(FROZEN_PARAM_SHAPES, None)
118
+ if frozen_param_shapes is not None:
119
+ if debug:
120
+ print(f"Found frozen_param_shapes: {frozen_param_shapes}")
121
+ param_names += list(frozen_param_shapes.keys())
122
+
123
+ # handle shared params
124
+ shared_params = [[k, v] for k, v in state_dict["shared_params"].items()]
125
+
126
+ ds_version = state_dict.get(DS_VERSION, None)
127
+
128
+ frozen_param_fragments = state_dict.get(FROZEN_PARAM_FRAGMENTS, None)
129
+
130
+ z_model_state = zero_model_state(buffers=buffers,
131
+ param_shapes=param_shapes,
132
+ shared_params=shared_params,
133
+ ds_version=ds_version,
134
+ frozen_param_shapes=frozen_param_shapes,
135
+ frozen_param_fragments=frozen_param_fragments)
136
+ zero_model_states.append(z_model_state)
137
+
138
+ return zero_model_states
139
+
140
+
141
+ def parse_optim_states(files, ds_checkpoint_dir):
142
+
143
+ total_files = len(files)
144
+ state_dicts = []
145
+ for f in files:
146
+ state_dict = torch.load(f, map_location=device)
147
+ # immediately discard the potentially huge 2 optimizer states as we only care for fp32 master weights
148
+ # and also handle the case where it was already removed by another helper script
149
+ state_dict["optimizer_state_dict"].pop("optimizer_state_dict", None)
150
+ state_dicts.append(state_dict)
151
+
152
+ if not ZERO_STAGE in state_dicts[0][OPTIMIZER_STATE_DICT]:
153
+ raise ValueError(f"{files[0]} is not a zero checkpoint")
154
+ zero_stage = state_dicts[0][OPTIMIZER_STATE_DICT][ZERO_STAGE]
155
+ world_size = state_dicts[0][OPTIMIZER_STATE_DICT][PARTITION_COUNT]
156
+
157
+ # For ZeRO-2 each param group can have different partition_count as data parallelism for expert
158
+ # parameters can be different from data parallelism for non-expert parameters. So we can just
159
+ # use the max of the partition_count to get the dp world_size.
160
+
161
+ if type(world_size) is list:
162
+ world_size = max(world_size)
163
+
164
+ if world_size != total_files:
165
+ raise ValueError(
166
+ f"Expected {world_size} of '*_optim_states.pt' under '{ds_checkpoint_dir}' but found {total_files} files. "
167
+ "Possibly due to an overwrite of an old checkpoint, or a checkpoint didn't get saved by one or more processes."
168
+ )
169
+
170
+ # the groups are named differently in each stage
171
+ if zero_stage <= 2:
172
+ fp32_groups_key = SINGLE_PARTITION_OF_FP32_GROUPS
173
+ elif zero_stage == 3:
174
+ fp32_groups_key = FP32_FLAT_GROUPS
175
+ else:
176
+ raise ValueError(f"unknown zero stage {zero_stage}")
177
+
178
+ if zero_stage <= 2:
179
+ fp32_flat_groups = [state_dicts[i][OPTIMIZER_STATE_DICT][fp32_groups_key] for i in range(len(state_dicts))]
180
+ elif zero_stage == 3:
181
+ # if there is more than one param group, there will be multiple flattened tensors - one
182
+ # flattened tensor per group - for simplicity merge them into a single tensor
183
+ #
184
+ # XXX: could make the script more memory efficient for when there are multiple groups - it
185
+ # will require matching the sub-lists of param_shapes for each param group flattened tensor
186
+
187
+ fp32_flat_groups = [
188
+ torch.cat(state_dicts[i][OPTIMIZER_STATE_DICT][fp32_groups_key], 0) for i in range(len(state_dicts))
189
+ ]
190
+
191
+ return zero_stage, world_size, fp32_flat_groups
192
+
193
+
194
+ def _get_fp32_state_dict_from_zero_checkpoint(ds_checkpoint_dir, exclude_frozen_parameters):
195
+ """
196
+ Returns fp32 state_dict reconstructed from ds checkpoint
197
+
198
+ Args:
199
+ - ``ds_checkpoint_dir``: path to the deepspeed checkpoint folder (where the optimizer files are)
200
+
201
+ """
202
+ print(f"Processing zero checkpoint '{ds_checkpoint_dir}'")
203
+
204
+ optim_files = get_optim_files(ds_checkpoint_dir)
205
+ zero_stage, world_size, fp32_flat_groups = parse_optim_states(optim_files, ds_checkpoint_dir)
206
+ print(f"Detected checkpoint of type zero stage {zero_stage}, world_size: {world_size}")
207
+
208
+ model_files = get_model_state_files(ds_checkpoint_dir)
209
+
210
+ zero_model_states = parse_model_states(model_files)
211
+ print(f'Parsing checkpoint created by deepspeed=={zero_model_states[0].ds_version}')
212
+
213
+ if zero_stage <= 2:
214
+ return _get_fp32_state_dict_from_zero2_checkpoint(world_size, fp32_flat_groups, zero_model_states,
215
+ exclude_frozen_parameters)
216
+ elif zero_stage == 3:
217
+ return _get_fp32_state_dict_from_zero3_checkpoint(world_size, fp32_flat_groups, zero_model_states,
218
+ exclude_frozen_parameters)
219
+
220
+
221
+ def _zero2_merge_frozen_params(state_dict, zero_model_states):
222
+ if zero_model_states[0].frozen_param_shapes is None or len(zero_model_states[0].frozen_param_shapes) == 0:
223
+ return
224
+
225
+ frozen_param_shapes = zero_model_states[0].frozen_param_shapes
226
+ frozen_param_fragments = zero_model_states[0].frozen_param_fragments
227
+
228
+ if debug:
229
+ num_elem = sum(s.numel() for s in frozen_param_shapes.values())
230
+ print(f'rank 0: {FROZEN_PARAM_SHAPES}.numel = {num_elem}')
231
+
232
+ wanted_params = len(frozen_param_shapes)
233
+ wanted_numel = sum(s.numel() for s in frozen_param_shapes.values())
234
+ avail_numel = sum([p.numel() for p in frozen_param_fragments.values()])
235
+ print(f'Frozen params: Have {avail_numel} numels to process.')
236
+ print(f'Frozen params: Need {wanted_numel} numels in {wanted_params} params')
237
+
238
+ total_params = 0
239
+ total_numel = 0
240
+ for name, shape in frozen_param_shapes.items():
241
+ total_params += 1
242
+ unpartitioned_numel = shape.numel()
243
+ total_numel += unpartitioned_numel
244
+
245
+ state_dict[name] = frozen_param_fragments[name]
246
+
247
+ if debug:
248
+ print(f"{name} full shape: {shape} unpartitioned numel {unpartitioned_numel} ")
249
+
250
+ print(f"Reconstructed Frozen fp32 state dict with {total_params} params {total_numel} elements")
251
+
252
+
253
+ def _has_callable(obj, fn):
254
+ attr = getattr(obj, fn, None)
255
+ return callable(attr)
256
+
257
+
258
+ def _zero2_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states):
259
+ param_shapes = zero_model_states[0].param_shapes
260
+
261
+ # Reconstruction protocol:
262
+ #
263
+ # XXX: document this
264
+
265
+ if debug:
266
+ for i in range(world_size):
267
+ for j in range(len(fp32_flat_groups[0])):
268
+ print(f"{FP32_FLAT_GROUPS}[{i}][{j}].shape={fp32_flat_groups[i][j].shape}")
269
+
270
+ # XXX: memory usage doubles here (zero2)
271
+ num_param_groups = len(fp32_flat_groups[0])
272
+ merged_single_partition_of_fp32_groups = []
273
+ for i in range(num_param_groups):
274
+ merged_partitions = [sd[i] for sd in fp32_flat_groups]
275
+ full_single_fp32_vector = torch.cat(merged_partitions, 0)
276
+ merged_single_partition_of_fp32_groups.append(full_single_fp32_vector)
277
+ avail_numel = sum(
278
+ [full_single_fp32_vector.numel() for full_single_fp32_vector in merged_single_partition_of_fp32_groups])
279
+
280
+ if debug:
281
+ wanted_params = sum([len(shapes) for shapes in param_shapes])
282
+ wanted_numel = sum([sum(shape.numel() for shape in shapes.values()) for shapes in param_shapes])
283
+ # not asserting if there is a mismatch due to possible padding
284
+ print(f"Have {avail_numel} numels to process.")
285
+ print(f"Need {wanted_numel} numels in {wanted_params} params.")
286
+
287
+ # params
288
+ # XXX: for huge models that can't fit into the host's RAM we will have to recode this to support
289
+ # out-of-core computing solution
290
+ total_numel = 0
291
+ total_params = 0
292
+ for shapes, full_single_fp32_vector in zip(param_shapes, merged_single_partition_of_fp32_groups):
293
+ offset = 0
294
+ avail_numel = full_single_fp32_vector.numel()
295
+ for name, shape in shapes.items():
296
+
297
+ unpartitioned_numel = shape.numel() if _has_callable(shape, 'numel') else math.prod(shape)
298
+ total_numel += unpartitioned_numel
299
+ total_params += 1
300
+
301
+ if debug:
302
+ print(f"{name} full shape: {shape} unpartitioned numel {unpartitioned_numel} ")
303
+ state_dict[name] = full_single_fp32_vector.narrow(0, offset, unpartitioned_numel).view(shape)
304
+ offset += unpartitioned_numel
305
+
306
+ # Z2 started to align to 2*world_size to improve nccl performance. Therefore both offset and
307
+ # avail_numel can differ by anywhere between 0..2*world_size. Due to two unrelated complex
308
+ # paddings performed in the code it's almost impossible to predict the exact numbers w/o the
309
+ # live optimizer object, so we are checking that the numbers are within the right range
310
+ align_to = 2 * world_size
311
+
312
+ def zero2_align(x):
313
+ return align_to * math.ceil(x / align_to)
314
+
315
+ if debug:
316
+ print(f"original offset={offset}, avail_numel={avail_numel}")
317
+
318
+ offset = zero2_align(offset)
319
+ avail_numel = zero2_align(avail_numel)
320
+
321
+ if debug:
322
+ print(f"aligned offset={offset}, avail_numel={avail_numel}")
323
+
324
+ # Sanity check
325
+ if offset != avail_numel:
326
+ raise ValueError(f"consumed {offset} numels out of {avail_numel} - something is wrong")
327
+
328
+ print(f"Reconstructed fp32 state dict with {total_params} params {total_numel} elements")
329
+
330
+
331
+ def _get_fp32_state_dict_from_zero2_checkpoint(world_size, fp32_flat_groups, zero_model_states,
332
+ exclude_frozen_parameters):
333
+ state_dict = OrderedDict()
334
+
335
+ # buffers
336
+ buffers = zero_model_states[0].buffers
337
+ state_dict.update(buffers)
338
+ if debug:
339
+ print(f"added {len(buffers)} buffers")
340
+
341
+ if not exclude_frozen_parameters:
342
+ _zero2_merge_frozen_params(state_dict, zero_model_states)
343
+
344
+ _zero2_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states)
345
+
346
+ # recover shared parameters
347
+ for pair in zero_model_states[0].shared_params:
348
+ if pair[1] in state_dict:
349
+ state_dict[pair[0]] = state_dict[pair[1]]
350
+
351
+ return state_dict
352
+
353
+
354
+ def zero3_partitioned_param_info(unpartitioned_numel, world_size):
355
+ remainder = unpartitioned_numel % world_size
356
+ padding_numel = (world_size - remainder) if remainder else 0
357
+ partitioned_numel = math.ceil(unpartitioned_numel / world_size)
358
+ return partitioned_numel, padding_numel
359
+
360
+
361
+ def _zero3_merge_frozen_params(state_dict, world_size, zero_model_states):
362
+ if zero_model_states[0].frozen_param_shapes is None or len(zero_model_states[0].frozen_param_shapes) == 0:
363
+ return
364
+
365
+ if debug:
366
+ for i in range(world_size):
367
+ num_elem = sum(s.numel() for s in zero_model_states[i].frozen_param_fragments.values())
368
+ print(f'rank {i}: {FROZEN_PARAM_SHAPES}.numel = {num_elem}')
369
+
370
+ frozen_param_shapes = zero_model_states[0].frozen_param_shapes
371
+ wanted_params = len(frozen_param_shapes)
372
+ wanted_numel = sum(s.numel() for s in frozen_param_shapes.values())
373
+ avail_numel = sum([p.numel() for p in zero_model_states[0].frozen_param_fragments.values()]) * world_size
374
+ print(f'Frozen params: Have {avail_numel} numels to process.')
375
+ print(f'Frozen params: Need {wanted_numel} numels in {wanted_params} params')
376
+
377
+ total_params = 0
378
+ total_numel = 0
379
+ for name, shape in zero_model_states[0].frozen_param_shapes.items():
380
+ total_params += 1
381
+ unpartitioned_numel = shape.numel()
382
+ total_numel += unpartitioned_numel
383
+
384
+ param_frags = tuple(model_state.frozen_param_fragments[name] for model_state in zero_model_states)
385
+ state_dict[name] = torch.cat(param_frags, 0).narrow(0, 0, unpartitioned_numel).view(shape)
386
+
387
+ partitioned_numel, partitioned_padding_numel = zero3_partitioned_param_info(unpartitioned_numel, world_size)
388
+
389
+ if debug:
390
+ print(
391
+ f"Frozen params: {total_params} {name} full shape: {shape} partition0 numel={partitioned_numel} partitioned_padding_numel={partitioned_padding_numel}"
392
+ )
393
+
394
+ print(f"Reconstructed Frozen fp32 state dict with {total_params} params {total_numel} elements")
395
+
396
+
397
+ def _zero3_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states):
398
+ param_shapes = zero_model_states[0].param_shapes
399
+ avail_numel = fp32_flat_groups[0].numel() * world_size
400
+ # Reconstruction protocol: For zero3 we need to zip the partitions together at boundary of each
401
+ # param, re-consolidating each param, while dealing with padding if any
402
+
403
+ # merge list of dicts, preserving order
404
+ param_shapes = {k: v for d in param_shapes for k, v in d.items()}
405
+
406
+ if debug:
407
+ for i in range(world_size):
408
+ print(f"{FP32_FLAT_GROUPS}[{i}].shape={fp32_flat_groups[i].shape}")
409
+
410
+ wanted_params = len(param_shapes)
411
+ wanted_numel = sum(shape.numel() for shape in param_shapes.values())
412
+ # not asserting if there is a mismatch due to possible padding
413
+ avail_numel = fp32_flat_groups[0].numel() * world_size
414
+ print(f"Trainable params: Have {avail_numel} numels to process.")
415
+ print(f"Trainable params: Need {wanted_numel} numels in {wanted_params} params.")
416
+
417
+ # params
418
+ # XXX: for huge models that can't fit into the host's RAM we will have to recode this to support
419
+ # out-of-core computing solution
420
+ offset = 0
421
+ total_numel = 0
422
+ total_params = 0
423
+ for name, shape in param_shapes.items():
424
+
425
+ unpartitioned_numel = shape.numel()
426
+ total_numel += unpartitioned_numel
427
+ total_params += 1
428
+
429
+ partitioned_numel, partitioned_padding_numel = zero3_partitioned_param_info(unpartitioned_numel, world_size)
430
+
431
+ if debug:
432
+ print(
433
+ f"Trainable params: {total_params} {name} full shape: {shape} partition0 numel={partitioned_numel} partitioned_padding_numel={partitioned_padding_numel}"
434
+ )
435
+
436
+ # XXX: memory usage doubles here
437
+ state_dict[name] = torch.cat(
438
+ tuple(fp32_flat_groups[i].narrow(0, offset, partitioned_numel) for i in range(world_size)),
439
+ 0).narrow(0, 0, unpartitioned_numel).view(shape)
440
+ offset += partitioned_numel
441
+
442
+ offset *= world_size
443
+
444
+ # Sanity check
445
+ if offset != avail_numel:
446
+ raise ValueError(f"consumed {offset} numels out of {avail_numel} - something is wrong")
447
+
448
+ print(f"Reconstructed Trainable fp32 state dict with {total_params} params {total_numel} elements")
449
+
450
+
451
+ def _get_fp32_state_dict_from_zero3_checkpoint(world_size, fp32_flat_groups, zero_model_states,
452
+ exclude_frozen_parameters):
453
+ state_dict = OrderedDict()
454
+
455
+ # buffers
456
+ buffers = zero_model_states[0].buffers
457
+ state_dict.update(buffers)
458
+ if debug:
459
+ print(f"added {len(buffers)} buffers")
460
+
461
+ if not exclude_frozen_parameters:
462
+ _zero3_merge_frozen_params(state_dict, world_size, zero_model_states)
463
+
464
+ _zero3_merge_trainable_params(state_dict, world_size, fp32_flat_groups, zero_model_states)
465
+
466
+ # recover shared parameters
467
+ for pair in zero_model_states[0].shared_params:
468
+ if pair[1] in state_dict:
469
+ state_dict[pair[0]] = state_dict[pair[1]]
470
+
471
+ return state_dict
472
+
473
+
474
+ def get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir, tag=None, exclude_frozen_parameters=False):
475
+ """
476
+ Convert ZeRO 2 or 3 checkpoint into a single fp32 consolidated state_dict that can be loaded with
477
+ ``load_state_dict()`` and used for training without DeepSpeed or shared with others, for example
478
+ via a model hub.
479
+
480
+ Args:
481
+ - ``checkpoint_dir``: path to the desired checkpoint folder
482
+ - ``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``
483
+ - ``exclude_frozen_parameters``: exclude frozen parameters
484
+
485
+ Returns:
486
+ - pytorch ``state_dict``
487
+
488
+ Note: this approach may not work if your application doesn't have sufficient free CPU memory and
489
+ you may need to use the offline approach using the ``zero_to_fp32.py`` script that is saved with
490
+ the checkpoint.
491
+
492
+ A typical usage might be ::
493
+
494
+ from deepspeed.utils.zero_to_fp32 import get_fp32_state_dict_from_zero_checkpoint
495
+ # do the training and checkpoint saving
496
+ state_dict = get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir) # already on cpu
497
+ model = model.cpu() # move to cpu
498
+ model.load_state_dict(state_dict)
499
+ # submit to model hub or save the model to share with others
500
+
501
+ In this example the ``model`` will no longer be usable in the deepspeed context of the same
502
+ application. i.e. you will need to re-initialize the deepspeed engine, since
503
+ ``model.load_state_dict(state_dict)`` will remove all the deepspeed magic from it.
504
+
505
+ If you want it all done for you, use ``load_state_dict_from_zero_checkpoint`` instead.
506
+
507
+ """
508
+ if tag is None:
509
+ latest_path = os.path.join(checkpoint_dir, 'latest')
510
+ if os.path.isfile(latest_path):
511
+ with open(latest_path, 'r') as fd:
512
+ tag = fd.read().strip()
513
+ else:
514
+ raise ValueError(f"Unable to find 'latest' file at {latest_path}")
515
+
516
+ ds_checkpoint_dir = os.path.join(checkpoint_dir, tag)
517
+
518
+ if not os.path.isdir(ds_checkpoint_dir):
519
+ raise FileNotFoundError(f"Directory '{ds_checkpoint_dir}' doesn't exist")
520
+
521
+ return _get_fp32_state_dict_from_zero_checkpoint(ds_checkpoint_dir, exclude_frozen_parameters)
522
+
523
+
524
+ def convert_zero_checkpoint_to_fp32_state_dict(checkpoint_dir, output_file, tag=None, exclude_frozen_parameters=False):
525
+ """
526
+ Convert ZeRO 2 or 3 checkpoint into a single fp32 consolidated ``state_dict`` file that can be
527
+ loaded with ``torch.load(file)`` + ``load_state_dict()`` and used for training without DeepSpeed.
528
+
529
+ Args:
530
+ - ``checkpoint_dir``: path to the desired checkpoint folder. (one that contains the tag-folder, like ``global_step14``)
531
+ - ``output_file``: path to the pytorch fp32 state_dict output file (e.g. path/pytorch_model.bin)
532
+ - ``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``
533
+ - ``exclude_frozen_parameters``: exclude frozen parameters
534
+ """
535
+
536
+ state_dict = get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir, tag, exclude_frozen_parameters)
537
+ print(f"Saving fp32 state dict to {output_file}")
538
+ torch.save(state_dict, output_file)
539
+
540
+
541
+ def load_state_dict_from_zero_checkpoint(model, checkpoint_dir, tag=None):
542
+ """
543
+ 1. Put the provided model to cpu
544
+ 2. Convert ZeRO 2 or 3 checkpoint into a single fp32 consolidated ``state_dict``
545
+ 3. Load it into the provided model
546
+
547
+ Args:
548
+ - ``model``: the model object to update
549
+ - ``checkpoint_dir``: path to the desired checkpoint folder. (one that contains the tag-folder, like ``global_step14``)
550
+ - ``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``
551
+
552
+ Returns:
553
+ - ``model`: modified model
554
+
555
+ Make sure you have plenty of CPU memory available before you call this function. If you don't
556
+ have enough use the ``zero_to_fp32.py`` utility to do the conversion. You will find it
557
+ conveniently placed for you in the checkpoint folder.
558
+
559
+ A typical usage might be ::
560
+
561
+ from deepspeed.utils.zero_to_fp32 import load_state_dict_from_zero_checkpoint
562
+ model = load_state_dict_from_zero_checkpoint(trainer.model, checkpoint_dir)
563
+ # submit to model hub or save the model to share with others
564
+
565
+ Note, that once this was run, the ``model`` will no longer be usable in the deepspeed context
566
+ of the same application. i.e. you will need to re-initialize the deepspeed engine, since
567
+ ``model.load_state_dict(state_dict)`` will remove all the deepspeed magic from it.
568
+
569
+ """
570
+ logger.info(f"Extracting fp32 weights")
571
+ state_dict = get_fp32_state_dict_from_zero_checkpoint(checkpoint_dir, tag)
572
+
573
+ logger.info(f"Overwriting model with fp32 weights")
574
+ model = model.cpu()
575
+ model.load_state_dict(state_dict, strict=False)
576
+
577
+ return model
578
+
579
+
580
+ if __name__ == "__main__":
581
+
582
+ parser = argparse.ArgumentParser()
583
+ parser.add_argument("checkpoint_dir",
584
+ type=str,
585
+ help="path to the desired checkpoint folder, e.g., path/checkpoint-12")
586
+ parser.add_argument(
587
+ "output_file",
588
+ type=str,
589
+ help="path to the pytorch fp32 state_dict output file (e.g. path/checkpoint-12/pytorch_model.bin)")
590
+ parser.add_argument("-t",
591
+ "--tag",
592
+ type=str,
593
+ default=None,
594
+ help="checkpoint tag used as a unique identifier for checkpoint. e.g., global_step1")
595
+ parser.add_argument("--exclude_frozen_parameters", action='store_true', help="exclude frozen parameters")
596
+ parser.add_argument("-d", "--debug", action='store_true', help="enable debug")
597
+ args = parser.parse_args()
598
+
599
+ debug = args.debug
600
+
601
+ convert_zero_checkpoint_to_fp32_state_dict(args.checkpoint_dir,
602
+ args.output_file,
603
+ tag=args.tag,
604
+ exclude_frozen_parameters=args.exclude_frozen_parameters)
v0-20251012-211828/checkpoint-448/added_tokens.json ADDED
@@ -0,0 +1,24 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "</tool_call>": 151658,
3
+ "<tool_call>": 151657,
4
+ "<|box_end|>": 151649,
5
+ "<|box_start|>": 151648,
6
+ "<|endoftext|>": 151643,
7
+ "<|file_sep|>": 151664,
8
+ "<|fim_middle|>": 151660,
9
+ "<|fim_pad|>": 151662,
10
+ "<|fim_prefix|>": 151659,
11
+ "<|fim_suffix|>": 151661,
12
+ "<|im_end|>": 151645,
13
+ "<|im_start|>": 151644,
14
+ "<|image_pad|>": 151655,
15
+ "<|object_ref_end|>": 151647,
16
+ "<|object_ref_start|>": 151646,
17
+ "<|quad_end|>": 151651,
18
+ "<|quad_start|>": 151650,
19
+ "<|repo_name|>": 151663,
20
+ "<|video_pad|>": 151656,
21
+ "<|vision_end|>": 151653,
22
+ "<|vision_pad|>": 151654,
23
+ "<|vision_start|>": 151652
24
+ }
v0-20251012-211828/checkpoint-448/args.json ADDED
@@ -0,0 +1,369 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model": "/data/minimax-dialogue/users/ruobai/cogito/base_model/Qwen2.5-7B",
3
+ "model_type": "qwen2_5",
4
+ "model_revision": null,
5
+ "task_type": "causal_lm",
6
+ "torch_dtype": "bfloat16",
7
+ "attn_impl": "flash_attn",
8
+ "num_labels": null,
9
+ "problem_type": null,
10
+ "rope_scaling": null,
11
+ "device_map": null,
12
+ "max_memory": {},
13
+ "local_repo_path": null,
14
+ "init_strategy": null,
15
+ "template": "qwen2_5",
16
+ "system": null,
17
+ "max_length": 16240,
18
+ "truncation_strategy": "delete",
19
+ "max_pixels": null,
20
+ "agent_template": null,
21
+ "norm_bbox": null,
22
+ "response_prefix": null,
23
+ "padding_side": "right",
24
+ "loss_scale": "default",
25
+ "sequence_parallel_size": 1,
26
+ "use_chat_template": true,
27
+ "template_backend": "swift",
28
+ "dataset": [
29
+ "/data/minimax-dialogue/users/ruobai/cogito/base_dataset/broswer_training_data/merged_data.jsonl"
30
+ ],
31
+ "val_dataset": [],
32
+ "split_dataset_ratio": 0.001,
33
+ "data_seed": 42,
34
+ "dataset_num_proc": 100,
35
+ "load_from_cache_file": true,
36
+ "dataset_shuffle": true,
37
+ "val_dataset_shuffle": false,
38
+ "streaming": false,
39
+ "interleave_prob": null,
40
+ "stopping_strategy": "first_exhausted",
41
+ "shuffle_buffer_size": 1000,
42
+ "download_mode": "reuse_dataset_if_exists",
43
+ "columns": {},
44
+ "strict": false,
45
+ "remove_unused_columns": false,
46
+ "model_name": [
47
+ null,
48
+ null
49
+ ],
50
+ "model_author": [
51
+ null,
52
+ null
53
+ ],
54
+ "custom_dataset_info": [],
55
+ "quant_method": null,
56
+ "quant_bits": null,
57
+ "hqq_axis": null,
58
+ "bnb_4bit_compute_dtype": "bfloat16",
59
+ "bnb_4bit_quant_type": "nf4",
60
+ "bnb_4bit_use_double_quant": true,
61
+ "bnb_4bit_quant_storage": null,
62
+ "max_new_tokens": 64,
63
+ "temperature": 0.0,
64
+ "top_k": null,
65
+ "top_p": null,
66
+ "repetition_penalty": null,
67
+ "num_beams": 1,
68
+ "stream": false,
69
+ "stop_words": [],
70
+ "logprobs": false,
71
+ "top_logprobs": null,
72
+ "ckpt_dir": null,
73
+ "lora_modules": [],
74
+ "tuner_backend": "peft",
75
+ "train_type": "full",
76
+ "adapters": [],
77
+ "external_plugins": [],
78
+ "seed": 42,
79
+ "model_kwargs": {},
80
+ "load_args": false,
81
+ "load_data_args": false,
82
+ "use_hf": true,
83
+ "hub_token": null,
84
+ "custom_register_path": [],
85
+ "ddp_timeout": 1800,
86
+ "ddp_backend": "nccl",
87
+ "ignore_args_error": false,
88
+ "use_swift_lora": false,
89
+ "output_dir": "/data/minimax-dialogue/users/ruobai/swift/model/qwen2p5_7B_ba_1012/v0-20251012-211828",
90
+ "overwrite_output_dir": false,
91
+ "do_train": false,
92
+ "do_eval": false,
93
+ "do_predict": false,
94
+ "eval_strategy": "epoch",
95
+ "prediction_loss_only": false,
96
+ "per_device_train_batch_size": 1,
97
+ "per_device_eval_batch_size": 1,
98
+ "per_gpu_train_batch_size": null,
99
+ "per_gpu_eval_batch_size": null,
100
+ "gradient_accumulation_steps": 2,
101
+ "eval_accumulation_steps": null,
102
+ "eval_delay": 0,
103
+ "torch_empty_cache_steps": null,
104
+ "learning_rate": 1e-05,
105
+ "weight_decay": 0.05,
106
+ "adam_beta1": 0.9,
107
+ "adam_beta2": 0.95,
108
+ "adam_epsilon": 1e-08,
109
+ "max_grad_norm": 1.0,
110
+ "num_train_epochs": 4.0,
111
+ "max_steps": -1,
112
+ "lr_scheduler_type": "cosine",
113
+ "lr_scheduler_kwargs": null,
114
+ "warmup_ratio": 0.05,
115
+ "warmup_steps": 0,
116
+ "log_level": "passive",
117
+ "log_level_replica": "warning",
118
+ "log_on_each_node": true,
119
+ "logging_dir": "/data/minimax-dialogue/users/ruobai/swift/model/qwen2p5_7B_ba_1012/v0-20251012-211828/runs",
120
+ "logging_strategy": "steps",
121
+ "logging_first_step": true,
122
+ "logging_steps": 1,
123
+ "logging_nan_inf_filter": true,
124
+ "save_strategy": "epoch",
125
+ "save_steps": 500,
126
+ "save_total_limit": null,
127
+ "save_safetensors": true,
128
+ "save_on_each_node": false,
129
+ "save_only_model": false,
130
+ "restore_callback_states_from_checkpoint": false,
131
+ "no_cuda": false,
132
+ "use_cpu": false,
133
+ "use_mps_device": false,
134
+ "jit_mode_eval": false,
135
+ "use_ipex": false,
136
+ "bf16": true,
137
+ "fp16": false,
138
+ "fp16_opt_level": "O1",
139
+ "half_precision_backend": "auto",
140
+ "bf16_full_eval": false,
141
+ "fp16_full_eval": false,
142
+ "tf32": null,
143
+ "local_rank": 0,
144
+ "tpu_num_cores": null,
145
+ "tpu_metrics_debug": false,
146
+ "debug": null,
147
+ "dataloader_drop_last": false,
148
+ "eval_steps": 20.0,
149
+ "dataloader_num_workers": 48,
150
+ "dataloader_prefetch_factor": null,
151
+ "past_index": -1,
152
+ "run_name": "/data/minimax-dialogue/users/ruobai/swift/model/qwen2p5_7B_ba_1012/v0-20251012-211828",
153
+ "disable_tqdm": null,
154
+ "label_names": null,
155
+ "load_best_model_at_end": false,
156
+ "metric_for_best_model": "loss",
157
+ "greater_is_better": false,
158
+ "ignore_data_skip": false,
159
+ "fsdp": "",
160
+ "fsdp_min_num_params": 0,
161
+ "fsdp_config": null,
162
+ "tp_size": 0,
163
+ "fsdp_transformer_layer_cls_to_wrap": null,
164
+ "accelerator_config": {
165
+ "dispatch_batches": false
166
+ },
167
+ "deepspeed": {
168
+ "fp16": {
169
+ "enabled": "auto",
170
+ "loss_scale": 0,
171
+ "loss_scale_window": 1000,
172
+ "initial_scale_power": 16,
173
+ "hysteresis": 2,
174
+ "min_loss_scale": 1
175
+ },
176
+ "bf16": {
177
+ "enabled": "auto"
178
+ },
179
+ "zero_optimization": {
180
+ "stage": 3,
181
+ "offload_optimizer": {
182
+ "device": "none",
183
+ "pin_memory": true
184
+ },
185
+ "offload_param": {
186
+ "device": "none",
187
+ "pin_memory": true
188
+ },
189
+ "overlap_comm": false,
190
+ "contiguous_gradients": true,
191
+ "sub_group_size": 1000000000.0,
192
+ "reduce_bucket_size": "auto",
193
+ "zero_quantized_weights": false,
194
+ "zero_quantized_gradients": false,
195
+ "stage3_prefetch_bucket_size": "auto",
196
+ "stage3_param_persistence_threshold": "auto",
197
+ "stage3_max_live_parameters": 1000000000.0,
198
+ "stage3_max_reuse_distance": 1000000000.0,
199
+ "stage3_gather_16bit_weights_on_model_save": true
200
+ },
201
+ "gradient_accumulation_steps": "auto",
202
+ "gradient_clipping": "auto",
203
+ "steps_per_print": 2000,
204
+ "train_batch_size": "auto",
205
+ "train_micro_batch_size_per_gpu": "auto",
206
+ "wall_clock_breakdown": false
207
+ },
208
+ "label_smoothing_factor": 0.0,
209
+ "optim": "adamw_torch",
210
+ "optim_args": null,
211
+ "adafactor": false,
212
+ "group_by_length": false,
213
+ "length_column_name": "length",
214
+ "report_to": [
215
+ "wandb"
216
+ ],
217
+ "ddp_find_unused_parameters": null,
218
+ "ddp_bucket_cap_mb": null,
219
+ "ddp_broadcast_buffers": null,
220
+ "dataloader_pin_memory": true,
221
+ "dataloader_persistent_workers": false,
222
+ "skip_memory_metrics": true,
223
+ "use_legacy_prediction_loop": false,
224
+ "push_to_hub": false,
225
+ "resume_from_checkpoint": null,
226
+ "hub_model_id": null,
227
+ "hub_strategy": "every_save",
228
+ "hub_private_repo": null,
229
+ "hub_always_push": false,
230
+ "gradient_checkpointing": true,
231
+ "gradient_checkpointing_kwargs": null,
232
+ "include_inputs_for_metrics": false,
233
+ "include_for_metrics": [],
234
+ "eval_do_concat_batches": true,
235
+ "fp16_backend": "auto",
236
+ "push_to_hub_model_id": null,
237
+ "push_to_hub_organization": null,
238
+ "push_to_hub_token": null,
239
+ "mp_parameters": "",
240
+ "auto_find_batch_size": false,
241
+ "full_determinism": false,
242
+ "torchdynamo": null,
243
+ "ray_scope": "last",
244
+ "torch_compile": false,
245
+ "torch_compile_backend": null,
246
+ "torch_compile_mode": null,
247
+ "include_tokens_per_second": false,
248
+ "include_num_input_tokens_seen": false,
249
+ "neftune_noise_alpha": null,
250
+ "optim_target_modules": null,
251
+ "batch_eval_metrics": false,
252
+ "eval_on_start": false,
253
+ "use_liger_kernel": false,
254
+ "eval_use_gather_object": false,
255
+ "average_tokens_across_devices": false,
256
+ "sortish_sampler": false,
257
+ "predict_with_generate": false,
258
+ "generation_max_length": null,
259
+ "generation_num_beams": null,
260
+ "generation_config": null,
261
+ "check_model": true,
262
+ "acc_strategy": "token",
263
+ "train_dataloader_shuffle": true,
264
+ "max_epochs": null,
265
+ "metric_warmup_step": 0,
266
+ "fsdp_num": 1,
267
+ "acc_steps": 1,
268
+ "eval_use_evalscope": false,
269
+ "eval_datasets": [],
270
+ "eval_limit": null,
271
+ "eval_datasets_args": null,
272
+ "eval_generation_config": null,
273
+ "freeze_parameters": [],
274
+ "freeze_parameters_regex": null,
275
+ "freeze_parameters_ratio": 0.0,
276
+ "trainable_parameters": [],
277
+ "trainable_parameters_regex": null,
278
+ "freeze_llm": false,
279
+ "freeze_vit": true,
280
+ "freeze_aligner": false,
281
+ "target_modules": [
282
+ "all-linear"
283
+ ],
284
+ "target_regex": null,
285
+ "modules_to_save": [],
286
+ "lora_rank": 8,
287
+ "lora_alpha": 32,
288
+ "lora_dropout": 0.05,
289
+ "lora_bias": "none",
290
+ "lora_dtype": null,
291
+ "lorap_lr_ratio": null,
292
+ "use_rslora": false,
293
+ "use_dora": false,
294
+ "lora_ga_batch_size": 2,
295
+ "lora_ga_iters": 2,
296
+ "lora_ga_max_length": 1024,
297
+ "lora_ga_direction": "ArB2r",
298
+ "lora_ga_scale": "stable",
299
+ "lora_ga_stable_gamma": 16,
300
+ "init_weights": true,
301
+ "fourier_n_frequency": 2000,
302
+ "fourier_scaling": 300.0,
303
+ "boft_block_size": 4,
304
+ "boft_block_num": 0,
305
+ "boft_n_butterfly_factor": 1,
306
+ "boft_dropout": 0.0,
307
+ "vera_rank": 256,
308
+ "vera_projection_prng_key": 0,
309
+ "vera_dropout": 0.0,
310
+ "vera_d_initial": 0.1,
311
+ "adapter_act": "gelu",
312
+ "adapter_length": 128,
313
+ "use_galore": false,
314
+ "galore_target_modules": null,
315
+ "galore_rank": 128,
316
+ "galore_update_proj_gap": 50,
317
+ "galore_scale": 1.0,
318
+ "galore_proj_type": "std",
319
+ "galore_optim_per_parameter": false,
320
+ "galore_with_embedding": false,
321
+ "galore_quantization": false,
322
+ "galore_proj_quant": false,
323
+ "galore_proj_bits": 4,
324
+ "galore_proj_group_size": 256,
325
+ "galore_cos_threshold": 0.4,
326
+ "galore_gamma_proj": 2,
327
+ "galore_queue_size": 5,
328
+ "adalora_target_r": 8,
329
+ "adalora_init_r": 12,
330
+ "adalora_tinit": 0,
331
+ "adalora_tfinal": 0,
332
+ "adalora_deltaT": 1,
333
+ "adalora_beta1": 0.85,
334
+ "adalora_beta2": 0.85,
335
+ "adalora_orth_reg_weight": 0.5,
336
+ "llamapro_num_new_blocks": 4,
337
+ "llamapro_num_groups": null,
338
+ "lisa_activated_layers": 0,
339
+ "lisa_step_interval": 20,
340
+ "reft_layer_key": null,
341
+ "reft_layers": null,
342
+ "reft_rank": 4,
343
+ "reft_intervention_type": "LoreftIntervention",
344
+ "reft_args": null,
345
+ "swanlab_token": null,
346
+ "swanlab_project": null,
347
+ "swanlab_workspace": null,
348
+ "swanlab_exp_name": null,
349
+ "swanlab_mode": "cloud",
350
+ "add_version": true,
351
+ "resume_only_model": false,
352
+ "create_checkpoint_symlink": false,
353
+ "packing": true,
354
+ "lazy_tokenize": false,
355
+ "loss_type": null,
356
+ "optimizer": null,
357
+ "metric": null,
358
+ "zero_hpz_partition_size": null,
359
+ "rank": 0,
360
+ "global_world_size": 8,
361
+ "local_world_size": 8,
362
+ "model_suffix": "Qwen2.5-7B",
363
+ "model_info": "ModelInfo(model_type='qwen2_5', model_dir='/data/minimax-dialogue/users/ruobai/cogito/base_model/Qwen2.5-7B', torch_dtype=torch.bfloat16, max_model_len=131072, quant_method=None, quant_bits=None, rope_scaling=None, config=None, task_type='causal_lm', num_labels=None)",
364
+ "model_meta": "ModelMeta(model_type='qwen2_5', model_groups=[ModelGroup(models=[Model(ms_model_id='Qwen/Qwen2.5-0.5B-Instruct', hf_model_id='Qwen/Qwen2.5-0.5B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-1.5B-Instruct', hf_model_id='Qwen/Qwen2.5-1.5B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-3B-Instruct', hf_model_id='Qwen/Qwen2.5-3B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-7B-Instruct', hf_model_id='Qwen/Qwen2.5-7B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-14B-Instruct', hf_model_id='Qwen/Qwen2.5-14B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-32B-Instruct', hf_model_id='Qwen/Qwen2.5-32B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-72B-Instruct', hf_model_id='Qwen/Qwen2.5-72B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-0.5B', hf_model_id='Qwen/Qwen2.5-0.5B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-1.5B', hf_model_id='Qwen/Qwen2.5-1.5B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-3B', hf_model_id='Qwen/Qwen2.5-3B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-7B', hf_model_id='Qwen/Qwen2.5-7B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-14B', hf_model_id='Qwen/Qwen2.5-14B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-32B', hf_model_id='Qwen/Qwen2.5-32B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-72B', hf_model_id='Qwen/Qwen2.5-72B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-0.5B-Instruct-GPTQ-Int4', hf_model_id='Qwen/Qwen2.5-0.5B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-1.5B-Instruct-GPTQ-Int4', hf_model_id='Qwen/Qwen2.5-1.5B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-3B-Instruct-GPTQ-Int4', hf_model_id='Qwen/Qwen2.5-3B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-7B-Instruct-GPTQ-Int4', hf_model_id='Qwen/Qwen2.5-7B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-14B-Instruct-GPTQ-Int4', hf_model_id='Qwen/Qwen2.5-14B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-32B-Instruct-GPTQ-Int4', hf_model_id='Qwen/Qwen2.5-32B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-72B-Instruct-GPTQ-Int4', hf_model_id='Qwen/Qwen2.5-72B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-0.5B-Instruct-GPTQ-Int8', hf_model_id='Qwen/Qwen2.5-0.5B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-1.5B-Instruct-GPTQ-Int8', hf_model_id='Qwen/Qwen2.5-1.5B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-3B-Instruct-GPTQ-Int8', hf_model_id='Qwen/Qwen2.5-3B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-7B-Instruct-GPTQ-Int8', hf_model_id='Qwen/Qwen2.5-7B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-14B-Instruct-GPTQ-Int8', hf_model_id='Qwen/Qwen2.5-14B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-32B-Instruct-GPTQ-Int8', hf_model_id='Qwen/Qwen2.5-32B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-72B-Instruct-GPTQ-Int8', hf_model_id='Qwen/Qwen2.5-72B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-0.5B-Instruct-AWQ', hf_model_id='Qwen/Qwen2.5-0.5B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-1.5B-Instruct-AWQ', hf_model_id='Qwen/Qwen2.5-1.5B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-3B-Instruct-AWQ', hf_model_id='Qwen/Qwen2.5-3B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-7B-Instruct-AWQ', hf_model_id='Qwen/Qwen2.5-7B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-14B-Instruct-AWQ', hf_model_id='Qwen/Qwen2.5-14B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-32B-Instruct-AWQ', hf_model_id='Qwen/Qwen2.5-32B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-72B-Instruct-AWQ', hf_model_id='Qwen/Qwen2.5-72B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None)], ignore_patterns=None, requires=None, tags=[]), ModelGroup(models=[Model(ms_model_id='Qwen/Qwen2.5-Coder-0.5B-Instruct', hf_model_id='Qwen/Qwen2.5-Coder-0.5B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-1.5B-Instruct', hf_model_id='Qwen/Qwen2.5-Coder-1.5B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-3B-Instruct', hf_model_id='Qwen/Qwen2.5-Coder-3B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-7B-Instruct', hf_model_id='Qwen/Qwen2.5-Coder-7B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-14B-Instruct', hf_model_id='Qwen/Qwen2.5-Coder-14B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-32B-Instruct', hf_model_id='Qwen/Qwen2.5-Coder-32B-Instruct', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-0.5B', hf_model_id='Qwen/Qwen2.5-Coder-0.5B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-1.5B', hf_model_id='Qwen/Qwen2.5-Coder-1.5B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-3B', hf_model_id='Qwen/Qwen2.5-Coder-3B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-7B', hf_model_id='Qwen/Qwen2.5-Coder-7B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-14B', hf_model_id='Qwen/Qwen2.5-Coder-14B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-32B', hf_model_id='Qwen/Qwen2.5-Coder-32B', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-0.5B-Instruct-AWQ', hf_model_id='Qwen/Qwen2.5-Coder-0.5B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-1.5B-Instruct-AWQ', hf_model_id='Qwen/Qwen2.5-Coder-1.5B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-3B-Instruct-AWQ', hf_model_id='Qwen/Qwen2.5-Coder-3B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-7B-Instruct-AWQ', hf_model_id='Qwen/Qwen2.5-Coder-7B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-14B-Instruct-AWQ', hf_model_id='Qwen/Qwen2.5-Coder-14B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-32B-Instruct-AWQ', hf_model_id='Qwen/Qwen2.5-Coder-32B-Instruct-AWQ', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-0.5B-Instruct-GPTQ-Int4', hf_model_id='Qwen/Qwen2.5-Coder-0.5B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-0.5B-Instruct-GPTQ-Int8', hf_model_id='Qwen/Qwen2.5-Coder-0.5B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-1.5B-Instruct-GPTQ-Int4', hf_model_id='Qwen/Qwen2.5-Coder-1.5B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-1.5B-Instruct-GPTQ-Int8', hf_model_id='Qwen/Qwen2.5-Coder-1.5B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-3B-Instruct-GPTQ-Int4', hf_model_id='Qwen/Qwen2.5-Coder-3B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-3B-Instruct-GPTQ-Int8', hf_model_id='Qwen/Qwen2.5-Coder-3B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-7B-Instruct-GPTQ-Int4', hf_model_id='Qwen/Qwen2.5-Coder-7B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-7B-Instruct-GPTQ-Int8', hf_model_id='Qwen/Qwen2.5-Coder-7B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-14B-Instruct-GPTQ-Int4', hf_model_id='Qwen/Qwen2.5-Coder-14B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-14B-Instruct-GPTQ-Int8', hf_model_id='Qwen/Qwen2.5-Coder-14B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-32B-Instruct-GPTQ-Int4', hf_model_id='Qwen/Qwen2.5-Coder-32B-Instruct-GPTQ-Int4', model_path=None, ms_revision=None, hf_revision=None), Model(ms_model_id='Qwen/Qwen2.5-Coder-32B-Instruct-GPTQ-Int8', hf_model_id='Qwen/Qwen2.5-Coder-32B-Instruct-GPTQ-Int8', model_path=None, ms_revision=None, hf_revision=None)], ignore_patterns=None, requires=None, tags=['coding'])], template='qwen2_5', get_function=<function get_model_tokenizer_with_flash_attn at 0x7f3ff78a7920>, model_arch='llama', architectures=['Qwen2ForCausalLM'], additional_saved_files=[], torch_dtype=None, is_multimodal=False, is_reward=False, task_type=None, ignore_patterns=None, requires=['transformers>=4.37'], tags=[])",
365
+ "model_dir": "/data/minimax-dialogue/users/ruobai/cogito/base_model/Qwen2.5-7B",
366
+ "hub": "<class 'swift.hub.hub.HFHub'>",
367
+ "evaluation_strategy": "epoch",
368
+ "training_args": "Seq2SeqTrainingArguments(output_dir='/data/minimax-dialogue/users/ruobai/swift/model/qwen2p5_7B_ba_1012/v0-20251012-211828', overwrite_output_dir=False, do_train=False, do_eval=True, do_predict=False, eval_strategy=<IntervalStrategy.EPOCH: 'epoch'>, prediction_loss_only=False, per_device_train_batch_size=1, per_device_eval_batch_size=1, per_gpu_train_batch_size=None, per_gpu_eval_batch_size=None, gradient_accumulation_steps=2, eval_accumulation_steps=None, eval_delay=0, torch_empty_cache_steps=None, learning_rate=1e-05, weight_decay=0.05, adam_beta1=0.9, adam_beta2=0.95, adam_epsilon=1e-08, max_grad_norm=1.0, num_train_epochs=4.0, max_steps=-1, lr_scheduler_type=<SchedulerType.COSINE: 'cosine'>, lr_scheduler_kwargs=None, warmup_ratio=0.05, warmup_steps=0, log_level='passive', log_level_replica='warning', log_on_each_node=True, logging_dir='/data/minimax-dialogue/users/ruobai/swift/model/qwen2p5_7B_ba_1012/v0-20251012-211828/runs', logging_strategy=<IntervalStrategy.STEPS: 'steps'>, logging_first_step=True, logging_steps=1, logging_nan_inf_filter=True, save_strategy=<SaveStrategy.EPOCH: 'epoch'>, save_steps=500, save_total_limit=None, save_safetensors=True, save_on_each_node=False, save_only_model=False, restore_callback_states_from_checkpoint=False, no_cuda=False, use_cpu=False, use_mps_device=False, seed=42, data_seed=42, jit_mode_eval=False, use_ipex=False, bf16=True, fp16=False, fp16_opt_level='O1', half_precision_backend='auto', bf16_full_eval=False, fp16_full_eval=False, tf32=None, local_rank=0, ddp_backend='nccl', tpu_num_cores=None, tpu_metrics_debug=False, debug=[], dataloader_drop_last=False, eval_steps=20.0, dataloader_num_workers=48, dataloader_prefetch_factor=10, past_index=-1, run_name='/data/minimax-dialogue/users/ruobai/swift/model/qwen2p5_7B_ba_1012/v0-20251012-211828', disable_tqdm=False, remove_unused_columns=False, label_names=None, load_best_model_at_end=False, metric_for_best_model='loss', greater_is_better=False, ignore_data_skip=False, fsdp=[], fsdp_min_num_params=0, fsdp_config={'min_num_params': 0, 'xla': False, 'xla_fsdp_v2': False, 'xla_fsdp_grad_ckpt': False}, tp_size=0, fsdp_transformer_layer_cls_to_wrap=None, accelerator_config=AcceleratorConfig(split_batches=False, dispatch_batches=False, even_batches=True, use_seedable_sampler=True, non_blocking=False, gradient_accumulation_kwargs=None, use_configured_state=False), deepspeed={'fp16': {'enabled': 'auto', 'loss_scale': 0, 'loss_scale_window': 1000, 'initial_scale_power': 16, 'hysteresis': 2, 'min_loss_scale': 1}, 'bf16': {'enabled': 'auto'}, 'zero_optimization': {'stage': 3, 'offload_optimizer': {'device': 'none', 'pin_memory': True}, 'offload_param': {'device': 'none', 'pin_memory': True}, 'overlap_comm': False, 'contiguous_gradients': True, 'sub_group_size': 1000000000.0, 'reduce_bucket_size': 'auto', 'zero_quantized_weights': False, 'zero_quantized_gradients': False, 'stage3_prefetch_bucket_size': 'auto', 'stage3_param_persistence_threshold': 'auto', 'stage3_max_live_parameters': 1000000000.0, 'stage3_max_reuse_distance': 1000000000.0, 'stage3_gather_16bit_weights_on_model_save': True}, 'gradient_accumulation_steps': 'auto', 'gradient_clipping': 'auto', 'steps_per_print': 2000, 'train_batch_size': 'auto', 'train_micro_batch_size_per_gpu': 'auto', 'wall_clock_breakdown': False}, label_smoothing_factor=0.0, optim=<OptimizerNames.ADAMW_TORCH: 'adamw_torch'>, optim_args=None, adafactor=False, group_by_length=False, length_column_name='length', report_to=['wandb'], ddp_find_unused_parameters=None, ddp_bucket_cap_mb=None, ddp_broadcast_buffers=None, dataloader_pin_memory=True, dataloader_persistent_workers=False, skip_memory_metrics=True, use_legacy_prediction_loop=False, push_to_hub=False, resume_from_checkpoint=None, hub_model_id=None, hub_strategy=<HubStrategy.EVERY_SAVE: 'every_save'>, hub_token=None, hub_private_repo=None, hub_always_push=False, gradient_checkpointing=True, gradient_checkpointing_kwargs=None, include_inputs_for_metrics=False, include_for_metrics=[], eval_do_concat_batches=True, fp16_backend='auto', push_to_hub_model_id=None, push_to_hub_organization=None, push_to_hub_token=None, mp_parameters='', auto_find_batch_size=False, full_determinism=False, torchdynamo=None, ray_scope='last', ddp_timeout=1800, torch_compile=False, torch_compile_backend=None, torch_compile_mode=None, include_tokens_per_second=None, include_num_input_tokens_seen=None, neftune_noise_alpha=None, optim_target_modules=None, batch_eval_metrics=False, eval_on_start=False, use_liger_kernel=False, eval_use_gather_object=False, average_tokens_across_devices=None, sortish_sampler=False, predict_with_generate=False, generation_max_length=None, generation_num_beams=None, generation_config=None, check_model=True, acc_strategy='token', train_dataloader_shuffle=True, max_epochs=None, metric_warmup_step=0, fsdp_num=1, acc_steps=1, eval_use_evalscope=False, eval_datasets=[], eval_limit=None, eval_datasets_args=None, eval_generation_config=None, train_type='full', optimizer=None, local_repo_path=None, galore_config=None)"
369
+ }
v0-20251012-211828/checkpoint-448/config.json ADDED
@@ -0,0 +1,30 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "Qwen2ForCausalLM"
4
+ ],
5
+ "attention_dropout": 0.0,
6
+ "bos_token_id": 151643,
7
+ "eos_token_id": 151643,
8
+ "hidden_act": "silu",
9
+ "hidden_size": 3584,
10
+ "initializer_range": 0.02,
11
+ "intermediate_size": 18944,
12
+ "max_position_embeddings": 131072,
13
+ "max_window_layers": 28,
14
+ "model_type": "qwen2",
15
+ "num_attention_heads": 28,
16
+ "num_hidden_layers": 28,
17
+ "num_key_value_heads": 4,
18
+ "pad_token_id": 151643,
19
+ "rms_norm_eps": 1e-06,
20
+ "rope_scaling": null,
21
+ "rope_theta": 1000000.0,
22
+ "sliding_window": 131072,
23
+ "tie_word_embeddings": false,
24
+ "torch_dtype": "bfloat16",
25
+ "transformers_version": "4.51.3",
26
+ "use_cache": false,
27
+ "use_mrope": false,
28
+ "use_sliding_window": false,
29
+ "vocab_size": 152064
30
+ }
v0-20251012-211828/checkpoint-448/generation_config.json ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token_id": 151643,
3
+ "eos_token_id": 151643,
4
+ "max_new_tokens": 2048,
5
+ "transformers_version": "4.51.3"
6
+ }
v0-20251012-211828/checkpoint-448/latest ADDED
@@ -0,0 +1 @@
 
 
1
+ global_step448
v0-20251012-211828/checkpoint-448/merges.txt ADDED
The diff for this file is too large to render. See raw diff
 
v0-20251012-211828/checkpoint-448/model-00001-of-00004.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9ae082b6d4a3335bddf2e3ad359953f59cf7ba12511dd539b78be26d70a8c6c5
3
+ size 4877660776
v0-20251012-211828/checkpoint-448/model-00002-of-00004.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2e4a54c18739b011f1f716af6aae8494c22c946c0c2578c4f2003d37fa0fb027
3
+ size 4932751008
v0-20251012-211828/checkpoint-448/model-00003-of-00004.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:628b3a445a516905e16572aba05128d1a8d7a9f2d549f373ebffcf0b66f2dc11
3
+ size 4330865200
v0-20251012-211828/checkpoint-448/model-00004-of-00004.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cadea09cd03a13198884d136be0b7fe07bcbdea6787fb749cee4c5657cec7d9c
3
+ size 1089994880
v0-20251012-211828/checkpoint-448/model.safetensors.index.json ADDED
@@ -0,0 +1,346 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "metadata": {
3
+ "total_size": 15231233024
4
+ },
5
+ "weight_map": {
6
+ "lm_head.weight": "model-00004-of-00004.safetensors",
7
+ "model.embed_tokens.weight": "model-00001-of-00004.safetensors",
8
+ "model.layers.0.input_layernorm.weight": "model-00001-of-00004.safetensors",
9
+ "model.layers.0.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
10
+ "model.layers.0.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
11
+ "model.layers.0.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
12
+ "model.layers.0.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
13
+ "model.layers.0.self_attn.k_proj.bias": "model-00001-of-00004.safetensors",
14
+ "model.layers.0.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
15
+ "model.layers.0.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
16
+ "model.layers.0.self_attn.q_proj.bias": "model-00001-of-00004.safetensors",
17
+ "model.layers.0.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
18
+ "model.layers.0.self_attn.v_proj.bias": "model-00001-of-00004.safetensors",
19
+ "model.layers.0.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
20
+ "model.layers.1.input_layernorm.weight": "model-00001-of-00004.safetensors",
21
+ "model.layers.1.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
22
+ "model.layers.1.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
23
+ "model.layers.1.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
24
+ "model.layers.1.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
25
+ "model.layers.1.self_attn.k_proj.bias": "model-00001-of-00004.safetensors",
26
+ "model.layers.1.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
27
+ "model.layers.1.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
28
+ "model.layers.1.self_attn.q_proj.bias": "model-00001-of-00004.safetensors",
29
+ "model.layers.1.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
30
+ "model.layers.1.self_attn.v_proj.bias": "model-00001-of-00004.safetensors",
31
+ "model.layers.1.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
32
+ "model.layers.10.input_layernorm.weight": "model-00002-of-00004.safetensors",
33
+ "model.layers.10.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
34
+ "model.layers.10.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
35
+ "model.layers.10.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
36
+ "model.layers.10.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
37
+ "model.layers.10.self_attn.k_proj.bias": "model-00002-of-00004.safetensors",
38
+ "model.layers.10.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
39
+ "model.layers.10.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
40
+ "model.layers.10.self_attn.q_proj.bias": "model-00002-of-00004.safetensors",
41
+ "model.layers.10.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
42
+ "model.layers.10.self_attn.v_proj.bias": "model-00002-of-00004.safetensors",
43
+ "model.layers.10.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
44
+ "model.layers.11.input_layernorm.weight": "model-00002-of-00004.safetensors",
45
+ "model.layers.11.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
46
+ "model.layers.11.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
47
+ "model.layers.11.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
48
+ "model.layers.11.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
49
+ "model.layers.11.self_attn.k_proj.bias": "model-00002-of-00004.safetensors",
50
+ "model.layers.11.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
51
+ "model.layers.11.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
52
+ "model.layers.11.self_attn.q_proj.bias": "model-00002-of-00004.safetensors",
53
+ "model.layers.11.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
54
+ "model.layers.11.self_attn.v_proj.bias": "model-00002-of-00004.safetensors",
55
+ "model.layers.11.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
56
+ "model.layers.12.input_layernorm.weight": "model-00002-of-00004.safetensors",
57
+ "model.layers.12.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
58
+ "model.layers.12.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
59
+ "model.layers.12.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
60
+ "model.layers.12.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
61
+ "model.layers.12.self_attn.k_proj.bias": "model-00002-of-00004.safetensors",
62
+ "model.layers.12.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
63
+ "model.layers.12.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
64
+ "model.layers.12.self_attn.q_proj.bias": "model-00002-of-00004.safetensors",
65
+ "model.layers.12.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
66
+ "model.layers.12.self_attn.v_proj.bias": "model-00002-of-00004.safetensors",
67
+ "model.layers.12.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
68
+ "model.layers.13.input_layernorm.weight": "model-00002-of-00004.safetensors",
69
+ "model.layers.13.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
70
+ "model.layers.13.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
71
+ "model.layers.13.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
72
+ "model.layers.13.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
73
+ "model.layers.13.self_attn.k_proj.bias": "model-00002-of-00004.safetensors",
74
+ "model.layers.13.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
75
+ "model.layers.13.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
76
+ "model.layers.13.self_attn.q_proj.bias": "model-00002-of-00004.safetensors",
77
+ "model.layers.13.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
78
+ "model.layers.13.self_attn.v_proj.bias": "model-00002-of-00004.safetensors",
79
+ "model.layers.13.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
80
+ "model.layers.14.input_layernorm.weight": "model-00002-of-00004.safetensors",
81
+ "model.layers.14.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
82
+ "model.layers.14.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
83
+ "model.layers.14.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
84
+ "model.layers.14.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
85
+ "model.layers.14.self_attn.k_proj.bias": "model-00002-of-00004.safetensors",
86
+ "model.layers.14.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
87
+ "model.layers.14.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
88
+ "model.layers.14.self_attn.q_proj.bias": "model-00002-of-00004.safetensors",
89
+ "model.layers.14.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
90
+ "model.layers.14.self_attn.v_proj.bias": "model-00002-of-00004.safetensors",
91
+ "model.layers.14.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
92
+ "model.layers.15.input_layernorm.weight": "model-00002-of-00004.safetensors",
93
+ "model.layers.15.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
94
+ "model.layers.15.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
95
+ "model.layers.15.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
96
+ "model.layers.15.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
97
+ "model.layers.15.self_attn.k_proj.bias": "model-00002-of-00004.safetensors",
98
+ "model.layers.15.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
99
+ "model.layers.15.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
100
+ "model.layers.15.self_attn.q_proj.bias": "model-00002-of-00004.safetensors",
101
+ "model.layers.15.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
102
+ "model.layers.15.self_attn.v_proj.bias": "model-00002-of-00004.safetensors",
103
+ "model.layers.15.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
104
+ "model.layers.16.input_layernorm.weight": "model-00002-of-00004.safetensors",
105
+ "model.layers.16.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
106
+ "model.layers.16.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
107
+ "model.layers.16.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
108
+ "model.layers.16.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
109
+ "model.layers.16.self_attn.k_proj.bias": "model-00002-of-00004.safetensors",
110
+ "model.layers.16.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
111
+ "model.layers.16.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
112
+ "model.layers.16.self_attn.q_proj.bias": "model-00002-of-00004.safetensors",
113
+ "model.layers.16.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
114
+ "model.layers.16.self_attn.v_proj.bias": "model-00002-of-00004.safetensors",
115
+ "model.layers.16.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
116
+ "model.layers.17.input_layernorm.weight": "model-00002-of-00004.safetensors",
117
+ "model.layers.17.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
118
+ "model.layers.17.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
119
+ "model.layers.17.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
120
+ "model.layers.17.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
121
+ "model.layers.17.self_attn.k_proj.bias": "model-00002-of-00004.safetensors",
122
+ "model.layers.17.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
123
+ "model.layers.17.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
124
+ "model.layers.17.self_attn.q_proj.bias": "model-00002-of-00004.safetensors",
125
+ "model.layers.17.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
126
+ "model.layers.17.self_attn.v_proj.bias": "model-00002-of-00004.safetensors",
127
+ "model.layers.17.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
128
+ "model.layers.18.input_layernorm.weight": "model-00003-of-00004.safetensors",
129
+ "model.layers.18.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
130
+ "model.layers.18.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
131
+ "model.layers.18.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
132
+ "model.layers.18.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
133
+ "model.layers.18.self_attn.k_proj.bias": "model-00002-of-00004.safetensors",
134
+ "model.layers.18.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
135
+ "model.layers.18.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
136
+ "model.layers.18.self_attn.q_proj.bias": "model-00002-of-00004.safetensors",
137
+ "model.layers.18.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
138
+ "model.layers.18.self_attn.v_proj.bias": "model-00002-of-00004.safetensors",
139
+ "model.layers.18.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
140
+ "model.layers.19.input_layernorm.weight": "model-00003-of-00004.safetensors",
141
+ "model.layers.19.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
142
+ "model.layers.19.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
143
+ "model.layers.19.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
144
+ "model.layers.19.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
145
+ "model.layers.19.self_attn.k_proj.bias": "model-00003-of-00004.safetensors",
146
+ "model.layers.19.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
147
+ "model.layers.19.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
148
+ "model.layers.19.self_attn.q_proj.bias": "model-00003-of-00004.safetensors",
149
+ "model.layers.19.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
150
+ "model.layers.19.self_attn.v_proj.bias": "model-00003-of-00004.safetensors",
151
+ "model.layers.19.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
152
+ "model.layers.2.input_layernorm.weight": "model-00001-of-00004.safetensors",
153
+ "model.layers.2.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
154
+ "model.layers.2.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
155
+ "model.layers.2.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
156
+ "model.layers.2.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
157
+ "model.layers.2.self_attn.k_proj.bias": "model-00001-of-00004.safetensors",
158
+ "model.layers.2.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
159
+ "model.layers.2.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
160
+ "model.layers.2.self_attn.q_proj.bias": "model-00001-of-00004.safetensors",
161
+ "model.layers.2.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
162
+ "model.layers.2.self_attn.v_proj.bias": "model-00001-of-00004.safetensors",
163
+ "model.layers.2.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
164
+ "model.layers.20.input_layernorm.weight": "model-00003-of-00004.safetensors",
165
+ "model.layers.20.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
166
+ "model.layers.20.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
167
+ "model.layers.20.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
168
+ "model.layers.20.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
169
+ "model.layers.20.self_attn.k_proj.bias": "model-00003-of-00004.safetensors",
170
+ "model.layers.20.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
171
+ "model.layers.20.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
172
+ "model.layers.20.self_attn.q_proj.bias": "model-00003-of-00004.safetensors",
173
+ "model.layers.20.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
174
+ "model.layers.20.self_attn.v_proj.bias": "model-00003-of-00004.safetensors",
175
+ "model.layers.20.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
176
+ "model.layers.21.input_layernorm.weight": "model-00003-of-00004.safetensors",
177
+ "model.layers.21.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
178
+ "model.layers.21.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
179
+ "model.layers.21.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
180
+ "model.layers.21.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
181
+ "model.layers.21.self_attn.k_proj.bias": "model-00003-of-00004.safetensors",
182
+ "model.layers.21.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
183
+ "model.layers.21.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
184
+ "model.layers.21.self_attn.q_proj.bias": "model-00003-of-00004.safetensors",
185
+ "model.layers.21.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
186
+ "model.layers.21.self_attn.v_proj.bias": "model-00003-of-00004.safetensors",
187
+ "model.layers.21.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
188
+ "model.layers.22.input_layernorm.weight": "model-00003-of-00004.safetensors",
189
+ "model.layers.22.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
190
+ "model.layers.22.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
191
+ "model.layers.22.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
192
+ "model.layers.22.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
193
+ "model.layers.22.self_attn.k_proj.bias": "model-00003-of-00004.safetensors",
194
+ "model.layers.22.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
195
+ "model.layers.22.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
196
+ "model.layers.22.self_attn.q_proj.bias": "model-00003-of-00004.safetensors",
197
+ "model.layers.22.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
198
+ "model.layers.22.self_attn.v_proj.bias": "model-00003-of-00004.safetensors",
199
+ "model.layers.22.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
200
+ "model.layers.23.input_layernorm.weight": "model-00003-of-00004.safetensors",
201
+ "model.layers.23.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
202
+ "model.layers.23.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
203
+ "model.layers.23.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
204
+ "model.layers.23.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
205
+ "model.layers.23.self_attn.k_proj.bias": "model-00003-of-00004.safetensors",
206
+ "model.layers.23.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
207
+ "model.layers.23.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
208
+ "model.layers.23.self_attn.q_proj.bias": "model-00003-of-00004.safetensors",
209
+ "model.layers.23.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
210
+ "model.layers.23.self_attn.v_proj.bias": "model-00003-of-00004.safetensors",
211
+ "model.layers.23.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
212
+ "model.layers.24.input_layernorm.weight": "model-00003-of-00004.safetensors",
213
+ "model.layers.24.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
214
+ "model.layers.24.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
215
+ "model.layers.24.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
216
+ "model.layers.24.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
217
+ "model.layers.24.self_attn.k_proj.bias": "model-00003-of-00004.safetensors",
218
+ "model.layers.24.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
219
+ "model.layers.24.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
220
+ "model.layers.24.self_attn.q_proj.bias": "model-00003-of-00004.safetensors",
221
+ "model.layers.24.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
222
+ "model.layers.24.self_attn.v_proj.bias": "model-00003-of-00004.safetensors",
223
+ "model.layers.24.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
224
+ "model.layers.25.input_layernorm.weight": "model-00003-of-00004.safetensors",
225
+ "model.layers.25.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
226
+ "model.layers.25.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
227
+ "model.layers.25.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
228
+ "model.layers.25.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
229
+ "model.layers.25.self_attn.k_proj.bias": "model-00003-of-00004.safetensors",
230
+ "model.layers.25.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
231
+ "model.layers.25.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
232
+ "model.layers.25.self_attn.q_proj.bias": "model-00003-of-00004.safetensors",
233
+ "model.layers.25.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
234
+ "model.layers.25.self_attn.v_proj.bias": "model-00003-of-00004.safetensors",
235
+ "model.layers.25.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
236
+ "model.layers.26.input_layernorm.weight": "model-00003-of-00004.safetensors",
237
+ "model.layers.26.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
238
+ "model.layers.26.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
239
+ "model.layers.26.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
240
+ "model.layers.26.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
241
+ "model.layers.26.self_attn.k_proj.bias": "model-00003-of-00004.safetensors",
242
+ "model.layers.26.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
243
+ "model.layers.26.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
244
+ "model.layers.26.self_attn.q_proj.bias": "model-00003-of-00004.safetensors",
245
+ "model.layers.26.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
246
+ "model.layers.26.self_attn.v_proj.bias": "model-00003-of-00004.safetensors",
247
+ "model.layers.26.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
248
+ "model.layers.27.input_layernorm.weight": "model-00003-of-00004.safetensors",
249
+ "model.layers.27.mlp.down_proj.weight": "model-00003-of-00004.safetensors",
250
+ "model.layers.27.mlp.gate_proj.weight": "model-00003-of-00004.safetensors",
251
+ "model.layers.27.mlp.up_proj.weight": "model-00003-of-00004.safetensors",
252
+ "model.layers.27.post_attention_layernorm.weight": "model-00003-of-00004.safetensors",
253
+ "model.layers.27.self_attn.k_proj.bias": "model-00003-of-00004.safetensors",
254
+ "model.layers.27.self_attn.k_proj.weight": "model-00003-of-00004.safetensors",
255
+ "model.layers.27.self_attn.o_proj.weight": "model-00003-of-00004.safetensors",
256
+ "model.layers.27.self_attn.q_proj.bias": "model-00003-of-00004.safetensors",
257
+ "model.layers.27.self_attn.q_proj.weight": "model-00003-of-00004.safetensors",
258
+ "model.layers.27.self_attn.v_proj.bias": "model-00003-of-00004.safetensors",
259
+ "model.layers.27.self_attn.v_proj.weight": "model-00003-of-00004.safetensors",
260
+ "model.layers.3.input_layernorm.weight": "model-00001-of-00004.safetensors",
261
+ "model.layers.3.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
262
+ "model.layers.3.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
263
+ "model.layers.3.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
264
+ "model.layers.3.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
265
+ "model.layers.3.self_attn.k_proj.bias": "model-00001-of-00004.safetensors",
266
+ "model.layers.3.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
267
+ "model.layers.3.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
268
+ "model.layers.3.self_attn.q_proj.bias": "model-00001-of-00004.safetensors",
269
+ "model.layers.3.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
270
+ "model.layers.3.self_attn.v_proj.bias": "model-00001-of-00004.safetensors",
271
+ "model.layers.3.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
272
+ "model.layers.4.input_layernorm.weight": "model-00001-of-00004.safetensors",
273
+ "model.layers.4.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
274
+ "model.layers.4.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
275
+ "model.layers.4.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
276
+ "model.layers.4.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
277
+ "model.layers.4.self_attn.k_proj.bias": "model-00001-of-00004.safetensors",
278
+ "model.layers.4.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
279
+ "model.layers.4.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
280
+ "model.layers.4.self_attn.q_proj.bias": "model-00001-of-00004.safetensors",
281
+ "model.layers.4.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
282
+ "model.layers.4.self_attn.v_proj.bias": "model-00001-of-00004.safetensors",
283
+ "model.layers.4.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
284
+ "model.layers.5.input_layernorm.weight": "model-00001-of-00004.safetensors",
285
+ "model.layers.5.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
286
+ "model.layers.5.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
287
+ "model.layers.5.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
288
+ "model.layers.5.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
289
+ "model.layers.5.self_attn.k_proj.bias": "model-00001-of-00004.safetensors",
290
+ "model.layers.5.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
291
+ "model.layers.5.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
292
+ "model.layers.5.self_attn.q_proj.bias": "model-00001-of-00004.safetensors",
293
+ "model.layers.5.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
294
+ "model.layers.5.self_attn.v_proj.bias": "model-00001-of-00004.safetensors",
295
+ "model.layers.5.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
296
+ "model.layers.6.input_layernorm.weight": "model-00001-of-00004.safetensors",
297
+ "model.layers.6.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
298
+ "model.layers.6.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
299
+ "model.layers.6.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
300
+ "model.layers.6.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
301
+ "model.layers.6.self_attn.k_proj.bias": "model-00001-of-00004.safetensors",
302
+ "model.layers.6.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
303
+ "model.layers.6.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
304
+ "model.layers.6.self_attn.q_proj.bias": "model-00001-of-00004.safetensors",
305
+ "model.layers.6.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
306
+ "model.layers.6.self_attn.v_proj.bias": "model-00001-of-00004.safetensors",
307
+ "model.layers.6.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
308
+ "model.layers.7.input_layernorm.weight": "model-00001-of-00004.safetensors",
309
+ "model.layers.7.mlp.down_proj.weight": "model-00001-of-00004.safetensors",
310
+ "model.layers.7.mlp.gate_proj.weight": "model-00001-of-00004.safetensors",
311
+ "model.layers.7.mlp.up_proj.weight": "model-00001-of-00004.safetensors",
312
+ "model.layers.7.post_attention_layernorm.weight": "model-00001-of-00004.safetensors",
313
+ "model.layers.7.self_attn.k_proj.bias": "model-00001-of-00004.safetensors",
314
+ "model.layers.7.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
315
+ "model.layers.7.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
316
+ "model.layers.7.self_attn.q_proj.bias": "model-00001-of-00004.safetensors",
317
+ "model.layers.7.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
318
+ "model.layers.7.self_attn.v_proj.bias": "model-00001-of-00004.safetensors",
319
+ "model.layers.7.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
320
+ "model.layers.8.input_layernorm.weight": "model-00002-of-00004.safetensors",
321
+ "model.layers.8.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
322
+ "model.layers.8.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
323
+ "model.layers.8.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
324
+ "model.layers.8.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
325
+ "model.layers.8.self_attn.k_proj.bias": "model-00001-of-00004.safetensors",
326
+ "model.layers.8.self_attn.k_proj.weight": "model-00001-of-00004.safetensors",
327
+ "model.layers.8.self_attn.o_proj.weight": "model-00001-of-00004.safetensors",
328
+ "model.layers.8.self_attn.q_proj.bias": "model-00001-of-00004.safetensors",
329
+ "model.layers.8.self_attn.q_proj.weight": "model-00001-of-00004.safetensors",
330
+ "model.layers.8.self_attn.v_proj.bias": "model-00001-of-00004.safetensors",
331
+ "model.layers.8.self_attn.v_proj.weight": "model-00001-of-00004.safetensors",
332
+ "model.layers.9.input_layernorm.weight": "model-00002-of-00004.safetensors",
333
+ "model.layers.9.mlp.down_proj.weight": "model-00002-of-00004.safetensors",
334
+ "model.layers.9.mlp.gate_proj.weight": "model-00002-of-00004.safetensors",
335
+ "model.layers.9.mlp.up_proj.weight": "model-00002-of-00004.safetensors",
336
+ "model.layers.9.post_attention_layernorm.weight": "model-00002-of-00004.safetensors",
337
+ "model.layers.9.self_attn.k_proj.bias": "model-00002-of-00004.safetensors",
338
+ "model.layers.9.self_attn.k_proj.weight": "model-00002-of-00004.safetensors",
339
+ "model.layers.9.self_attn.o_proj.weight": "model-00002-of-00004.safetensors",
340
+ "model.layers.9.self_attn.q_proj.bias": "model-00002-of-00004.safetensors",
341
+ "model.layers.9.self_attn.q_proj.weight": "model-00002-of-00004.safetensors",
342
+ "model.layers.9.self_attn.v_proj.bias": "model-00002-of-00004.safetensors",
343
+ "model.layers.9.self_attn.v_proj.weight": "model-00002-of-00004.safetensors",
344
+ "model.norm.weight": "model-00003-of-00004.safetensors"
345
+ }
346
+ }
v0-20251012-211828/checkpoint-448/rng_state_0.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5ff22c72cb50d4b6353cf336b950d2c54115c739606544de9a8d3b0fab0ef188
3
+ size 15920
v0-20251012-211828/checkpoint-448/rng_state_1.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c253ad05d3ffa98dc1e5291ec640e6158218602254206e6fb97ea82185040775
3
+ size 15920
v0-20251012-211828/checkpoint-448/rng_state_2.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:edcdbb53b869133655de3ceb14325bb161e7aa238640244125393ff4afcb7363
3
+ size 15920
v0-20251012-211828/checkpoint-448/rng_state_3.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b095e9807a468e05241a1d389fd326265bee89778655d3c00298c2f6abcee791
3
+ size 15920
v0-20251012-211828/checkpoint-448/rng_state_4.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a4da3805c54028ef89f9cce32b640836c54b432b251fae2dea3d0182fa96f403
3
+ size 15920
v0-20251012-211828/checkpoint-448/rng_state_5.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:08bb3c8d2c1053a6b93a1342acb589ea4695b3fdb4dfeaf3675191c7e3390d3d
3
+ size 15920
v0-20251012-211828/checkpoint-448/rng_state_6.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3cfebda0f5a673869a40a8a82d7e308c843f9b55ec287861a6a0670b4c43c9d4
3
+ size 15920
v0-20251012-211828/checkpoint-448/rng_state_7.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:34bc6aae481e0bf889f93cc019a7692fbdaba7925f341ec202dd711aa1d2a724
3
+ size 15920
v0-20251012-211828/checkpoint-448/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0c199a071e8f3f4409364ebb7ec6e95774df918d3f419a206c18e162e774e54b
3
+ size 1064
v0-20251012-211828/checkpoint-448/special_tokens_map.json ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "additional_special_tokens": [
3
+ "<|im_start|>",
4
+ "<|im_end|>",
5
+ "<|object_ref_start|>",
6
+ "<|object_ref_end|>",
7
+ "<|box_start|>",
8
+ "<|box_end|>",
9
+ "<|quad_start|>",
10
+ "<|quad_end|>",
11
+ "<|vision_start|>",
12
+ "<|vision_end|>",
13
+ "<|vision_pad|>",
14
+ "<|image_pad|>",
15
+ "<|video_pad|>"
16
+ ],
17
+ "eos_token": {
18
+ "content": "<|endoftext|>",
19
+ "lstrip": false,
20
+ "normalized": false,
21
+ "rstrip": false,
22
+ "single_word": false
23
+ },
24
+ "pad_token": {
25
+ "content": "<|endoftext|>",
26
+ "lstrip": false,
27
+ "normalized": false,
28
+ "rstrip": false,
29
+ "single_word": false
30
+ }
31
+ }