DavidNguyen commited on
Commit
d1e32cd
·
verified ·
1 Parent(s): e6420e3

Delete sft/1M3

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. sft/1M3/Full_remoe/checkpoint-13786/added_tokens.json +0 -13
  2. sft/1M3/Full_remoe/checkpoint-13786/config.json +0 -203
  3. sft/1M3/Full_remoe/checkpoint-13786/generation_config.json +0 -12
  4. sft/1M3/Full_remoe/checkpoint-13786/latest +0 -1
  5. sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1412_llava...image_llava_model_args_595126/gpt_response/hallusion_output_vd_model.json +0 -0
  6. sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1412_llava...image_llava_model_args_595126/gpt_response/hallusion_output_vs_model.json +0 -0
  7. sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1412_llava...image_llava_model_args_595126/hallusion_bench_image.json +0 -0
  8. sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1412_llava...image_llava_model_args_595126/mathvista_testmini.json +0 -0
  9. sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1412_llava...image_llava_model_args_595126/rank0_metric_eval_done.txt +0 -1
  10. sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1412_llava...image_llava_model_args_595126/results.json +0 -146
  11. sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1412_llava...image_llava_model_args_595126/submissions/mathvista_testmini_scores.json +0 -0
  12. sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/ai2d.json +0 -0
  13. sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/infovqa_val.json +0 -3
  14. sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/mmbench_en_dev.json +0 -3
  15. sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/mme.json +0 -0
  16. sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/mmerealworld_lite.json +0 -3
  17. sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/mmmu_val.json +0 -0
  18. sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/mmstar.json +0 -0
  19. sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/rank0_metric_eval_done.txt +0 -1
  20. sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/rank1_metric_eval_done.txt +0 -1
  21. sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/rank2_metric_eval_done.txt +0 -1
  22. sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/results.json +0 -683
  23. sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/scienceqa_img.json +0 -0
  24. sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/seedbench_2_plus.json +0 -0
  25. sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/submissions/mmbench_en_dev_results.json +0 -1
  26. sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/submissions/mmbench_en_dev_results.xlsx +0 -3
  27. sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/submissions/textvqa_submission_2025-06-27-13-58-34.json +0 -0
  28. sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/textvqa_val.json +0 -3
  29. sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1459_llava...bench_llava_model_args_595126/ocrbench.json +0 -0
  30. sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1459_llava...bench_llava_model_args_595126/rank0_metric_eval_done.txt +0 -1
  31. sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1459_llava...bench_llava_model_args_595126/rank1_metric_eval_done.txt +0 -1
  32. sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1459_llava...bench_llava_model_args_595126/rank2_metric_eval_done.txt +0 -1
  33. sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1459_llava...bench_llava_model_args_595126/results.json +0 -67
  34. sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1459_llava...bench_llava_model_args_595126/results/ocrbench_results.txt +0 -18
  35. sft/1M3/Full_remoe/checkpoint-13786/model-00001-of-00003.safetensors +0 -3
  36. sft/1M3/Full_remoe/checkpoint-13786/model-00002-of-00003.safetensors +0 -3
  37. sft/1M3/Full_remoe/checkpoint-13786/model-00003-of-00003.safetensors +0 -3
  38. sft/1M3/Full_remoe/checkpoint-13786/model.safetensors.index.json +0 -0
  39. sft/1M3/Full_remoe/checkpoint-13786/rng_state_0.pth +0 -3
  40. sft/1M3/Full_remoe/checkpoint-13786/rng_state_1.pth +0 -3
  41. sft/1M3/Full_remoe/checkpoint-13786/rng_state_2.pth +0 -3
  42. sft/1M3/Full_remoe/checkpoint-13786/rng_state_3.pth +0 -3
  43. sft/1M3/Full_remoe/checkpoint-13786/special_tokens_map.json +0 -24
  44. sft/1M3/Full_remoe/checkpoint-13786/tokenizer.model +0 -3
  45. sft/1M3/Full_remoe/checkpoint-13786/tokenizer_config.json +0 -132
  46. sft/1M3/Full_remoe/checkpoint-13786/trainer_state.json +0 -3
  47. sft/1M3/Full_remoe/checkpoint-13786/training_args.bin +0 -3
  48. sft/1M3/Full_remoe/checkpoint-13786/zero_to_fp32.py +0 -604
  49. sft/1M3/Full_remoe/checkpoint-20679/added_tokens.json +0 -13
  50. sft/1M3/Full_remoe/checkpoint-20679/config.json +0 -203
