MatanBT commited on
Commit
0b913ee
·
verified ·
1 Parent(s): 8c5981d

Upload folder using huggingface_hub

Browse files
.gitattributes CHANGED
@@ -40,3 +40,4 @@ checkpoint-75/tokenizer.json filter=lfs diff=lfs merge=lfs -text
40
  checkpoint-125/tokenizer.json filter=lfs diff=lfs merge=lfs -text
41
  checkpoint-65/tokenizer.json filter=lfs diff=lfs merge=lfs -text
42
  checkpoint-250/tokenizer.json filter=lfs diff=lfs merge=lfs -text
 
 
40
  checkpoint-125/tokenizer.json filter=lfs diff=lfs merge=lfs -text
41
  checkpoint-65/tokenizer.json filter=lfs diff=lfs merge=lfs -text
42
  checkpoint-250/tokenizer.json filter=lfs diff=lfs merge=lfs -text
43
+ checkpoint-300/tokenizer.json filter=lfs diff=lfs merge=lfs -text
checkpoint-300/added_tokens.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ {
2
+ "<image_soft_token>": 262144
3
+ }
checkpoint-300/config.json ADDED
@@ -0,0 +1,64 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "_sliding_window_pattern": 6,
3
+ "architectures": [
4
+ "Gemma3ForCausalLM"
5
+ ],
6
+ "attention_bias": false,
7
+ "attention_dropout": 0.0,
8
+ "attn_logit_softcapping": null,
9
+ "bos_token_id": 2,
10
+ "cache_implementation": "hybrid",
11
+ "dtype": "bfloat16",
12
+ "eos_token_id": 1,
13
+ "final_logit_softcapping": null,
14
+ "head_dim": 256,
15
+ "hidden_activation": "gelu_pytorch_tanh",
16
+ "hidden_size": 1152,
17
+ "initializer_range": 0.02,
18
+ "intermediate_size": 6912,
19
+ "layer_types": [
20
+ "sliding_attention",
21
+ "sliding_attention",
22
+ "sliding_attention",
23
+ "sliding_attention",
24
+ "sliding_attention",
25
+ "full_attention",
26
+ "sliding_attention",
27
+ "sliding_attention",
28
+ "sliding_attention",
29
+ "sliding_attention",
30
+ "sliding_attention",
31
+ "full_attention",
32
+ "sliding_attention",
33
+ "sliding_attention",
34
+ "sliding_attention",
35
+ "sliding_attention",
36
+ "sliding_attention",
37
+ "full_attention",
38
+ "sliding_attention",
39
+ "sliding_attention",
40
+ "sliding_attention",
41
+ "sliding_attention",
42
+ "sliding_attention",
43
+ "full_attention",
44
+ "sliding_attention",
45
+ "sliding_attention"
46
+ ],
47
+ "max_position_embeddings": 32768,
48
+ "model_type": "gemma3_text",
49
+ "num_attention_heads": 4,
50
+ "num_hidden_layers": 26,
51
+ "num_key_value_heads": 1,
52
+ "pad_token_id": 0,
53
+ "query_pre_attn_scalar": 256,
54
+ "rms_norm_eps": 1e-06,
55
+ "rope_local_base_freq": 10000,
56
+ "rope_scaling": null,
57
+ "rope_theta": 1000000,
58
+ "sliding_window": 512,
59
+ "sliding_window_pattern": 6,
60
+ "transformers_version": "4.57.1",
61
+ "use_bidirectional_attention": false,
62
+ "use_cache": true,
63
+ "vocab_size": 262144
64
+ }
checkpoint-300/generation_config.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token_id": 2,
3
+ "cache_implementation": "hybrid",
4
+ "do_sample": true,
5
+ "eos_token_id": [
6
+ 1,
7
+ 106
8
+ ],
9
+ "pad_token_id": 0,
10
+ "top_k": 64,
11
+ "top_p": 0.95,
12
+ "transformers_version": "4.57.1"
13
+ }
checkpoint-300/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:34a639938f53f443c0dc92996c1fea7ea8618b2deeb5f92b5cf41d16551951f2
3
+ size 1999811208
checkpoint-300/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:001560c4be4c072b6dcd158fe795b79fc3772406c480917a864a3e70c8b2f674
3
+ size 3999834207
checkpoint-300/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1255b82ec9f462fef6f8c2f4a8ee876c4b6d998a53c99546a4eddcc1aabaefc7
3
+ size 14645
checkpoint-300/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6899f9f82cea064fe52ac09813b47d526ed686cbe04f0925eb7e665a0e61bfae
3
+ size 1465
checkpoint-300/special_tokens_map.json ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "boi_token": "<start_of_image>",
3
+ "bos_token": {
4
+ "content": "<bos>",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false
9
+ },
10
+ "eoi_token": "<end_of_image>",
11
+ "eos_token": {
12
+ "content": "<eos>",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false
17
+ },
18
+ "image_token": "<image_soft_token>",
19
+ "pad_token": "<eos>",
20
+ "unk_token": {
21
+ "content": "<unk>",
22
+ "lstrip": false,
23
+ "normalized": false,
24
+ "rstrip": false,
25
+ "single_word": false
26
+ }
27
+ }
checkpoint-300/tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:db5eba787198c9cb1d12d0dc4d653f46747749da078219b0b3f4a11b7487eb19
3
+ size 33384832
checkpoint-300/tokenizer.model ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1299c11d7cf632ef3b4e11937501358ada021bbdf7c47638d13c0ee982f2e79c
3
+ size 4689074
checkpoint-300/tokenizer_config.json ADDED
The diff for this file is too large to render. See raw diff
 
