shenmishajing commited on
Commit
5d32c24
·
1 Parent(s): 58a08b7

upload model

Browse files
.gitattributes CHANGED
@@ -33,3 +33,5 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ training_args.bin filter=lfs diff=lfs merge=lfs -text
37
+ model.safetensors filter=lfs diff=lfs merge=lfs -text
added_tokens.json ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "</box>": 92552,
3
+ "</img>": 92545,
4
+ "</quad>": 92548,
5
+ "</ref>": 92550,
6
+ "<IMG_CONTEXT>": 92546,
7
+ "<box>": 92551,
8
+ "<img>": 92544,
9
+ "<quad>": 92547,
10
+ "<ref>": 92549
11
+ }
config.json ADDED
@@ -0,0 +1,143 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_commit_hash": null,
3
+ "architectures": [
4
+ "InternVLChatModel"
5
+ ],
6
+ "auto_map": {
7
+ "AutoConfig": "configuration_internvl_chat.InternVLChatConfig",
8
+ "AutoModel": "modeling_internvl_chat.InternVLChatModel",
9
+ "AutoModelForCausalLM": "modeling_internvl_chat.InternVLChatModel"
10
+ },
11
+ "downsample_ratio": 0.5,
12
+ "dynamic_image_size": true,
13
+ "force_image_size": 448,
14
+ "llm_config": {
15
+ "_name_or_path": "internlm/internlm2-chat-1_8b",
16
+ "add_cross_attention": false,
17
+ "architectures": [
18
+ "InternLM2ForCausalLM"
19
+ ],
20
+ "attn_implementation": "flash_attention_2",
21
+ "auto_map": {
22
+ "AutoConfig": "configuration_internlm2.InternLM2Config",
23
+ "AutoModel": "modeling_internlm2.InternLM2ForCausalLM",
24
+ "AutoModelForCausalLM": "modeling_internlm2.InternLM2ForCausalLM"
25
+ },
26
+ "bad_words_ids": null,
27
+ "begin_suppress_tokens": null,
28
+ "bias": false,
29
+ "bos_token_id": 1,
30
+ "chunk_size_feed_forward": 0,
31
+ "cross_attention_hidden_size": null,
32
+ "decoder_start_token_id": null,
33
+ "diversity_penalty": 0.0,
34
+ "do_sample": false,
35
+ "early_stopping": false,
36
+ "encoder_no_repeat_ngram_size": 0,
37
+ "eos_token_id": 2,
38
+ "exponential_decay_length_penalty": null,
39
+ "finetuning_task": null,
40
+ "forced_bos_token_id": null,
41
+ "forced_eos_token_id": null,
42
+ "hidden_act": "silu",
43
+ "hidden_size": 2048,
44
+ "id2label": {
45
+ "0": "LABEL_0",
46
+ "1": "LABEL_1"
47
+ },
48
+ "initializer_range": 0.02,
49
+ "intermediate_size": 8192,
50
+ "is_decoder": false,
51
+ "is_encoder_decoder": false,
52
+ "label2id": {
53
+ "LABEL_0": 0,
54
+ "LABEL_1": 1
55
+ },
56
+ "length_penalty": 1.0,
57
+ "max_length": 20,
58
+ "max_position_embeddings": 32768,
59
+ "min_length": 0,
60
+ "model_type": "internlm2",
61
+ "no_repeat_ngram_size": 0,
62
+ "num_attention_heads": 16,
63
+ "num_beam_groups": 1,
64
+ "num_beams": 1,
65
+ "num_hidden_layers": 24,
66
+ "num_key_value_heads": 8,
67
+ "num_return_sequences": 1,
68
+ "output_attentions": false,
69
+ "output_hidden_states": false,
70
+ "output_scores": false,
71
+ "pad_token_id": 2,
72
+ "prefix": null,
73
+ "problem_type": null,
74
+ "pruned_heads": {},
75
+ "remove_invalid_values": false,
76
+ "repetition_penalty": 1.0,
77
+ "return_dict": true,
78
+ "return_dict_in_generate": false,
79
+ "rms_norm_eps": 1e-05,
80
+ "rope_scaling": {
81
+ "factor": 2.0,
82
+ "type": "dynamic"
83
+ },
84
+ "rope_theta": 1000000,
85
+ "sep_token_id": null,
86
+ "suppress_tokens": null,
87
+ "task_specific_params": null,
88
+ "temperature": 1.0,
89
+ "tf_legacy_loss": false,
90
+ "tie_encoder_decoder": false,
91
+ "tie_word_embeddings": false,
92
+ "tokenizer_class": null,
93
+ "top_k": 50,
94
+ "top_p": 1.0,
95
+ "torch_dtype": "bfloat16",
96
+ "torchscript": false,
97
+ "transformers_version": "4.37.2",
98
+ "typical_p": 1.0,
99
+ "use_bfloat16": true,
100
+ "use_cache": true,
101
+ "vocab_size": 92553
102
+ },
103
+ "max_dynamic_patch": 12,
104
+ "min_dynamic_patch": 1,
105
+ "model_type": "internvl_chat",
106
+ "ps_version": "v2",
107
+ "select_layer": -1,
108
+ "template": "internlm2-chat",
109
+ "torch_dtype": "bfloat16",
110
+ "use_backbone_lora": 0,
111
+ "use_llm_lora": 0,
112
+ "use_thumbnail": true,
113
+ "vision_config": {
114
+ "architectures": [
115
+ "InternVisionModel"
116
+ ],
117
+ "attention_dropout": 0.0,
118
+ "drop_path_rate": 0.0,
119
+ "dropout": 0.0,
120
+ "hidden_act": "gelu",
121
+ "hidden_size": 1024,
122
+ "image_size": 448,
123
+ "initializer_factor": 1.0,
124
+ "initializer_range": 0.02,
125
+ "intermediate_size": 4096,
126
+ "layer_norm_eps": 1e-06,
127
+ "model_type": "intern_vit_6b",
128
+ "norm_type": "layer_norm",
129
+ "num_attention_heads": 16,
130
+ "num_channels": 3,
131
+ "num_hidden_layers": 24,
132
+ "output_attentions": false,
133
+ "output_hidden_states": false,
134
+ "patch_size": 14,
135
+ "qk_normalization": false,
136
+ "qkv_bias": true,
137
+ "return_dict": true,
138
+ "torch_dtype": "bfloat16",
139
+ "transformers_version": "4.37.2",
140
+ "use_bfloat16": true,
141
+ "use_flash_attn": true
142
+ }
143
+ }
generation_config.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_from_model_config": true,
3
+ "transformers_version": "4.37.2",
4
+ "eos_token_id": [
5
+ 92542,
6
+ 92543
7
+ ]
8
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5825fc69d95f626b0462b80f1d05bb169184d2b874c6e691dda733ab0e312689
3
+ size 4428549526
optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:45bbed20ff493b5ee4da34cb088b89344db020e8e21a28664437c7aa529a4fd8
3
+ size 6832499370
preprocessor_config.json ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "crop_size": 448,
3
+ "do_center_crop": true,
4
+ "do_normalize": true,
5
+ "do_resize": true,
6
+ "feature_extractor_type": "CLIPFeatureExtractor",
7
+ "image_mean": [
8
+ 0.485,
9
+ 0.456,
10
+ 0.406
11
+ ],
12
+ "image_std": [
13
+ 0.229,
14
+ 0.224,
15
+ 0.225
16
+ ],
17
+ "resample": 3,
18
+ "size": 448
19
+ }
scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:29e5cdd2694bbb0dfbe9b312be1fa3c6fe622689e3083fb2cc0d2bd79d9542a9
3
+ size 1064
special_tokens_map.json ADDED
@@ -0,0 +1,47 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "additional_special_tokens": [
3
+ "<|im_start|>",
4
+ "<|im_end|>",
5
+ "<|action_start|>",
6
+ "<|action_end|>",
7
+ "<|interpreter|>",
8
+ "<|plugin|>",
9
+ "<img>",
10
+ "</img>",
11
+ "<IMG_CONTEXT>",
12
+ "<quad>",
13
+ "</quad>",
14
+ "<ref>",
15
+ "</ref>",
16
+ "<box>",
17
+ "</box>"
18
+ ],
19
+ "bos_token": {
20
+ "content": "<s>",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false
25
+ },
26
+ "eos_token": {
27
+ "content": "</s>",
28
+ "lstrip": false,
29
+ "normalized": false,
30
+ "rstrip": false,
31
+ "single_word": false
32
+ },
33
+ "pad_token": {
34
+ "content": "</s>",
35
+ "lstrip": false,
36
+ "normalized": false,
37
+ "rstrip": false,
38
+ "single_word": false
39
+ },
40
+ "unk_token": {
41
+ "content": "<unk>",
42
+ "lstrip": false,
43
+ "normalized": false,
44
+ "rstrip": false,
45
+ "single_word": false
46
+ }
47
+ }
tokenizer_config.json ADDED
@@ -0,0 +1,179 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "added_tokens_decoder": {
3
+ "0": {
4
+ "content": "<unk>",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false,
9
+ "special": true
10
+ },
11
+ "1": {
12
+ "content": "<s>",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false,
17
+ "special": true
18
+ },
19
+ "2": {
20
+ "content": "</s>",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false,
25
+ "special": true
26
+ },
27
+ "92538": {
28
+ "content": "<|plugin|>",
29
+ "lstrip": false,
30
+ "normalized": false,
31
+ "rstrip": false,
32
+ "single_word": false,
33
+ "special": true
34
+ },
35
+ "92539": {
36
+ "content": "<|interpreter|>",
37
+ "lstrip": false,
38
+ "normalized": false,
39
+ "rstrip": false,
40
+ "single_word": false,
41
+ "special": true
42
+ },
43
+ "92540": {
44
+ "content": "<|action_end|>",
45
+ "lstrip": false,
46
+ "normalized": false,
47
+ "rstrip": false,
48
+ "single_word": false,
49
+ "special": true
50
+ },
51
+ "92541": {
52
+ "content": "<|action_start|>",
53
+ "lstrip": false,
54
+ "normalized": false,
55
+ "rstrip": false,
56
+ "single_word": false,
57
+ "special": true
58
+ },
59
+ "92542": {
60
+ "content": "<|im_end|>",
61
+ "lstrip": false,
62
+ "normalized": false,
63
+ "rstrip": false,
64
+ "single_word": false,
65
+ "special": true
66
+ },
67
+ "92543": {
68
+ "content": "<|im_start|>",
69
+ "lstrip": false,
70
+ "normalized": false,
71
+ "rstrip": false,
72
+ "single_word": false,
73
+ "special": true
74
+ },
75
+ "92544": {
76
+ "content": "<img>",
77
+ "lstrip": false,
78
+ "normalized": false,
79
+ "rstrip": false,
80
+ "single_word": false,
81
+ "special": true
82
+ },
83
+ "92545": {
84
+ "content": "</img>",
85
+ "lstrip": false,
86
+ "normalized": false,
87
+ "rstrip": false,
88
+ "single_word": false,
89
+ "special": true
90
+ },
91
+ "92546": {
92
+ "content": "<IMG_CONTEXT>",
93
+ "lstrip": false,
94
+ "normalized": false,
95
+ "rstrip": false,
96
+ "single_word": false,
97
+ "special": true
98
+ },
99
+ "92547": {
100
+ "content": "<quad>",
101
+ "lstrip": false,
102
+ "normalized": false,
103
+ "rstrip": false,
104
+ "single_word": false,
105
+ "special": true
106
+ },
107
+ "92548": {
108
+ "content": "</quad>",
109
+ "lstrip": false,
110
+ "normalized": false,
111
+ "rstrip": false,
112
+ "single_word": false,
113
+ "special": true
114
+ },
115
+ "92549": {
116
+ "content": "<ref>",
117
+ "lstrip": false,
118
+ "normalized": false,
119
+ "rstrip": false,
120
+ "single_word": false,
121
+ "special": true
122
+ },
123
+ "92550": {
124
+ "content": "</ref>",
125
+ "lstrip": false,
126
+ "normalized": false,
127
+ "rstrip": false,
128
+ "single_word": false,
129
+ "special": true
130
+ },
131
+ "92551": {
132
+ "content": "<box>",
133
+ "lstrip": false,
134
+ "normalized": false,
135
+ "rstrip": false,
136
+ "single_word": false,
137
+ "special": true
138
+ },
139
+ "92552": {
140
+ "content": "</box>",
141
+ "lstrip": false,
142
+ "normalized": false,
143
+ "rstrip": false,
144
+ "single_word": false,
145
+ "special": true
146
+ }
147
+ },
148
+ "additional_special_tokens": [
149
+ "<|im_start|>",
150
+ "<|im_end|>",
151
+ "<|action_start|>",
152
+ "<|action_end|>",
153
+ "<|interpreter|>",
154
+ "<|plugin|>",
155
+ "<img>",
156
+ "</img>",
157
+ "<IMG_CONTEXT>",
158
+ "<quad>",
159
+ "</quad>",
160
+ "<ref>",
161
+ "</ref>",
162
+ "<box>",
163
+ "</box>"
164
+ ],
165
+ "auto_map": {
166
+ "AutoTokenizer": [
167
+ "tokenization_internlm2.InternLM2Tokenizer",
168
+ null
169
+ ]
170
+ },
171
+ "bos_token": "<s>",
172
+ "chat_template": "{{ bos_token }}{% for message in messages %}{{'<|im_start|>' + message['role'] + '\n' + message['content'] + '<|im_end|>' + '\n'}}{% endfor %}{% if add_generation_prompt %}{{ '<|im_start|>assistant\n' }}{% endif %}",
173
+ "clean_up_tokenization_spaces": false,
174
+ "eos_token": "</s>",
175
+ "model_max_length": 8192,
176
+ "pad_token": "</s>",
177
+ "tokenizer_class": "InternLM2Tokenizer",
178
+ "unk_token": "<unk>"
179
+ }
trainer_state.json ADDED
@@ -0,0 +1,131 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_metric": null,
3
+ "best_model_checkpoint": null,
4
+ "epoch": 2.830188679245283,
5
+ "eval_steps": 1,
6
+ "global_step": 39,
7
+ "is_hyper_param_search": false,
8
+ "is_local_process_zero": true,
9
+ "is_world_process_zero": true,
10
+ "log_history": [
11
+ {
12
+ "epoch": 0.07547169811320754,
13
+ "grad_norm": 111.0,
14
+ "learning_rate": 6e-06,
15
+ "loss": 1.6151,
16
+ "step": 1
17
+ },
18
+ {
19
+ "epoch": 0.22641509433962265,
20
+ "grad_norm": 65.0,
21
+ "learning_rate": 1.8e-05,
22
+ "loss": 1.5894,
23
+ "step": 3
24
+ },
25
+ {
26
+ "epoch": 0.4528301886792453,
27
+ "grad_norm": 286.0,
28
+ "learning_rate": 2.9936012644425518e-05,
29
+ "loss": 1.9202,
30
+ "step": 6
31
+ },
32
+ {
33
+ "epoch": 0.6792452830188679,
34
+ "grad_norm": 175.0,
35
+ "learning_rate": 2.8987083441065335e-05,
36
+ "loss": 1.5979,
37
+ "step": 9
38
+ },
39
+ {
40
+ "epoch": 0.9056603773584906,
41
+ "grad_norm": 89.5,
42
+ "learning_rate": 2.6970258409203596e-05,
43
+ "loss": 1.4915,
44
+ "step": 12
45
+ },
46
+ {
47
+ "epoch": 1.0754716981132075,
48
+ "grad_norm": 128.0,
49
+ "learning_rate": 2.4039519545688848e-05,
50
+ "loss": 1.0238,
51
+ "step": 15
52
+ },
53
+ {
54
+ "epoch": 1.3018867924528301,
55
+ "grad_norm": 96.5,
56
+ "learning_rate": 2.0418624992807297e-05,
57
+ "loss": 1.2875,
58
+ "step": 18
59
+ },
60
+ {
61
+ "epoch": 1.5283018867924527,
62
+ "grad_norm": 122.0,
63
+ "learning_rate": 1.638402539194953e-05,
64
+ "loss": 1.1833,
65
+ "step": 21
66
+ },
67
+ {
68
+ "epoch": 1.7547169811320755,
69
+ "grad_norm": 87.5,
70
+ "learning_rate": 1.2243757232751444e-05,
71
+ "loss": 1.1228,
72
+ "step": 24
73
+ },
74
+ {
75
+ "epoch": 1.9811320754716981,
76
+ "grad_norm": 67.0,
77
+ "learning_rate": 8.313924663351927e-06,
78
+ "loss": 1.1144,
79
+ "step": 27
80
+ },
81
+ {
82
+ "epoch": 2.150943396226415,
83
+ "grad_norm": 100.5,
84
+ "learning_rate": 4.894565345301642e-06,
85
+ "loss": 0.7711,
86
+ "step": 30
87
+ },
88
+ {
89
+ "epoch": 2.3773584905660377,
90
+ "grad_norm": 60.25,
91
+ "learning_rate": 2.2467429640557903e-06,
92
+ "loss": 1.0037,
93
+ "step": 33
94
+ },
95
+ {
96
+ "epoch": 2.6037735849056602,
97
+ "grad_norm": 67.5,
98
+ "learning_rate": 5.726153524077144e-07,
99
+ "loss": 1.0549,
100
+ "step": 36
101
+ },
102
+ {
103
+ "epoch": 2.830188679245283,
104
+ "grad_norm": 50.75,
105
+ "learning_rate": 0.0,
106
+ "loss": 1.0016,
107
+ "step": 39
108
+ }
109
+ ],
110
+ "logging_steps": 3,
111
+ "max_steps": 39,
112
+ "num_input_tokens_seen": 0,
113
+ "num_train_epochs": 3,
114
+ "save_steps": 500,
115
+ "stateful_callbacks": {
116
+ "TrainerControl": {
117
+ "args": {
118
+ "should_epoch_stop": false,
119
+ "should_evaluate": true,
120
+ "should_log": false,
121
+ "should_save": true,
122
+ "should_training_stop": true
123
+ },
124
+ "attributes": {}
125
+ }
126
+ },
127
+ "total_flos": 0.0,
128
+ "train_batch_size": 1,
129
+ "trial_name": null,
130
+ "trial_params": null
131
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8571b2549806a355ade2e01bfe8b3afd9faa28beee154a72c054876ea38fb024
3
+ size 5304