Delete sft/1M3
Browse filesThis view is limited to 50 files because it contains too many changes.
See raw diff
- sft/1M3/Full_remoe/checkpoint-13786/added_tokens.json +0 -13
- sft/1M3/Full_remoe/checkpoint-13786/config.json +0 -203
- sft/1M3/Full_remoe/checkpoint-13786/generation_config.json +0 -12
- sft/1M3/Full_remoe/checkpoint-13786/latest +0 -1
- sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1412_llava...image_llava_model_args_595126/gpt_response/hallusion_output_vd_model.json +0 -0
- sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1412_llava...image_llava_model_args_595126/gpt_response/hallusion_output_vs_model.json +0 -0
- sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1412_llava...image_llava_model_args_595126/hallusion_bench_image.json +0 -0
- sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1412_llava...image_llava_model_args_595126/mathvista_testmini.json +0 -0
- sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1412_llava...image_llava_model_args_595126/rank0_metric_eval_done.txt +0 -1
- sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1412_llava...image_llava_model_args_595126/results.json +0 -146
- sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1412_llava...image_llava_model_args_595126/submissions/mathvista_testmini_scores.json +0 -0
- sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/ai2d.json +0 -0
- sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/infovqa_val.json +0 -3
- sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/mmbench_en_dev.json +0 -3
- sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/mme.json +0 -0
- sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/mmerealworld_lite.json +0 -3
- sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/mmmu_val.json +0 -0
- sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/mmstar.json +0 -0
- sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/rank0_metric_eval_done.txt +0 -1
- sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/rank1_metric_eval_done.txt +0 -1
- sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/rank2_metric_eval_done.txt +0 -1
- sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/results.json +0 -683
- sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/scienceqa_img.json +0 -0
- sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/seedbench_2_plus.json +0 -0
- sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/submissions/mmbench_en_dev_results.json +0 -1
- sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/submissions/mmbench_en_dev_results.xlsx +0 -3
- 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
- sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1428_llava...l_mme_llava_model_args_595126/textvqa_val.json +0 -3
- sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1459_llava...bench_llava_model_args_595126/ocrbench.json +0 -0
- sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1459_llava...bench_llava_model_args_595126/rank0_metric_eval_done.txt +0 -1
- sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1459_llava...bench_llava_model_args_595126/rank1_metric_eval_done.txt +0 -1
- sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1459_llava...bench_llava_model_args_595126/rank2_metric_eval_done.txt +0 -1
- sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1459_llava...bench_llava_model_args_595126/results.json +0 -67
- sft/1M3/Full_remoe/checkpoint-13786/logs/0627_1459_llava...bench_llava_model_args_595126/results/ocrbench_results.txt +0 -18
- sft/1M3/Full_remoe/checkpoint-13786/model-00001-of-00003.safetensors +0 -3
- sft/1M3/Full_remoe/checkpoint-13786/model-00002-of-00003.safetensors +0 -3
- sft/1M3/Full_remoe/checkpoint-13786/model-00003-of-00003.safetensors +0 -3
- sft/1M3/Full_remoe/checkpoint-13786/model.safetensors.index.json +0 -0
- sft/1M3/Full_remoe/checkpoint-13786/rng_state_0.pth +0 -3
- sft/1M3/Full_remoe/checkpoint-13786/rng_state_1.pth +0 -3
- sft/1M3/Full_remoe/checkpoint-13786/rng_state_2.pth +0 -3
- sft/1M3/Full_remoe/checkpoint-13786/rng_state_3.pth +0 -3
- sft/1M3/Full_remoe/checkpoint-13786/special_tokens_map.json +0 -24
- sft/1M3/Full_remoe/checkpoint-13786/tokenizer.model +0 -3
- sft/1M3/Full_remoe/checkpoint-13786/tokenizer_config.json +0 -132
- sft/1M3/Full_remoe/checkpoint-13786/trainer_state.json +0 -3
- sft/1M3/Full_remoe/checkpoint-13786/training_args.bin +0 -3
- sft/1M3/Full_remoe/checkpoint-13786/zero_to_fp32.py +0 -604
- sft/1M3/Full_remoe/checkpoint-20679/added_tokens.json +0 -13
- 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 |
-
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|