checkpoint-300/trainer_state.json ADDED
@@ -0,0 +1,244 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 12.0,
6
+ "eval_steps": 500,
7
+ "global_step": 300,
8
+ "is_hyper_param_search": false,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "epoch": 0.4,
14
+ "grad_norm": 10.75,
15
+ "learning_rate": 4.85e-05,
16
+ "loss": 2.3182,
17
+ "step": 10
18
+ },
19
+ {
20
+ "epoch": 0.8,
21
+ "grad_norm": 4.71875,
22
+ "learning_rate": 4.683333333333334e-05,
23
+ "loss": 1.674,
24
+ "step": 20
25
+ },
26
+ {
27
+ "epoch": 1.2,
28
+ "grad_norm": 5.90625,
29
+ "learning_rate": 4.516666666666667e-05,
30
+ "loss": 1.618,
31
+ "step": 30
32
+ },
33
+ {
34
+ "epoch": 1.6,
35
+ "grad_norm": 5.6875,
36
+ "learning_rate": 4.35e-05,
37
+ "loss": 1.5503,
38
+ "step": 40
39
+ },
40
+ {
41
+ "epoch": 2.0,
42
+ "grad_norm": 4.3125,
43
+ "learning_rate": 4.183333333333334e-05,
44
+ "loss": 1.5248,
45
+ "step": 50
46
+ },
47
+ {
48
+ "epoch": 2.4,
49
+ "grad_norm": 4.59375,
50
+ "learning_rate": 4.016666666666667e-05,
51
+ "loss": 1.3144,
52
+ "step": 60
53
+ },
54
+ {
55
+ "epoch": 2.8,
56
+ "grad_norm": 9.1875,
57
+ "learning_rate": 3.85e-05,
58
+ "loss": 1.2945,
59
+ "step": 70
60
+ },
61
+ {
62
+ "epoch": 3.2,
63
+ "grad_norm": 6.3125,
64
+ "learning_rate": 3.683333333333334e-05,
65
+ "loss": 1.2991,
66
+ "step": 80
67
+ },
68
+ {
69
+ "epoch": 3.6,
70
+ "grad_norm": 6.15625,
71
+ "learning_rate": 3.516666666666667e-05,
72
+ "loss": 1.052,
73
+ "step": 90
74
+ },
75
+ {
76
+ "epoch": 4.0,
77
+ "grad_norm": 8.375,
78
+ "learning_rate": 3.35e-05,
79
+ "loss": 1.1129,
80
+ "step": 100
81
+ },
82
+ {
83
+ "epoch": 4.4,
84
+ "grad_norm": 19.75,
85
+ "learning_rate": 3.183333333333334e-05,
86
+ "loss": 0.9219,
87
+ "step": 110
88
+ },
89
+ {
90
+ "epoch": 4.8,
91
+ "grad_norm": 8.375,
92
+ "learning_rate": 3.016666666666667e-05,
93
+ "loss": 0.7774,
94
+ "step": 120
95
+ },
96
+ {
97
+ "epoch": 5.2,
98
+ "grad_norm": 8.625,
99
+ "learning_rate": 2.8499999999999998e-05,
100
+ "loss": 0.7305,
101
+ "step": 130
102
+ },
103
+ {
104
+ "epoch": 5.6,
105
+ "grad_norm": 13.4375,
106
+ "learning_rate": 2.6833333333333333e-05,
107
+ "loss": 0.5266,
108
+ "step": 140
109
+ },
110
+ {
111
+ "epoch": 6.0,
112
+ "grad_norm": 6.0625,
113
+ "learning_rate": 2.5166666666666667e-05,
114
+ "loss": 0.6177,
115
+ "step": 150
116
+ },
117
+ {
118
+ "epoch": 6.4,
119
+ "grad_norm": 14.1875,
120
+ "learning_rate": 2.35e-05,
121
+ "loss": 0.3576,
122
+ "step": 160
123
+ },
124
+ {
125
+ "epoch": 6.8,
126
+ "grad_norm": 13.375,
127
+ "learning_rate": 2.1833333333333333e-05,
128
+ "loss": 0.3831,
129
+ "step": 170
130
+ },
131
+ {
132
+ "epoch": 7.2,
133
+ "grad_norm": 11.25,
134
+ "learning_rate": 2.0166666666666668e-05,
135
+ "loss": 0.2907,
136
+ "step": 180
137
+ },
138
+ {
139
+ "epoch": 7.6,
140
+ "grad_norm": 8.75,
141
+ "learning_rate": 1.85e-05,
142
+ "loss": 0.2107,
143
+ "step": 190
144
+ },
145
+ {
146
+ "epoch": 8.0,
147
+ "grad_norm": 13.375,
148
+ "learning_rate": 1.6833333333333334e-05,
149
+ "loss": 0.2369,
150
+ "step": 200
151
+ },
152
+ {
153
+ "epoch": 8.4,
154
+ "grad_norm": 5.84375,
155
+ "learning_rate": 1.5166666666666668e-05,
156
+ "loss": 0.1433,
157
+ "step": 210
158
+ },
159
+ {
160
+ "epoch": 8.8,
161
+ "grad_norm": 5.5,
162
+ "learning_rate": 1.3500000000000001e-05,
163
+ "loss": 0.159,
164
+ "step": 220
165
+ },
166
+ {
167
+ "epoch": 9.2,
168
+ "grad_norm": 3.3125,
169
+ "learning_rate": 1.1833333333333334e-05,
170
+ "loss": 0.1216,
171
+ "step": 230
172
+ },
173
+ {
174
+ "epoch": 9.6,
175
+ "grad_norm": 5.03125,
176
+ "learning_rate": 1.0166666666666667e-05,
177
+ "loss": 0.0976,
178
+ "step": 240
179
+ },
180
+ {
181
+ "epoch": 10.0,
182
+ "grad_norm": 3.9375,
183
+ "learning_rate": 8.500000000000002e-06,
184
+ "loss": 0.0994,
185
+ "step": 250
186
+ },
187
+ {
188
+ "epoch": 10.4,
189
+ "grad_norm": 3.140625,
190
+ "learning_rate": 6.833333333333333e-06,
191
+ "loss": 0.0757,
192
+ "step": 260
193
+ },
194
+ {
195
+ "epoch": 10.8,
196
+ "grad_norm": 6.3125,
197
+ "learning_rate": 5.166666666666667e-06,
198
+ "loss": 0.0823,
199
+ "step": 270
200
+ },
201
+ {
202
+ "epoch": 11.2,
203
+ "grad_norm": 1.9921875,
204
+ "learning_rate": 3.5000000000000004e-06,
205
+ "loss": 0.0683,
206
+ "step": 280
207
+ },
208
+ {
209
+ "epoch": 11.6,
210
+ "grad_norm": 2.234375,
211
+ "learning_rate": 1.8333333333333335e-06,
212
+ "loss": 0.0657,
213
+ "step": 290
214
+ },
215
+ {
216
+ "epoch": 12.0,
217
+ "grad_norm": 2.484375,
218
+ "learning_rate": 1.6666666666666668e-07,
219
+ "loss": 0.0702,
220
+ "step": 300
221
+ }
222
+ ],
223
+ "logging_steps": 10,
224
+ "max_steps": 300,
225
+ "num_input_tokens_seen": 0,
226
+ "num_train_epochs": 12,
227
+ "save_steps": 500,
228
+ "stateful_callbacks": {
229
+ "TrainerControl": {
230
+ "args": {
231
+ "should_epoch_stop": false,
232
+ "should_evaluate": false,
233
+ "should_log": false,
234
+ "should_save": true,
235
+ "should_training_stop": true
236
+ },
237
+ "attributes": {}
238
+ }
239
+ },
240
+ "total_flos": 643181012582400.0,
241
+ "train_batch_size": 4,
242
+ "trial_name": null,
243
+ "trial_params": null
244
+ }
checkpoint-300/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4742684a093fc361873457fae1211104c151d4d7203b326aafde35d86d727093
3
+ size 5777
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:51a67a16229c3a8e5242fae3d047b744f800596df5ae07b946778f74589cfbd0
3
  size 1999811208
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:34a639938f53f443c0dc92996c1fea7ea8618b2deeb5f92b5cf41d16551951f2
3
  size 1999811208
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:4f47d9dfeea297f70054819c45ec532a1a9cc3bfad2833427ed1061da98947ec
3
  size 5777
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4742684a093fc361873457fae1211104c151d4d7203b326aafde35d86d727093
3
  size 5777