sft/1M3/Full_remoe/checkpoint-13786/added_tokens.json DELETED
@@ -1,13 +0,0 @@
1
- {
2
- "<|assistant|>": 32001,
3
- "<|endoftext|>": 32000,
4
- "<|end|>": 32007,
5
- "<|placeholder1|>": 32002,
6
- "<|placeholder2|>": 32003,
7
- "<|placeholder3|>": 32004,
8
- "<|placeholder4|>": 32005,
9
- "<|placeholder5|>": 32008,
10
- "<|placeholder6|>": 32009,
11
- "<|system|>": 32006,
12
- "<|user|>": 32010
13
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
sft/1M3/Full_remoe/checkpoint-13786/config.json DELETED
@@ -1,203 +0,0 @@
1
- {
2
- "_name_or_path": "/cm/archive/namnv78/checkpoints/phi35-siglip224/pft",
3
- "architectures": [
4
- "LlavaPhiForCausalLM"
5
- ],
6
- "attention_bias": false,
7
- "attention_dropout": 0.0,
8
- "auto_map": {
9
- "AutoConfig": "configuration_phi3.Phi3Config",
10
- "AutoModelForCausalLM": "modeling_phi3.Phi3ForCausalLM"
11
- },
12
- "bal_comp_loss_coef": 0.01,
13
- "balance_loss_coef": 1.5e-323,
14
- "bos_token_id": 1,
15
- "clip_smoe": true,
16
- "diversity_loss_coef": 0.01,
17
- "dropout": false,
18
- "e_loss_coef": 0.001,
19
- "embd_pdrop": 0.0,
20
- "entropy_advance_loss": false,
21
- "eos_token_id": 32000,
22
- "freeze_backbone": false,
23
- "freeze_mm_mlp_adapter": false,
24
- "hidden_act": "silu",
25
- "hidden_size": 3072,
26
- "hybrid": false,
27
- "image_aspect_ratio": "pad",
28
- "init_weight": true,
29
- "initializer_range": 0.02,
30
- "intermediate_size": 8192,
31
- "is_cosine": false,
32
- "is_norm_weight": false,
33
- "local_rank": 0,
34
- "loss1": "balanceloss",
35
- "loss2": "zloss",
36
- "luna": false,
37
- "max_compete_in_iter": 3,
38
- "max_position_embeddings": 131072,
39
- "mlp_smoe": true,
40
- "mm_hidden_size": 1152,
41
- "mm_patch_merge_type": "flat",
42
- "mm_projector_lr": null,
43
- "mm_projector_type": "moe",
44
- "mm_use_im_patch_token": false,
45
- "mm_use_im_start_end": false,
46
- "mm_vision_select_feature": "patch",
47
- "mm_vision_select_layer": -2,
48
- "mm_vision_tower": "google/siglip-so400m-patch14-224",
49
- "model_name_or_path": "/cm/archive/namnv78/checkpoints/phi35-siglip224/pft",
50
- "model_type": "llava_phi",
51
- "moe_name": "remoe",
52
- "moe_relu_l1_reg_coeff_multiplier": 1.2,
53
- "mp_pixel_shuffle_factor": 1,
54
- "norm_softmax": false,
55
- "normalization": true,
56
- "num_attention_heads": 32,
57
- "num_experts": 4,
58
- "num_hidden_layers": 32,
59
- "num_key_value_heads": 32,
60
- "num_layers": 3,
61
- "num_selected": 2,
62
- "number_of_previous_tokens": 2,
63
- "original_max_position_embeddings": 4096,
64
- "pad_token_id": 32000,
65
- "pretrain_mm_mlp_adapter": "/cm/archive/namnv78/checkpoints/phi35-siglip224/pft/mm_projector.bin",
66
- "rate_compete": 0.2,
67
- "rate_flip": 0.05,
68
- "resid_pdrop": 0.0,
69
- "rms_norm_eps": 1e-05,
70
- "rope_scaling": {
71
- "long_factor": [
72
- 1.0800000429153442,
73
- 1.1100000143051147,
74
- 1.1399999856948853,
75
- 1.340000033378601,
76
- 1.5899999141693115,
77
- 1.600000023841858,
78
- 1.6200000047683716,
79
- 2.620000123977661,
80
- 3.2300000190734863,
81
- 3.2300000190734863,
82
- 4.789999961853027,
83
- 7.400000095367432,
84
- 7.700000286102295,
85
- 9.09000015258789,
86
- 12.199999809265137,
87
- 17.670000076293945,
88
- 24.46000099182129,
89
- 28.57000160217285,
90
- 30.420001983642578,
91
- 30.840002059936523,
92
- 32.590003967285156,
93
- 32.93000411987305,
94
- 42.320003509521484,
95
- 44.96000289916992,
96
- 50.340003967285156,
97
- 50.45000457763672,
98
- 57.55000305175781,
99
- 57.93000411987305,
100
- 58.21000289916992,
101
- 60.1400032043457,
102
- 62.61000442504883,
103
- 62.62000274658203,
104
- 62.71000289916992,
105
- 63.1400032043457,
106
- 63.1400032043457,
107
- 63.77000427246094,
108
- 63.93000411987305,
109
- 63.96000289916992,
110
- 63.970001220703125,
111
- 64.02999877929688,
112
- 64.06999969482422,
113
- 64.08000183105469,
114
- 64.12000274658203,
115
- 64.41000366210938,
116
- 64.4800033569336,
117
- 64.51000213623047,
118
- 64.52999877929688,
119
- 64.83999633789062
120
- ],
121
- "short_factor": [
122
- 1.0,
123
- 1.0199999809265137,
124
- 1.0299999713897705,
125
- 1.0299999713897705,
126
- 1.0499999523162842,
127
- 1.0499999523162842,
128
- 1.0499999523162842,
129
- 1.0499999523162842,
130
- 1.0499999523162842,
131
- 1.0699999332427979,
132
- 1.0999999046325684,
133
- 1.1099998950958252,
134
- 1.1599998474121094,
135
- 1.1599998474121094,
136
- 1.1699998378753662,
137
- 1.2899998426437378,
138
- 1.339999794960022,
139
- 1.679999828338623,
140
- 1.7899998426437378,
141
- 1.8199998140335083,
142
- 1.8499997854232788,
143
- 1.8799997568130493,
144
- 1.9099997282028198,
145
- 1.9399996995925903,
146
- 1.9899996519088745,
147
- 2.0199997425079346,
148
- 2.0199997425079346,
149
- 2.0199997425079346,
150
- 2.0199997425079346,
151
- 2.0199997425079346,
152
- 2.0199997425079346,
153
- 2.0299997329711914,
154
- 2.0299997329711914,
155
- 2.0299997329711914,
156
- 2.0299997329711914,
157
- 2.0299997329711914,
158
- 2.0299997329711914,
159
- 2.0299997329711914,
160
- 2.0299997329711914,
161
- 2.0299997329711914,
162
- 2.0799996852874756,
163
- 2.0899996757507324,
164
- 2.189999580383301,
165
- 2.2199995517730713,
166
- 2.5899994373321533,
167
- 2.729999542236328,
168
- 2.749999523162842,
169
- 2.8399994373321533
170
- ],
171
- "type": "longrope"
172
- },
173
- "rope_theta": 10000.0,
174
- "router_loss_coef": 0.01,
175
- "router_theta": 0.1,
176
- "router_z_loss_coef": 0.001,
177
- "scales": [
178
- 1,
179
- 3
180
- ],
181
- "sliding_window": 262144,
182
- "sparse_upcycling": true,
183
- "std_gate": 0.02,
184
- "strategy_train": "base",
185
- "tie_word_embeddings": false,
186
- "tokenizer_model_max_length": 2048,
187
- "tokenizer_padding_side": "right",
188
- "topk_max": 2,
189
- "topk_min": 1,
190
- "torch_dtype": "bfloat16",
191
- "training": true,
192
- "transformers_version": "4.43.0",
193
- "tune_mm_mlp_adapter": false,
194
- "unit_test": true,
195
- "use_cache": false,
196
- "use_mm_proj": true,
197
- "use_old": false,
198
- "version": "phi35",
199
- "vision_tower": "google/siglip-so400m-patch14-224",
200
- "vision_tower_dir": "/cm/archive/namnv78/checkpoints/phi35-siglip224/pft/clip.bin",
201
- "vocab_size": 32064,
202
- "warm_up": 0.05
203
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
sft/1M3/Full_remoe/checkpoint-13786/generation_config.json DELETED
@@ -1,12 +0,0 @@
1
- {
2
- "_from_model_config": true,
3
- "bos_token_id": 1,
4
- "do_sample": true,
5
- "eos_token_id": [
6
- 32007,
7
- 32001,
8
- 32000
9
- ],
10
- "pad_token_id": 32000,
11
- "transformers_version": "4.43.0"
12
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
sft/1M3/Full_remoe/checkpoint-13786/latest DELETED
@@ -1 +0,0 @@
1
- global_step13786
 
 
sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1412_llava...image_llava_model_args_595126/gpt_response/hallusion_output_vd_model.json DELETED
The diff for this file is too large to render. See raw diff
 
sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1412_llava...image_llava_model_args_595126/gpt_response/hallusion_output_vs_model.json DELETED
The diff for this file is too large to render. See raw diff
 
sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1412_llava...image_llava_model_args_595126/hallusion_bench_image.json DELETED
The diff for this file is too large to render. See raw diff
 
sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1412_llava...image_llava_model_args_595126/mathvista_testmini.json DELETED
The diff for this file is too large to render. See raw diff
 
sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1412_llava...image_llava_model_args_595126/rank0_metric_eval_done.txt DELETED
@@ -1 +0,0 @@
1
- rank 0 eval done
 
 
sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1412_llava...image_llava_model_args_595126/results.json DELETED
@@ -1,146 +0,0 @@
1
- {
2
- "results": {
3
- "hallusion_bench_image": {
4
- "aAcc,none": 44.164,
5
- "aAcc_stderr,none": "N/A",
6
- "fAcc,none": 14.4509,
7
- "fAcc_stderr,none": "N/A",
8
- "qAcc,none": 10.5495,
9
- "qAcc_stderr,none": "N/A",
10
- "alias": "hallusion_bench_image"
11
- },
12
- "mathvista_testmini": {
13
- "gpt_eval_score,none": 32.3,
14
- "gpt_eval_score_stderr,none": "N/A",
15
- "alias": "mathvista_testmini"
16
- }
17
- },
18
- "configs": {
19
- "hallusion_bench_image": {
20
- "task": "hallusion_bench_image",
21
- "dataset_path": "lmms-lab/HallusionBench",
22
- "dataset_kwargs": {
23
- "token": true
24
- },
25
- "test_split": "image",
26
- "doc_to_visual": "<function hb_doc_to_visual at 0x7fe2d0dbcee0>",
27
- "doc_to_text": "<function hb_doc_to_text at 0x7fe2d0c13670>",
28
- "doc_to_target": "gt_answer_details",
29
- "process_results": "<function hb_process_results at 0x7fe2d0c13e50>",
30
- "description": "",
31
- "target_delimiter": " ",
32
- "fewshot_delimiter": "\n\n",
33
- "metric_list": [
34
- {
35
- "metric": "aAcc",
36
- "aggregation": "<function hb_aggregation_result_aAcc at 0x7fe2d0a94820>",
37
- "higher_is_better": true
38
- },
39
- {
40
- "metric": "qAcc",
41
- "aggregation": "<function hb_aggregation_result_qAcc at 0x7fe2d0a94e50>",
42
- "higher_is_better": true
43
- },
44
- {
45
- "metric": "fAcc",
46
- "aggregation": "<function hb_aggregation_result_fAcc at 0x7fe2d0a60670>",
47
- "higher_is_better": true
48
- }
49
- ],
50
- "output_type": "generate_until",
51
- "generation_kwargs": {
52
- "max_new_tokens": 128,
53
- "temperature": 0.0,
54
- "top_p": 1.0,
55
- "num_beams": 1,
56
- "do_sample": false,
57
- "until": [
58
- "\n\n"
59
- ]
60
- },
61
- "repeats": 1,
62
- "should_decontaminate": false,
63
- "metadata": [
64
- {
65
- "version": 0.0
66
- }
67
- ],
68
- "model_specific_prompt_kwargs": {
69
- "default": {
70
- "pre_prompt": "",
71
- "post_prompt": ""
72
- }
73
- }
74
- },
75
- "mathvista_testmini": {
76
- "task": "mathvista_testmini",
77
- "dataset_path": "AI4Math/MathVista",
78
- "dataset_kwargs": {
79
- "token": true
80
- },
81
- "test_split": "testmini",
82
- "doc_to_visual": "<function mathvista_doc_to_visual at 0x7fe2a4fc89d0>",
83
- "doc_to_text": "<function mathvista_doc_to_text at 0x7fe2a4fd30d0>",
84
- "doc_to_target": "answer",
85
- "process_results": "<function mathvista_process_results at 0x7fe2a4bcb790>",
86
- "description": "",
87
- "target_delimiter": " ",
88
- "fewshot_delimiter": "\n\n",
89
- "metric_list": [
90
- {
91
- "metric": "gpt_eval_score",
92
- "aggregation": "<function mathvista_aggregate_results at 0x7fe2a4bd3e50>",
93
- "higher_is_better": true
94
- }
95
- ],
96
- "output_type": "generate_until",
97
- "generation_kwargs": {
98
- "until": [
99
- "ASSISTANT:"
100
- ],
101
- "max_new_tokens": 1024,
102
- "temperature": 0.0,
103
- "top_p": 1.0,
104
- "num_beams": 1,
105
- "do_sample": false,
106
- "image_aspect_ratio": "original"
107
- },
108
- "repeats": 1,
109
- "should_decontaminate": false,
110
- "model_specific_prompt_kwargs": {
111
- "default": {
112
- "shot_type": "format-prompt",
113
- "shot": 0,
114
- "use_caption": false,
115
- "use_ocr": false
116
- },
117
- "phi3v": {
118
- "shot_type": "solution"
119
- }
120
- },
121
- "model_specific_generation_kwargs": {
122
- "llava": {
123
- "image_aspect_ratio": "original"
124
- }
125
- }
126
- }
127
- },
128
- "versions": {
129
- "hallusion_bench_image": "Yaml",
130
- "mathvista_testmini": "Yaml"
131
- },
132
- "n-shot": {
133
- "hallusion_bench_image": 0,
134
- "mathvista_testmini": 0
135
- },
136
- "model_configs": {
137
- "model": "llava",
138
- "model_args": "pretrained=/cm/archive/namnv78/checkpoints/Xphi35-siglip224/SMOE/1M3/Full_remoe/checkpoint-13786,conv_template=phi35",
139
- "batch_size": "1",
140
- "device": null,
141
- "limit": null,
142
- "bootstrap_iters": 100000,
143
- "gen_kwargs": ""
144
- },
145
- "git_hash": "289c7fe5"
146
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1412_llava...image_llava_model_args_595126/submissions/mathvista_testmini_scores.json DELETED
The diff for this file is too large to render. See raw diff
 
sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/ai2d.json DELETED
The diff for this file is too large to render. See raw diff
 
sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/infovqa_val.json DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:f26b32df54ad8f9f5c8ad9413761458b2923524459e0498689101f28c8e7d4bd
3
- size 576448204
 
 
 
 
sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/mmbench_en_dev.json DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:1ef6a3572059b9b3123aeff247cb17eaa934c5f33b7c4fdb862f5d56d9564c26
3
- size 14568248
 
 
 
 
sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/mme.json DELETED
The diff for this file is too large to render. See raw diff
 
sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/mmerealworld_lite.json DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:e8320b5d0524c75293d97a563749adb88c6e9812000a4c17282dc9cc68c17cb4
3
- size 1994104281
 
 
 
 
sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/mmmu_val.json DELETED
The diff for this file is too large to render. See raw diff
 
sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/mmstar.json DELETED
The diff for this file is too large to render. See raw diff
 
sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/rank0_metric_eval_done.txt DELETED
@@ -1 +0,0 @@
1
- rank 0 eval done
 
 
sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/rank1_metric_eval_done.txt DELETED
@@ -1 +0,0 @@
1
- rank 1 eval done
 
 
sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/rank2_metric_eval_done.txt DELETED
@@ -1 +0,0 @@
1
- rank 2 eval done
 
 
sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/results.json DELETED
@@ -1,683 +0,0 @@
1
- {
2
- "results": {
3
- "ai2d": {
4
- "exact_match,flexible-extract": 0.614961139896373,
5
- "exact_match_stderr,flexible-extract": 0.00875805772380654,
6
- "alias": "ai2d"
7
- },
8
- "infovqa_val": {
9
- "anls,none": 0.25276686897536593,
10
- "anls_stderr,none": 0.008213137970952944,
11
- "alias": "infovqa_val"
12
- },
13
- "mmbench_en_dev": {
14
- "gpt_eval_score,none": 54.98281786941581,
15
- "gpt_eval_score_stderr,none": "N/A",
16
- "submission,none": null,
17
- "submission_stderr,none": "N/A",
18
- "alias": "mmbench_en_dev"
19
- },
20
- "mme": {
21
- "mme_cognition_score,none": 276.7857142857143,
22
- "mme_cognition_score_stderr,none": "N/A",
23
- "mme_percetion_score,none": 1103.8987595038016,
24
- "mme_percetion_score_stderr,none": "N/A",
25
- "alias": "mme"
26
- },
27
- "mmerealworld_lite": {
28
- "mme_realworld_score,none": 0.2636789994788953,
29
- "mme_realworld_score_stderr,none": "N/A",
30
- "alias": "mmerealworld_lite"
31
- },
32
- "mmmu_val": {
33
- "mmmu_acc,none": 0.40111,
34
- "mmmu_acc_stderr,none": "N/A",
35
- "alias": "mmmu_val"
36
- },
37
- "mmstar": {
38
- "coarse perception,none": 0.4745496248584787,
39
- "coarse perception_stderr,none": "N/A",
40
- "fine-grained perception,none": 0.26887128469663474,
41
- "fine-grained perception_stderr,none": "N/A",
42
- "instance reasoning,none": 0.3861223080142979,
43
- "instance reasoning_stderr,none": "N/A",
44
- "logical reasoning,none": 0.35995214906106,
45
- "logical reasoning_stderr,none": "N/A",
46
- "math,none": 0.39322150940033856,
47
- "math_stderr,none": "N/A",
48
- "science & technology,none": 0.25396890595797994,
49
- "science & technology_stderr,none": "N/A",
50
- "alias": "mmstar"
51
- },
52
- "scienceqa_img": {
53
- "exact_match,none": 0.7441745166088249,
54
- "exact_match_stderr,none": 0.00971771224462157,
55
- "alias": "scienceqa_img"
56
- },
57
- "seedbench_2_plus": {
58
- "seedbench_2_plus_Chart,none": 0.44814814814814813,
59
- "seedbench_2_plus_Chart_stderr,none": "N/A",
60
- "seedbench_2_plus_all,none": 0.45322793148880103,
61
- "seedbench_2_plus_all_stderr,none": "N/A",
62
- "seedbench_2_plus_Web,none": 0.4954545454545455,
63
- "seedbench_2_plus_Web_stderr,none": "N/A",
64
- "seedbench_2_plus_Map,none": 0.42379182156133827,
65
- "seedbench_2_plus_Map_stderr,none": "N/A",
66
- "alias": "seedbench_2_plus"
67
- },
68
- "textvqa_val": {
69
- "exact_match,none": 0.2150200002193451,
70
- "exact_match_stderr,none": 0.005632905626047891,
71
- "submission,none": null,
72
- "submission_stderr,none": "N/A",
73
- "alias": "textvqa_val"
74
- }
75
- },
76
- "configs": {
77
- "ai2d": {
78
- "task": "ai2d",
79
- "dataset_path": "lmms-lab/ai2d",
80
- "dataset_kwargs": {
81
- "token": true
82
- },
83
- "test_split": "test",
84
- "doc_to_visual": "<function ai2d_doc_to_visual at 0x7f4c749b3af0>",
85
- "doc_to_text": "<function ai2d_doc_to_text at 0x7f4c749b3dc0>",
86
- "doc_to_target": "<function ai2d_doc_to_target at 0x7f4c749b93a0>",
87
- "description": "",
88
- "target_delimiter": " ",
89
- "fewshot_delimiter": "\n\n",
90
- "metric_list": [
91
- {
92
- "metric": "exact_match",
93
- "aggregation": "mean",
94
- "higher_is_better": true,
95
- "ignore_case": true,
96
- "ignore_punctuation": true
97
- }
98
- ],
99
- "output_type": "generate_until",
100
- "generation_kwargs": {
101
- "max_new_tokens": 512,
102
- "temperature": 0.0,
103
- "do_sample": false,
104
- "until": [
105
- "\n\n"
106
- ]
107
- },
108
- "repeats": 1,
109
- "filter_list": [
110
- {
111
- "name": "flexible-extract",
112
- "filter": [
113
- {
114
- "function": "<class 'utils.MultiChoiceRegexFilter'>",
115
- "group_select": 0,
116
- "ignore_case": true,
117
- "ignore_punctuation": true,
118
- "regex_pattern": "([A-Z])\\."
119
- }
120
- ]
121
- }
122
- ],
123
- "should_decontaminate": false,
124
- "metadata": [
125
- {
126
- "version": 0.0
127
- }
128
- ],
129
- "model_specific_prompt_kwargs": {
130
- "default": {
131
- "prompt_format": "mcq",
132
- "pre_prompt": "",
133
- "post_prompt": "\nAnswer with the option's letter from the given choices directly."
134
- },
135
- "gpt4v": {
136
- "prompt_format": "mcq",
137
- "pre_prompt": "",
138
- "post_prompt": "\nAbove choices are given in {option}. {content} format.\nPlease answer with the option letter from the given choices directly."
139
- },
140
- "qwen_vl": {
141
- "prompt_format": "qa",
142
- "pre_prompt": "",
143
- "post_prompt": " Answer:"
144
- },
145
- "xcomposer2_4khd": {
146
- "prompt_format": "mcq_xcomposer",
147
- "pre_prompt": "[UNUSED_TOKEN_146]user\nQuestion: ",
148
- "post_prompt": "[UNUSED_TOKEN_145]\n[UNUSED_TOKEN_146]assistant\nThe answer is"
149
- }
150
- },
151
- "model_specific_target_kwargs": {
152
- "default": "mcq",
153
- "qwen_vl": "qa"
154
- }
155
- },
156
- "infovqa_val": {
157
- "task": "infovqa_val",
158
- "dataset_path": "lmms-lab/DocVQA",
159
- "dataset_name": "InfographicVQA",
160
- "dataset_kwargs": {
161
- "token": true
162
- },
163
- "test_split": "validation",
164
- "doc_to_visual": "<function infovqa_doc_to_visual at 0x7f4ca74199d0>",
165
- "doc_to_text": "<function infovqa_doc_to_text at 0x7f4ca7419ca0>",
166
- "doc_to_target": "answers",
167
- "description": "",
168
- "target_delimiter": " ",
169
- "fewshot_delimiter": "\n\n",
170
- "metric_list": [
171
- {
172
- "metric": "anls",
173
- "aggregation": "mean",
174
- "higher_is_better": true
175
- }
176
- ],
177
- "output_type": "generate_until",
178
- "generation_kwargs": {
179
- "max_new_tokens": 32,
180
- "temperature": 0.0,
181
- "do_sample": false,
182
- "until": [
183
- "\n\n"
184
- ]
185
- },
186
- "repeats": 1,
187
- "should_decontaminate": false,
188
- "model_specific_prompt_kwargs": {
189
- "default": {
190
- "pre_prompt": "",
191
- "post_prompt": "\nAnswer the question using a single word or phrase."
192
- }
193
- }
194
- },
195
- "mmbench_en_dev": {
196
- "task": "mmbench_en_dev",
197
- "dataset_path": "lmms-lab/MMBench",
198
- "dataset_name": "en",
199
- "dataset_kwargs": {
200
- "token": true
201
- },
202
- "test_split": "dev",
203
- "doc_to_visual": "<function mmbench_doc_to_visual at 0x7f4cd45bd430>",
204
- "doc_to_text": "<function mmbench_doc_to_text at 0x7f4cd45bd940>",
205
- "doc_to_target": "answer",
206
- "process_results": "<function mmbench_process_results at 0x7f4cd45bde50>",
207
- "description": "",
208
- "target_delimiter": " ",
209
- "fewshot_delimiter": "\n\n",
210
- "metric_list": [
211
- {
212
- "metric": "gpt_eval_score",
213
- "aggregation": "<function mmbench_aggregate_dev_results_eval at 0x7f4cd4631ca0>",
214
- "higher_is_better": true
215
- },
216
- {
217
- "metric": "submission",
218
- "aggregation": "<function mmbench_aggregate_dev_results_submission at 0x7f4cd45bd1f0>",
219
- "higher_is_better": true
220
- }
221
- ],
222
- "output_type": "generate_until",
223
- "generation_kwargs": {
224
- "until": [
225
- "ASSISTANT:"
226
- ],
227
- "max_new_tokens": 1024,
228
- "temperature": 0.0,
229
- "top_p": 1.0,
230
- "num_beams": 1,
231
- "do_sample": false,
232
- "image_aspect_ratio": "original"
233
- },
234
- "repeats": 1,
235
- "should_decontaminate": false,
236
- "model_specific_prompt_kwargs": {
237
- "default": {
238
- "pre_prompt": "",
239
- "post_prompt": "\nAnswer with the option's letter from the given choices directly."
240
- }
241
- },
242
- "model_specific_generation_kwargs": {
243
- "llava": {
244
- "image_aspect_ratio": "original"
245
- }
246
- }
247
- },
248
- "mme": {
249
- "task": "mme",
250
- "dataset_path": "lmms-lab/MME",
251
- "dataset_kwargs": {
252
- "token": false
253
- },
254
- "test_split": "test",
255
- "doc_to_visual": "<function mme_doc_to_visual at 0x7f4cd56dc8b0>",
256
- "doc_to_text": "<function mme_doc_to_text at 0x7f4cd463d160>",
257
- "doc_to_target": "answer",
258
- "process_results": "<function mme_process_results at 0x7f4cd463d700>",
259
- "description": "",
260
- "target_delimiter": " ",
261
- "fewshot_delimiter": "\n\n",
262
- "metric_list": [
263
- {
264
- "metric": "mme_percetion_score",
265
- "aggregation": "<function mme_aggregate_results at 0x7f4cd463dc10>",
266
- "higher_is_better": true
267
- },
268
- {
269
- "metric": "mme_cognition_score",
270
- "aggregation": "<function mme_aggregate_results at 0x7f4cd46450d0>",
271
- "higher_is_better": true
272
- }
273
- ],
274
- "output_type": "generate_until",
275
- "generation_kwargs": {
276
- "max_new_tokens": 16,
277
- "temperature": 0.0,
278
- "top_p": 1.0,
279
- "num_beams": 1,
280
- "do_sample": false,
281
- "until": [
282
- "\n\n"
283
- ]
284
- },
285
- "repeats": 1,
286
- "should_decontaminate": false,
287
- "metadata": [
288
- {
289
- "version": 0.0
290
- }
291
- ],
292
- "model_specific_prompt_kwargs": {
293
- "default": {
294
- "pre_prompt": "",
295
- "post_prompt": "\nAnswer the question using a single word or phrase."
296
- },
297
- "gpt4v": {
298
- "pre_prompt": "",
299
- "post_prompt": "\nAnswer the question with Yes or No."
300
- },
301
- "qwen_vl": {
302
- "pre_prompt": "",
303
- "post_prompt": " Answer:"
304
- },
305
- "otterhd": {
306
- "pre_prompt": "",
307
- "post_prompt": " Answer:"
308
- },
309
- "xcomposer2_4khd": {
310
- "pre_prompt": "[UNUSED_TOKEN_146]user\n",
311
- "post_prompt": " Answer this question briefly[UNUSED_TOKEN_145]\n[UNUSED_TOKEN_146]assistant\n"
312
- }
313
- }
314
- },
315
- "mmerealworld_lite": {
316
- "task": "mmerealworld_lite",
317
- "dataset_path": "yifanzhang114/MME-RealWorld-lite-lmms-eval",
318
- "dataset_kwargs": {
319
- "token": true
320
- },
321
- "test_split": "train",
322
- "doc_to_visual": "<function mme_realworld_doc_to_visual at 0x7f4c7515aca0>",
323
- "doc_to_text": "<function mme_realworld_doc_to_text at 0x7f4c7515e790>",
324
- "doc_to_target": "answer",
325
- "process_results": "<function mme_realworld_process_results at 0x7f4c75167430>",
326
- "description": "",
327
- "target_delimiter": " ",
328
- "fewshot_delimiter": "\n\n",
329
- "metric_list": [
330
- {
331
- "metric": "mme_realworld_score",
332
- "aggregation": "<function mme_realworld_aggregate_results at 0x7f4c75167f70>",
333
- "higher_is_better": true
334
- }
335
- ],
336
- "output_type": "generate_until",
337
- "generation_kwargs": {
338
- "max_new_tokens": 16,
339
- "temperature": 0.0,
340
- "top_p": 1.0,
341
- "num_beams": 1,
342
- "do_sample": false,
343
- "until": [
344
- "\n\n"
345
- ]
346
- },
347
- "repeats": 1,
348
- "should_decontaminate": false,
349
- "metadata": [
350
- {
351
- "version": 0.0
352
- }
353
- ],
354
- "model_specific_prompt_kwargs": {
355
- "default": {
356
- "pre_prompt": "",
357
- "post_prompt": "\nSelect the best answer to the above multiple-choice question based on the image. Respond with only the letter (A, B, C, D, or E) of the correct option."
358
- },
359
- "gpt4v": {
360
- "pre_prompt": "",
361
- "post_prompt": "\nSelect the best answer to the above multiple-choice question based on the image. Respond with only the letter (A, B, C, D, or E) of the correct option."
362
- },
363
- "xcomposer2_4khd": {
364
- "pre_prompt": "[UNUSED_TOKEN_146]user\n",
365
- "post_prompt": " Answer this question with A, B, C, or D.[UNUSED_TOKEN_145]\n[UNUSED_TOKEN_146]assistant\n"
366
- }
367
- }
368
- },
369
- "mmmu_val": {
370
- "task": "mmmu_val",
371
- "dataset_path": "lmms-lab/MMMU",
372
- "test_split": "validation",
373
- "doc_to_visual": "<function mmmu_doc_to_visual at 0x7f4c8746f700>",
374
- "doc_to_text": "<function mmmu_doc_to_text at 0x7f4c8743d430>",
375
- "doc_to_target": "answer",
376
- "process_results": "<function mmmu_process_results at 0x7f4c872ad310>",
377
- "description": "",
378
- "target_delimiter": " ",
379
- "fewshot_delimiter": "\n\n",
380
- "metric_list": [
381
- {
382
- "metric": "mmmu_acc",
383
- "aggregation": "<function mmmu_aggregate_results at 0x7f4c8706c280>",
384
- "higher_is_better": true
385
- }
386
- ],
387
- "output_type": "generate_until",
388
- "generation_kwargs": {
389
- "max_new_tokens": 128,
390
- "until": [
391
- "\n\n"
392
- ],
393
- "image_aspect_ratio": "original"
394
- },
395
- "repeats": 1,
396
- "should_decontaminate": false,
397
- "metadata": [
398
- {
399
- "version": 0.0
400
- }
401
- ],
402
- "model_specific_generation_kwargs": {
403
- "llava": {
404
- "image_aspect_ratio": "original"
405
- }
406
- }
407
- },
408
- "mmstar": {
409
- "task": "mmstar",
410
- "dataset_path": "Lin-Chen/MMStar",
411
- "dataset_kwargs": {
412
- "token": true
413
- },
414
- "test_split": "val",
415
- "doc_to_visual": "<function mmstar_doc_to_visual at 0x7f4c7490f160>",
416
- "doc_to_text": "<function mmstar_doc_to_text at 0x7f4c7490f5e0>",
417
- "doc_to_target": "answer",
418
- "process_results": "<function mmstar_process_results at 0x7f4c7490faf0>",
419
- "description": "",
420
- "target_delimiter": " ",
421
- "fewshot_delimiter": "\n\n",
422
- "metric_list": [
423
- {
424
- "metric": "coarse perception",
425
- "aggregation": "<function mmstar_aggregate_results at 0x7f4c7490ff70>",
426
- "higher_is_better": true
427
- },
428
- {
429
- "metric": "fine-grained perception",
430
- "aggregation": "<function mmstar_aggregate_results at 0x7f4c749193a0>",
431
- "higher_is_better": true
432
- },
433
- {
434
- "metric": "instance reasoning",
435
- "aggregation": "<function mmstar_aggregate_results at 0x7f4c74919790>",
436
- "higher_is_better": true
437
- },
438
- {
439
- "metric": "logical reasoning",
440
- "aggregation": "<function mmstar_aggregate_results at 0x7f4c74919b80>",
441
- "higher_is_better": true
442
- },
443
- {
444
- "metric": "science & technology",
445
- "aggregation": "<function mmstar_aggregate_results at 0x7f4c74919f70>",
446
- "higher_is_better": true
447
- },
448
- {
449
- "metric": "math",
450
- "aggregation": "<function mmstar_aggregate_results at 0x7f4c749213a0>",
451
- "higher_is_better": true
452
- }
453
- ],
454
- "output_type": "generate_until",
455
- "generation_kwargs": {
456
- "until": [
457
- "\n\n"
458
- ],
459
- "do_sample": false
460
- },
461
- "repeats": 1,
462
- "should_decontaminate": false,
463
- "metadata": [
464
- {
465
- "version": 0.0
466
- }
467
- ],
468
- "model_specific_prompt_kwargs": {
469
- "default": {
470
- "pre_prompt": "",
471
- "post_prompt": "\nAnswer with the option's letter from the given choices directly"
472
- }
473
- }
474
- },
475
- "scienceqa_img": {
476
- "task": "scienceqa_img",
477
- "dataset_path": "lmms-lab/ScienceQA",
478
- "dataset_name": "ScienceQA-IMG",
479
- "dataset_kwargs": {
480
- "token": true
481
- },
482
- "test_split": "test",
483
- "doc_to_visual": "<function sqa_doc_to_visual at 0x7f4c9f37bb80>",
484
- "doc_to_text": "<function sqa_doc_to_text at 0x7f4c9f37bdc0>",
485
- "doc_to_target": "<function sqa_doc_to_target at 0x7f4c9f2fc1f0>",
486
- "process_results": "<function sqa_process_results at 0x7f4c9f2fc550>",
487
- "description": "",
488
- "target_delimiter": " ",
489
- "fewshot_delimiter": "\n\n",
490
- "metric_list": [
491
- {
492
- "metric": "exact_match",
493
- "aggregation": "mean",
494
- "higher_is_better": true,
495
- "ignore_case": true,
496
- "ignore_punctuation": true
497
- }
498
- ],
499
- "output_type": "generate_until",
500
- "generation_kwargs": {
501
- "max_new_tokens": 16,
502
- "temperature": 0.0,
503
- "do_sample": false,
504
- "until": [
505
- "\n\n"
506
- ],
507
- "image_aspect_ratio": "original"
508
- },
509
- "repeats": 1,
510
- "should_decontaminate": false,
511
- "metadata": [
512
- {
513
- "version": 0.0
514
- }
515
- ],
516
- "model_specific_prompt_kwargs": {
517
- "default": {
518
- "format": "default",
519
- "pre_prompt": "",
520
- "post_prompt": "\nAnswer with the option's letter from the given choices directly."
521
- },
522
- "qwen_vl": {
523
- "format": "qwen_vl"
524
- },
525
- "idefics2": {
526
- "format": "default",
527
- "pre_prompt": "",
528
- "post_prompt": "\nAnswer:"
529
- }
530
- },
531
- "model_specific_generation_kwargs": {
532
- "llava": {
533
- "image_aspect_ratio": "original"
534
- }
535
- }
536
- },
537
- "seedbench_2_plus": {
538
- "task": "seedbench_2_plus",
539
- "dataset_path": "doolayer/SEED-Bench-2-Plus",
540
- "dataset_kwargs": {
541
- "token": true
542
- },
543
- "test_split": "test",
544
- "doc_to_visual": "<function seed_doc_to_visual at 0x7f4c9dcfcca0>",
545
- "doc_to_text": "<function seed_doc_to_text at 0x7f4c9dc5a310>",
546
- "doc_to_target": "answer",
547
- "process_results": "<function seed_process_result at 0x7f4c9dc5a820>",
548
- "description": "",
549
- "target_delimiter": " ",
550
- "fewshot_delimiter": "\n\n",
551
- "metric_list": [
552
- {
553
- "metric": "seedbench_2_plus_Chart",
554
- "aggregation": "<function seed_aggregation_result at 0x7f4c9dc5ad30>",
555
- "higher_is_better": true
556
- },
557
- {
558
- "metric": "seedbench_2_plus_Map",
559
- "aggregation": "<function seed_aggregation_result at 0x7f4c9dc7b1f0>",
560
- "higher_is_better": true
561
- },
562
- {
563
- "metric": "seedbench_2_plus_Web",
564
- "aggregation": "<function seed_aggregation_result at 0x7f4c9dc7b670>",
565
- "higher_is_better": true
566
- },
567
- {
568
- "metric": "seedbench_2_plus_all",
569
- "aggregation": "<function seed_aggregation_result at 0x7f4c9dc7baf0>",
570
- "higher_is_better": true
571
- }
572
- ],
573
- "output_type": "generate_until",
574
- "generation_kwargs": {
575
- "until": [
576
- "ASSISTANT:"
577
- ],
578
- "max_new_tokens": 16,
579
- "image_aspect_ratio": "original"
580
- },
581
- "repeats": 1,
582
- "should_decontaminate": false,
583
- "metadata": [
584
- {
585
- "version": 0.0
586
- }
587
- ],
588
- "model_specific_prompt_kwargs": {
589
- "llava": {
590
- "img_token": "<image>",
591
- "post_prompt": "Answer with the option's letter from the given choices directly."
592
- },
593
- "gpt4V": {
594
- "img_token": "<image>",
595
- "post_prompt": "Answer with the option's letter from the given choices directly."
596
- },
597
- "default": {
598
- "img_token": "<image>",
599
- "post_prompt": "Answer with the option's letter from the given choices directly."
600
- }
601
- }
602
- },
603
- "textvqa_val": {
604
- "task": "textvqa_val",
605
- "dataset_path": "lmms-lab/textvqa",
606
- "test_split": "validation",
607
- "doc_to_visual": "<function textvqa_doc_to_visual at 0x7f4ca0a8e790>",
608
- "doc_to_text": "<function textvqa_doc_to_text at 0x7f4ca0a8ec10>",
609
- "doc_to_target": "answer",
610
- "process_results": "<function textvqa_process_results at 0x7f4ca0a8eee0>",
611
- "description": "",
612
- "target_delimiter": " ",
613
- "fewshot_delimiter": "\n\n",
614
- "metric_list": [
615
- {
616
- "metric": "exact_match",
617
- "aggregation": "mean",
618
- "higher_is_better": true,
619
- "ignore_case": true,
620
- "ignore_punctuation": true
621
- },
622
- {
623
- "metric": "submission",
624
- "aggregation": "<function textvqa_aggregate_submissions at 0x7f4ca0a8e550>",
625
- "higher_is_better": true
626
- }
627
- ],
628
- "output_type": "generate_until",
629
- "generation_kwargs": {
630
- "until": [
631
- "ASSISTANT:"
632
- ]
633
- },
634
- "repeats": 1,
635
- "should_decontaminate": false,
636
- "model_specific_prompt_kwargs": {
637
- "default": {
638
- "pre_prompt": "",
639
- "post_prompt": "\nAnswer the question using a single word or phrase.",
640
- "ocr": false
641
- },
642
- "qwen_vl": {
643
- "pre_prompt": "",
644
- "post_prompt": " Answer:"
645
- }
646
- }
647
- }
648
- },
649
- "versions": {
650
- "ai2d": "Yaml",
651
- "infovqa_val": "Yaml",
652
- "mmbench_en_dev": "Yaml",
653
- "mme": "Yaml",
654
- "mmerealworld_lite": "Yaml",
655
- "mmmu_val": "Yaml",
656
- "mmstar": "Yaml",
657
- "scienceqa_img": "Yaml",
658
- "seedbench_2_plus": "Yaml",
659
- "textvqa_val": "Yaml"
660
- },
661
- "n-shot": {
662
- "ai2d": 0,
663
- "infovqa_val": 0,
664
- "mmbench_en_dev": 0,
665
- "mme": 0,
666
- "mmerealworld_lite": 0,
667
- "mmmu_val": 0,
668
- "mmstar": 0,
669
- "scienceqa_img": 0,
670
- "seedbench_2_plus": 0,
671
- "textvqa_val": 0
672
- },
673
- "model_configs": {
674
- "model": "llava",
675
- "model_args": "pretrained=/cm/archive/namnv78/checkpoints/Xphi35-siglip224/SMOE/1M3/Full_remoe/checkpoint-13786,conv_template=phi35",
676
- "batch_size": "1",
677
- "device": null,
678
- "limit": null,
679
- "bootstrap_iters": 100000,
680
- "gen_kwargs": ""
681
- },
682
- "git_hash": "289c7fe5"
683
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/scienceqa_img.json DELETED
The diff for this file is too large to render. See raw diff
 
sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/seedbench_2_plus.json DELETED
The diff for this file is too large to render. See raw diff
 
sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/submissions/mmbench_en_dev_results.json DELETED
@@ -1 +0,0 @@
1
- {"overall_acc": 0.5498281786941581, "category_acc": {"action_recognition": 0.7962962962962963, "attribute_comparison": 0.36363636363636365, "attribute_recognition": 0.6621621621621622, "celebrity_recognition": 0.48484848484848486, "function_reasoning": 0.7721518987341772, "future_prediction": 0.375, "identity_reasoning": 0.9333333333333333, "image_emotion": 0.54, "image_quality": 0.09433962264150944, "image_scene": 0.875, "image_style": 0.6981132075471698, "image_topic": 0.7222222222222222, "nature_relation": 0.3958333333333333, "object_localization": 0.2345679012345679, "ocr": 0.7948717948717948, "physical_property_reasoning": 0.49333333333333335, "physical_relation": 0.5416666666666666, "social_relation": 0.5813953488372093, "spatial_relationship": 0.17777777777777778, "structuralized_imagetext_understanding": 0.358974358974359}, "l2_category_acc": {"attribute_reasoning": 0.7035175879396985, "coarse_perception": 0.6283783783783784, "finegrained_perception (cross-instance)": 0.46853146853146854, "finegrained_perception (instance-level)": 0.5017064846416383, "logic_reasoning": 0.3644067796610169, "relation_reasoning": 0.4956521739130435}}
 
 
sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/submissions/mmbench_en_dev_results.xlsx DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:2d0e0fe22a7c47e99366258abf96d09e305355def5dc23c3e7e2f2f78e51fdda
3
- size 843188
 
 
 
 
sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/submissions/textvqa_submission_2025-06-27-13-58-34.json DELETED
The diff for this file is too large to render. See raw diff
 
sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/textvqa_val.json DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:6c92ae4eb14b5fb41c01b5c572fb62bd0815727bfc0aababd01756efebff0fbc
3
- size 13260575
 
 
 
 
sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1459_llava...bench_llava_model_args_595126/ocrbench.json DELETED
The diff for this file is too large to render. See raw diff
 
sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1459_llava...bench_llava_model_args_595126/rank0_metric_eval_done.txt DELETED
@@ -1 +0,0 @@
1
- rank 0 eval done
 
 
sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1459_llava...bench_llava_model_args_595126/rank1_metric_eval_done.txt DELETED
@@ -1 +0,0 @@
1
- rank 1 eval done
 
 
sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1459_llava...bench_llava_model_args_595126/rank2_metric_eval_done.txt DELETED
@@ -1 +0,0 @@
1
- rank 2 eval done
 
 
sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1459_llava...bench_llava_model_args_595126/results.json DELETED
@@ -1,67 +0,0 @@
1
- {
2
- "results": {
3
- "ocrbench": {
4
- "ocrbench_accuracy,none": 0.191,
5
- "ocrbench_accuracy_stderr,none": "N/A",
6
- "alias": "ocrbench"
7
- }
8
- },
9
- "configs": {
10
- "ocrbench": {
11
- "task": "ocrbench",
12
- "dataset_path": "echo840/OCRBench",
13
- "dataset_kwargs": {
14
- "token": true
15
- },
16
- "test_split": "test",
17
- "doc_to_visual": "<function ocrbench_doc_to_visual at 0x7fb593092b80>",
18
- "doc_to_text": "<function ocrbench_doc_to_text at 0x7fb592b56160>",
19
- "doc_to_target": "answer",
20
- "process_results": "<function ocrbench_process_results at 0x7fb592b564c0>",
21
- "description": "",
22
- "target_delimiter": " ",
23
- "fewshot_delimiter": "\n\n",
24
- "metric_list": [
25
- {
26
- "metric": "ocrbench_accuracy",
27
- "aggregation": "<function ocrbench_aggregate_accuracy at 0x7fb592b56820>",
28
- "higher_is_better": true
29
- }
30
- ],
31
- "output_type": "generate_until",
32
- "generation_kwargs": {
33
- "max_new_tokens": 128,
34
- "temperature": 0.0,
35
- "top_p": 1.0,
36
- "num_beams": 1,
37
- "do_sample": false,
38
- "until": [
39
- "\n\n"
40
- ]
41
- },
42
- "repeats": 1,
43
- "should_decontaminate": false,
44
- "metadata": [
45
- {
46
- "version": 0.0
47
- }
48
- ]
49
- }
50
- },
51
- "versions": {
52
- "ocrbench": "Yaml"
53
- },
54
- "n-shot": {
55
- "ocrbench": 0
56
- },
57
- "model_configs": {
58
- "model": "llava",
59
- "model_args": "pretrained=/cm/archive/namnv78/checkpoints/Xphi35-siglip224/SMOE/1M3/Full_remoe/checkpoint-13786,conv_template=phi35",
60
- "batch_size": "1",
61
- "device": null,
62
- "limit": null,
63
- "bootstrap_iters": 100000,
64
- "gen_kwargs": ""
65
- },
66
- "git_hash": "289c7fe5"
67
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1459_llava...bench_llava_model_args_595126/results/ocrbench_results.txt DELETED
@@ -1,18 +0,0 @@
1
- ######################### OCRBench #############################
2
- Text Recognition(Total 300): 109
3
- ---------------- Details of Recognition Score ------------------
4
- Regular Text Recognition(Total 50): 23
5
- Irregular Text Recognition(Total 50): 22
6
- Artistic Text Recognition(Total 50): 36
7
- Handwriting Recognition(Total 50): 15
8
- Digit String Recognition(Total 50): 6
9
- Non-Semantic Text Recognition(Total 50): 7
10
- ----------------------------------------------------------------
11
- Scene Text-centric VQA(Total 200): 62
12
- ----------------------------------------------------------------
13
- Doc-oriented VQA(Total 200): 17
14
- ----------------------------------------------------------------
15
- Key Information Extraction(Total 200): 2
16
- Handwritten Mathematical Expression Recognition(Total 100): 1
17
- --------------------- Final Score ------------------------------
18
- Final Score(Total 1000): 191
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
sft/1M3/Full_remoe/checkpoint-13786/model-00001-of-00003.safetensors DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:b61b40c5a7d6b823014de1f3aac0bd78c5e5870ce6db656b12990131d8830e92
3
- size 4972489328
 
 
 
 
sft/1M3/Full_remoe/checkpoint-13786/model-00002-of-00003.safetensors DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:915ec471ac15b80f811f6966233edb819c8be0375f68dcfd0cbe9347ada2590a
3
- size 4985529648
 
 
 
 
sft/1M3/Full_remoe/checkpoint-13786/model-00003-of-00003.safetensors DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:26658d3efca94e9d3bd68dccf60b1eca21d4751d582b4d067c20986d699486aa
3
- size 248943552
 
 
 
 
sft/1M3/Full_remoe/checkpoint-13786/model.safetensors.index.json DELETED
The diff for this file is too large to render. See raw diff
 
sft/1M3/Full_remoe/checkpoint-13786/rng_state_0.pth DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:d98bcd53a09c8bc1238dda64e113dd6869500cd3502c9a4d572b3f9ffea5dc29
3
- size 14960
 
 
 
 
sft/1M3/Full_remoe/checkpoint-13786/rng_state_1.pth DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:f793f5c89da7b5544f5038c073ba86532c704fdd2918ea5c27bfab848c20ad52
3
- size 14960
 
 
 
 
sft/1M3/Full_remoe/checkpoint-13786/rng_state_2.pth DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:cee33fb501fe6367a1ba811f1189b771eecbe30ed36d18486d5aeb9e083e4fdf
3
- size 14960
 
 
 
 
sft/1M3/Full_remoe/checkpoint-13786/rng_state_3.pth DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:d1aa2a4d8920b433e30b4dffe2a189b30027ce66925ec5b6b0ae0f1ccd678ec7
3
- size 14960
 
 
 
 
sft/1M3/Full_remoe/checkpoint-13786/special_tokens_map.json DELETED
@@ -1,24 +0,0 @@
1
- {
2
- "bos_token": {
3
- "content": "<s>",
4
- "lstrip": false,
5
- "normalized": false,
6
- "rstrip": false,
7
- "single_word": false
8
- },
9
- "eos_token": {
10
- "content": "<|endoftext|>",
11
- "lstrip": false,
12
- "normalized": false,
13
- "rstrip": false,
14
- "single_word": false
15
- },
16
- "pad_token": "<unk>",
17
- "unk_token": {
18
- "content": "<unk>",
19
- "lstrip": false,
20
- "normalized": false,
21
- "rstrip": false,
22
- "single_word": false
23
- }
24
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
sft/1M3/Full_remoe/checkpoint-13786/tokenizer.model DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:9e556afd44213b6bd1be2b850ebbbd98f5481437a8021afaf58ee7fb1818d347
3
- size 499723
 
 
 
 
sft/1M3/Full_remoe/checkpoint-13786/tokenizer_config.json DELETED
@@ -1,132 +0,0 @@
1
- {
2
- "add_bos_token": false,
3
- "add_eos_token": false,
4
- "add_prefix_space": true,
5
- "added_tokens_decoder": {
6
- "0": {
7
- "content": "<unk>",
8
- "lstrip": false,
9
- "normalized": false,
10
- "rstrip": false,
11
- "single_word": false,
12
- "special": true
13
- },
14
- "1": {
15
- "content": "<s>",
16
- "lstrip": false,
17
- "normalized": false,
18
- "rstrip": false,
19
- "single_word": false,
20
- "special": true
21
- },
22
- "2": {
23
- "content": "</s>",
24
- "lstrip": false,
25
- "normalized": false,
26
- "rstrip": true,
27
- "single_word": false,
28
- "special": false
29
- },
30
- "32000": {
31
- "content": "<|endoftext|>",
32
- "lstrip": false,
33
- "normalized": false,
34
- "rstrip": false,
35
- "single_word": false,
36
- "special": true
37
- },
38
- "32001": {
39
- "content": "<|assistant|>",
40
- "lstrip": false,
41
- "normalized": false,
42
- "rstrip": true,
43
- "single_word": false,
44
- "special": true
45
- },
46
- "32002": {
47
- "content": "<|placeholder1|>",
48
- "lstrip": false,
49
- "normalized": false,
50
- "rstrip": true,
51
- "single_word": false,
52
- "special": true
53
- },
54
- "32003": {
55
- "content": "<|placeholder2|>",
56
- "lstrip": false,
57
- "normalized": false,
58
- "rstrip": true,
59
- "single_word": false,
60
- "special": true
61
- },
62
- "32004": {
63
- "content": "<|placeholder3|>",
64
- "lstrip": false,
65
- "normalized": false,
66
- "rstrip": true,
67
- "single_word": false,
68
- "special": true
69
- },
70
- "32005": {
71
- "content": "<|placeholder4|>",
72
- "lstrip": false,
73
- "normalized": false,
74
- "rstrip": true,
75
- "single_word": false,
76
- "special": true
77
- },
78
- "32006": {
79
- "content": "<|system|>",
80
- "lstrip": false,
81
- "normalized": false,
82
- "rstrip": true,
83
- "single_word": false,
84
- "special": true
85
- },
86
- "32007": {
87
- "content": "<|end|>",
88
- "lstrip": false,
89
- "normalized": false,
90
- "rstrip": true,
91
- "single_word": false,
92
- "special": true
93
- },
94
- "32008": {
95
- "content": "<|placeholder5|>",
96
- "lstrip": false,
97
- "normalized": false,
98
- "rstrip": true,
99
- "single_word": false,
100
- "special": true
101
- },
102
- "32009": {
103
- "content": "<|placeholder6|>",
104
- "lstrip": false,
105
- "normalized": false,
106
- "rstrip": true,
107
- "single_word": false,
108
- "special": true
109
- },
110
- "32010": {
111
- "content": "<|user|>",
112
- "lstrip": false,
113
- "normalized": false,
114
- "rstrip": true,
115
- "single_word": false,
116
- "special": true
117
- }
118
- },
119
- "bos_token": "<s>",
120
- "chat_template": "{% for message in messages %}{% if message['role'] == 'system' and message['content'] %}{{'<|system|>\n' + message['content'] + '<|end|>\n'}}{% elif message['role'] == 'user' %}{{'<|user|>\n' + message['content'] + '<|end|>\n'}}{% elif message['role'] == 'assistant' %}{{'<|assistant|>\n' + message['content'] + '<|end|>\n'}}{% endif %}{% endfor %}{% if add_generation_prompt %}{{ '<|assistant|>\n' }}{% else %}{{ eos_token }}{% endif %}",
121
- "clean_up_tokenization_spaces": false,
122
- "eos_token": "<|endoftext|>",
123
- "legacy": false,
124
- "model_max_length": 2048,
125
- "pad_token": "<unk>",
126
- "padding_side": "right",
127
- "sp_model_kwargs": {},
128
- "spaces_between_special_tokens": false,
129
- "tokenizer_class": "LlamaTokenizer",
130
- "unk_token": "<unk>",
131
- "use_default_system_prompt": false
132
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
sft/1M3/Full_remoe/checkpoint-13786/trainer_state.json DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:ef3ee1b637156b86e778b44896c89f7a3ec5372e63369a290b21ac0e38205939
3
- size 11276881
 
 
 
 
sft/1M3/Full_remoe/checkpoint-13786/training_args.bin DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:3babbfcec145033842a4b1261cac4f82423da7e133b1a6117c45af84c17ebc6d
3
- size 7992
 
 
 
 
sft/1M3/Full_remoe/checkpoint-13786/zero_to_fp32.py DELETED
@@ -1,604 +0,0 @@
1
- #!/usr/bin/env python
2
-
3
- # Copyright (c) Microsoft Corporation.
4
- # SPDX-License-Identifier: Apache-2.0
5
-
6
- # DeepSpeed Team
7
-
8
- # This script extracts fp32 consolidated weights from a zero 1, 2 and 3 DeepSpeed checkpoints. It gets
9
- # copied into the top level checkpoint dir, so the user can easily do the conversion at any point in
10
- # the future. Once extracted, the weights don't require DeepSpeed and can be used in any
11
- # application.
12
- #
13
- # example: 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)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
sft/1M3/Full_remoe/checkpoint-20679/added_tokens.json DELETED
@@ -1,13 +0,0 @@
1
- {
2
- "<|assistant|>": 32001,
3
- "<|endoftext|>": 32000,
4
- "<|end|>": 32007,
5
- "<|placeholder1|>": 32002,
6
- "<|placeholder2|>": 32003,
7
- "<|placeholder3|>": 32004,
8
- "<|placeholder4|>": 32005,
9
- "<|placeholder5|>": 32008,
10
- "<|placeholder6|>": 32009,
11
- "<|system|>": 32006,
12
- "<|user|>": 32010
13
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
sft/1M3/Full_remoe/checkpoint-20679/config.json DELETED
@@ -1,203 +0,0 @@
1
- {
2
- "_name_or_path": "/cm/archive/namnv78/checkpoints/phi35-siglip224/pft",
3
- "architectures": [
4
- "LlavaPhiForCausalLM"
5
- ],
6
- "attention_bias": false,
7
- "attention_dropout": 0.0,
8
- "auto_map": {
9
- "AutoConfig": "configuration_phi3.Phi3Config",
10
- "AutoModelForCausalLM": "modeling_phi3.Phi3ForCausalLM"
11
- },
12
- "bal_comp_loss_coef": 0.01,
13
- "balance_loss_coef": 1.5e-323,
14
- "bos_token_id": 1,
15
- "clip_smoe": true,
16
- "diversity_loss_coef": 0.01,
17
- "dropout": false,
18
- "e_loss_coef": 0.001,
19
- "embd_pdrop": 0.0,
20
- "entropy_advance_loss": false,
21
- "eos_token_id": 32000,
22
- "freeze_backbone": false,
23
- "freeze_mm_mlp_adapter": false,
24
- "hidden_act": "silu",
25
- "hidden_size": 3072,
26
- "hybrid": false,
27
- "image_aspect_ratio": "pad",
28
- "init_weight": true,
29
- "initializer_range": 0.02,
30
- "intermediate_size": 8192,
31
- "is_cosine": false,
32
- "is_norm_weight": false,
33
- "local_rank": 0,
34
- "loss1": "balanceloss",
35
- "loss2": "zloss",
36
- "luna": false,
37
- "max_compete_in_iter": 3,
38
- "max_position_embeddings": 131072,
39
- "mlp_smoe": true,
40
- "mm_hidden_size": 1152,
41
- "mm_patch_merge_type": "flat",
42
- "mm_projector_lr": null,
43
- "mm_projector_type": "moe",
44
- "mm_use_im_patch_token": false,
45
- "mm_use_im_start_end": false,
46
- "mm_vision_select_feature": "patch",
47
- "mm_vision_select_layer": -2,
48
- "mm_vision_tower": "google/siglip-so400m-patch14-224",
49
- "model_name_or_path": "/cm/archive/namnv78/checkpoints/phi35-siglip224/pft",
50
- "model_type": "llava_phi",
51
- "moe_name": "remoe",
52
- "moe_relu_l1_reg_coeff_multiplier": 1.2,
53
- "mp_pixel_shuffle_factor": 1,
54
- "norm_softmax": false,
55
- "normalization": true,
56
- "num_attention_heads": 32,
57
- "num_experts": 4,
58
- "num_hidden_layers": 32,
59
- "num_key_value_heads": 32,
60
- "num_layers": 3,
61
- "num_selected": 2,
62
- "number_of_previous_tokens": 2,
63
- "original_max_position_embeddings": 4096,
64
- "pad_token_id": 32000,
65
- "pretrain_mm_mlp_adapter": "/cm/archive/namnv78/checkpoints/phi35-siglip224/pft/mm_projector.bin",
66
- "rate_compete": 0.2,
67
- "rate_flip": 0.05,
68
- "resid_pdrop": 0.0,
69
- "rms_norm_eps": 1e-05,
70
- "rope_scaling": {
71
- "long_factor": [
72
- 1.0800000429153442,
73
- 1.1100000143051147,
74
- 1.1399999856948853,
75
- 1.340000033378601,
76
- 1.5899999141693115,
77
- 1.600000023841858,
78
- 1.6200000047683716,
79
- 2.620000123977661,
80
- 3.2300000190734863,
81
- 3.2300000190734863,
82
- 4.789999961853027,
83
- 7.400000095367432,
84
- 7.700000286102295,
85
- 9.09000015258789,
86
- 12.199999809265137,
87
- 17.670000076293945,
88
- 24.46000099182129,
89
- 28.57000160217285,
90
- 30.420001983642578,
91
- 30.840002059936523,
92
- 32.590003967285156,
93
- 32.93000411987305,
94
- 42.320003509521484,
95
- 44.96000289916992,
96
- 50.340003967285156,
97
- 50.45000457763672,
98
- 57.55000305175781,
99
- 57.93000411987305,
100
- 58.21000289916992,
101
- 60.1400032043457,
102
- 62.61000442504883,
103
- 62.62000274658203,
104
- 62.71000289916992,
105
- 63.1400032043457,
106
- 63.1400032043457,
107
- 63.77000427246094,
108
- 63.93000411987305,
109
- 63.96000289916992,
110
- 63.970001220703125,
111
- 64.02999877929688,
112
- 64.06999969482422,
113
- 64.08000183105469,
114
- 64.12000274658203,
115
- 64.41000366210938,
116
- 64.4800033569336,
117
- 64.51000213623047,
118
- 64.52999877929688,
119
- 64.83999633789062
120
- ],
121
- "short_factor": [
122
- 1.0,
123
- 1.0199999809265137,
124
- 1.0299999713897705,
125
- 1.0299999713897705,
126
- 1.0499999523162842,
127
- 1.0499999523162842,
128
- 1.0499999523162842,
129
- 1.0499999523162842,
130
- 1.0499999523162842,
131
- 1.0699999332427979,
132
- 1.0999999046325684,
133
- 1.1099998950958252,
134
- 1.1599998474121094,
135
- 1.1599998474121094,
136
- 1.1699998378753662,
137
- 1.2899998426437378,
138
- 1.339999794960022,
139
- 1.679999828338623,
140
- 1.7899998426437378,
141
- 1.8199998140335083,
142
- 1.8499997854232788,
143
- 1.8799997568130493,
144
- 1.9099997282028198,
145
- 1.9399996995925903,
146
- 1.9899996519088745,
147
- 2.0199997425079346,
148
- 2.0199997425079346,
149
- 2.0199997425079346,
150
- 2.0199997425079346,
151
- 2.0199997425079346,
152
- 2.0199997425079346,
153
- 2.0299997329711914,
154
- 2.0299997329711914,
155
- 2.0299997329711914,
156
- 2.0299997329711914,
157
- 2.0299997329711914,
158
- 2.0299997329711914,
159
- 2.0299997329711914,
160
- 2.0299997329711914,
161
- 2.0299997329711914,
162
- 2.0799996852874756,
163
- 2.0899996757507324,
164
- 2.189999580383301,
165
- 2.2199995517730713,
166
- 2.5899994373321533,
167
- 2.729999542236328,
168
- 2.749999523162842,
169
- 2.8399994373321533
170
- ],
171
- "type": "longrope"
172
- },
173
- "rope_theta": 10000.0,
174
- "router_loss_coef": 0.01,
175
- "router_theta": 0.1,
176
- "router_z_loss_coef": 0.001,
177
- "scales": [
178
- 1,
179
- 3
180
- ],
181
- "sliding_window": 262144,
182
- "sparse_upcycling": true,
183
- "std_gate": 0.02,
184
- "strategy_train": "base",
185
- "tie_word_embeddings": false,
186
- "tokenizer_model_max_length": 2048,
187
- "tokenizer_padding_side": "right",
188
- "topk_max": 2,
189
- "topk_min": 1,
190
- "torch_dtype": "bfloat16",
191
- "training": true,
192
- "transformers_version": "4.43.0",
193
- "tune_mm_mlp_adapter": false,
194
- "unit_test": true,
195
- "use_cache": false,
196
- "use_mm_proj": true,
197
- "use_old": false,
198
- "version": "phi35",
199
- "vision_tower": "google/siglip-so400m-patch14-224",
200
- "vision_tower_dir": "/cm/archive/namnv78/checkpoints/phi35-siglip224/pft/clip.bin",
201
- "vocab_size": 32064,
202
- "warm_up": 0.05
203
- }