leonardlin commited on
Commit
35d1e5e
·
verified ·
1 Parent(s): 493a3d8

Upload folder using huggingface_hub

Browse files
.gitattributes CHANGED
@@ -33,3 +33,4 @@ 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
+ tokenizer.json filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,278 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: transformers
3
+ license: gemma
4
+ base_model: google/gemma-3-27b-it
5
+ tags:
6
+ - generated_from_trainer
7
+ datasets:
8
+ - shisa-ai/shisa-v2-best-of-n-athenev2-tulu70b-llama33-only-no-sysprompt
9
+ - shisa-ai/shisa-v2-roleplaying-sft
10
+ - shisa-ai/translation_set_april_6
11
+ - shisa-ai/rewild-set-deepseek-subset
12
+ - shisa-ai/magpie-ultra-set
13
+ - shisa-ai/magpie-advanced-questions-set
14
+ - shisa-ai/japan-magpie-set
15
+ - shisa-ai/shisa-v2-instruction-following-sft
16
+ model-index:
17
+ - name: outputs/ablation-196-finalsft2-shisa-v2-gemma3-27b
18
+ results: []
19
+ ---
20
+
21
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
22
+ should probably proofread and complete it, then remove this comment. -->
23
+
24
+ [<img src="https://raw.githubusercontent.com/axolotl-ai-cloud/axolotl/main/image/axolotl-badge-web.png" alt="Built with Axolotl" width="200" height="32"/>](https://github.com/axolotl-ai-cloud/axolotl)
25
+ <details><summary>See axolotl config</summary>
26
+
27
+ axolotl version: `0.8.0.dev0`
28
+ ```yaml
29
+ # train w/ shisa-ai/shisa-v1-athenev2-reannotated-filtered
30
+
31
+ base_model: google/gemma-3-27b-it
32
+
33
+ load_in_8bit: false
34
+ load_in_4bit: false
35
+ strict: false
36
+
37
+ datasets:
38
+ - path: shisa-ai/shisa-v2-best-of-n-athenev2-tulu70b-llama33-only-no-sysprompt
39
+ type: chat_template
40
+ field_messages: conversations
41
+ message_field_role: from
42
+ message_field_content: value
43
+ - path: shisa-ai/shisa-v2-roleplaying-sft
44
+ type: chat_template
45
+ field_messages: conversations
46
+ message_property_mappings:
47
+ role: role
48
+ content: content
49
+ roles:
50
+ system:
51
+ - system
52
+ assistant:
53
+ - gpt
54
+ - model
55
+ - assistant
56
+ user:
57
+ - human
58
+ - user
59
+ roles_to_train: ["assistant"]
60
+ - path: shisa-ai/translation_set_april_6
61
+ split: train[:25%]
62
+ type: chat_template
63
+ field_messages: conversations
64
+ message_property_mappings:
65
+ role: role
66
+ content: content
67
+ roles:
68
+ system:
69
+ - system
70
+ assistant:
71
+ - gpt
72
+ - model
73
+ - assistant
74
+ user:
75
+ - human
76
+ - user
77
+ roles_to_train: ["assistant"]
78
+ - path: shisa-ai/rewild-set-deepseek-subset
79
+ split: train[:25%]
80
+ type: chat_template
81
+ field_messages: conversations
82
+ message_property_mappings:
83
+ role: role
84
+ content: content
85
+ roles:
86
+ system:
87
+ - system
88
+ assistant:
89
+ - gpt
90
+ - model
91
+ - assistant
92
+ user:
93
+ - human
94
+ - user
95
+ roles_to_train: ["assistant"]
96
+ - path: shisa-ai/magpie-ultra-set
97
+ split: train[:8%]
98
+ type: chat_template
99
+ field_messages: conversations
100
+ message_property_mappings:
101
+ role: role
102
+ content: content
103
+ roles:
104
+ system:
105
+ - system
106
+ assistant:
107
+ - gpt
108
+ - model
109
+ - assistant
110
+ user:
111
+ - human
112
+ - user
113
+ roles_to_train: ["assistant"]
114
+ - path: shisa-ai/magpie-advanced-questions-set
115
+ split: train[:8%]
116
+ type: chat_template
117
+ field_messages: conversations
118
+ message_property_mappings:
119
+ role: role
120
+ content: content
121
+ roles:
122
+ system:
123
+ - system
124
+ assistant:
125
+ - gpt
126
+ - model
127
+ - assistant
128
+ user:
129
+ - human
130
+ - user
131
+ roles_to_train: ["assistant"]
132
+ - path: shisa-ai/japan-magpie-set
133
+ split: train
134
+ type: chat_template
135
+ field_messages: conversations
136
+ message_property_mappings:
137
+ role: role
138
+ content: content
139
+ roles:
140
+ system:
141
+ - system
142
+ assistant:
143
+ - gpt
144
+ - model
145
+ - assistant
146
+ user:
147
+ - human
148
+ - user
149
+ roles_to_train: ["assistant"]
150
+ - path: shisa-ai/shisa-v2-instruction-following-sft
151
+ split: train[:50%]
152
+ type: chat_template
153
+ field_messages: conversations
154
+ message_property_mappings:
155
+ role: role
156
+ content: content
157
+ roles:
158
+ system:
159
+ - system
160
+ assistant:
161
+ - gpt
162
+ - model
163
+ - assistant
164
+ user:
165
+ - human
166
+ - user
167
+ roles_to_train: ["assistant"]
168
+
169
+ dataset_prepared_path: last_run_prepared
170
+ val_set_size: 0.05
171
+ output_dir: ./outputs/ablation-196-finalsft2-shisa-v2-gemma3-27b
172
+
173
+ sequence_len: 8192
174
+ sample_packing: true
175
+ pad_to_sequence_len: true
176
+
177
+ # marginal difference
178
+ neftune_noise_alpha: 5
179
+
180
+ use_wandb: true
181
+ wandb_project: shisa-v2
182
+ wandb_entity: augmxnt
183
+ wandb_name: ablation-196-finalsft2-shisa-v2-gemma3-27b
184
+
185
+ gradient_accumulation_steps: 2
186
+ micro_batch_size: 1
187
+ num_epochs: 3
188
+ optimizer: paged_adamw_8bit
189
+ lr_scheduler: linear
190
+ learning_rate: 5.4e-6
191
+
192
+ train_on_inputs: false
193
+ group_by_length: false
194
+ bf16: auto
195
+ fp16:
196
+ tf32: false
197
+
198
+ gradient_checkpointing: true
199
+ gradient_checkpointing_kwargs:
200
+ use_reentrant: false
201
+ early_stopping_patience:
202
+ resume_from_checkpoint:
203
+ logging_steps: 1
204
+ xformers_attention:
205
+ flash_attention: true
206
+
207
+ warmup_steps: 100
208
+ evals_per_epoch: 2
209
+ eval_table_size:
210
+ saves_per_epoch: 0
211
+ save_total_limit: 1 # Only store a single checkpoint
212
+ debug:
213
+ deepspeed: zero3_bf16.json
214
+ weight_decay: 1e-4
215
+ fsdp:
216
+ fsdp_config:
217
+ special_tokens:
218
+
219
+ ```
220
+
221
+ </details><br>
222
+
223
+ # outputs/ablation-196-finalsft2-shisa-v2-gemma3-27b
224
+
225
+ This model is a fine-tuned version of [google/gemma-3-27b-it](https://huggingface.co/google/gemma-3-27b-it) on the shisa-ai/shisa-v2-best-of-n-athenev2-tulu70b-llama33-only-no-sysprompt, the shisa-ai/shisa-v2-roleplaying-sft, the shisa-ai/translation_set_april_6, the shisa-ai/rewild-set-deepseek-subset, the shisa-ai/magpie-ultra-set, the shisa-ai/magpie-advanced-questions-set, the shisa-ai/japan-magpie-set and the shisa-ai/shisa-v2-instruction-following-sft datasets.
226
+ It achieves the following results on the evaluation set:
227
+ - Loss: 0.5417
228
+
229
+ ## Model description
230
+
231
+ More information needed
232
+
233
+ ## Intended uses & limitations
234
+
235
+ More information needed
236
+
237
+ ## Training and evaluation data
238
+
239
+ More information needed
240
+
241
+ ## Training procedure
242
+
243
+ ### Training hyperparameters
244
+
245
+ The following hyperparameters were used during training:
246
+ - learning_rate: 5.4e-06
247
+ - train_batch_size: 1
248
+ - eval_batch_size: 1
249
+ - seed: 42
250
+ - distributed_type: multi-GPU
251
+ - num_devices: 64
252
+ - gradient_accumulation_steps: 2
253
+ - total_train_batch_size: 128
254
+ - total_eval_batch_size: 64
255
+ - optimizer: Use OptimizerNames.PAGED_ADAMW_8BIT with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
256
+ - lr_scheduler_type: linear
257
+ - lr_scheduler_warmup_steps: 100
258
+ - num_epochs: 3.0
259
+
260
+ ### Training results
261
+
262
+ | Training Loss | Epoch | Step | Validation Loss |
263
+ |:-------------:|:------:|:----:|:---------------:|
264
+ | 11.3287 | 0.0027 | 1 | 5.4644 |
265
+ | 1.1409 | 0.4993 | 182 | 0.5644 |
266
+ | 1.0588 | 0.9986 | 364 | 0.5344 |
267
+ | 0.9207 | 1.4966 | 546 | 0.5322 |
268
+ | 0.8979 | 1.9959 | 728 | 0.5245 |
269
+ | 0.7673 | 2.4938 | 910 | 0.5432 |
270
+ | 0.7521 | 2.9931 | 1092 | 0.5417 |
271
+
272
+
273
+ ### Framework versions
274
+
275
+ - Transformers 4.50.0
276
+ - Pytorch 2.6.0+cu124
277
+ - Datasets 3.4.1
278
+ - Tokenizers 0.21.1
added_tokens.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ {
2
+ "<image_soft_token>": 262144
3
+ }
config.json ADDED
@@ -0,0 +1,60 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "Gemma3ForConditionalGeneration"
4
+ ],
5
+ "boi_token_index": 255999,
6
+ "eoi_token_index": 256000,
7
+ "eos_token_id": 1,
8
+ "image_token_index": 262144,
9
+ "initializer_range": 0.02,
10
+ "mm_tokens_per_image": 256,
11
+ "model_type": "gemma3",
12
+ "text_config": {
13
+ "attention_bias": false,
14
+ "attention_dropout": 0.0,
15
+ "attn_logit_softcapping": null,
16
+ "cache_implementation": "hybrid",
17
+ "final_logit_softcapping": null,
18
+ "head_dim": 128,
19
+ "hidden_activation": "gelu_pytorch_tanh",
20
+ "hidden_size": 5376,
21
+ "initializer_range": 0.02,
22
+ "intermediate_size": 21504,
23
+ "max_position_embeddings": 131072,
24
+ "model_type": "gemma3_text",
25
+ "num_attention_heads": 32,
26
+ "num_hidden_layers": 62,
27
+ "num_key_value_heads": 16,
28
+ "query_pre_attn_scalar": 168,
29
+ "rms_norm_eps": 1e-06,
30
+ "rope_local_base_freq": 10000.0,
31
+ "rope_scaling": {
32
+ "factor": 8.0,
33
+ "rope_type": "linear"
34
+ },
35
+ "rope_theta": 1000000.0,
36
+ "sliding_window": 1024,
37
+ "sliding_window_pattern": 6,
38
+ "torch_dtype": "bfloat16",
39
+ "use_cache": false,
40
+ "vocab_size": 262208
41
+ },
42
+ "torch_dtype": "bfloat16",
43
+ "transformers_version": "4.50.0",
44
+ "use_cache": false,
45
+ "vision_config": {
46
+ "attention_dropout": 0.0,
47
+ "hidden_act": "gelu_pytorch_tanh",
48
+ "hidden_size": 1152,
49
+ "image_size": 896,
50
+ "intermediate_size": 4304,
51
+ "layer_norm_eps": 1e-06,
52
+ "model_type": "siglip_vision_model",
53
+ "num_attention_heads": 16,
54
+ "num_channels": 3,
55
+ "num_hidden_layers": 27,
56
+ "patch_size": 14,
57
+ "torch_dtype": "bfloat16",
58
+ "vision_use_head": false
59
+ }
60
+ }
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.50.0"
13
+ }
model-00001-of-00012.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b7cfe65620decaff9d737059e37e8496249b11ecf5a80381506e516cef336cd9
3
+ size 4854573696
model-00002-of-00012.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e782dd22381a0675ffa4bedf91b121fb7af4f471c391cf5c80f769cadd35bfef
3
+ size 4954792944
model-00003-of-00012.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:edc1111415f73c3acc038d8f4a50cbdb180577b7762f0a2cd8f304ad3303093a
3
+ size 4954792976
model-00004-of-00012.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:eacbc025f831dad7c95da476be6e10d2db5b843295710db47771594221165356
3
+ size 4954793016
model-00005-of-00012.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f11f7c13d8f08e4c707e5ca5d5f103da0f70bde2e6953184f25b81729c937af7
3
+ size 4954793016
model-00006-of-00012.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:93336fdc39591d2e43fd1ee9d67180155f4b29028ee45629500d9d105d85c299
3
+ size 4954793016
model-00007-of-00012.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aba120fe6156ccd6f35db21283cf1b990848eaae13a5e0f21c3b9a2ae1e794d0
3
+ size 4954793016
model-00008-of-00012.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:53253af4d3cd151692386e7ed23012710a7fa00cdd7e6ebcf9c7b6a7db35ec23
3
+ size 4954793016
model-00009-of-00012.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:561ef3fbe7ece9bcf5b0c62043f2a65be529ebcb6ccc51ed77a3900a8f115d98
3
+ size 4954793016
model-00010-of-00012.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:067608f9cf13317a73fd04228f65aa4d393a9650f7feddcdd1dd8ad1c7221475
3
+ size 4954793016
model-00011-of-00012.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:73c47430843465d4f024d671ce885658bd6c56a4a9cacec36f7ab6179622ae53
3
+ size 4954793016
model-00012-of-00012.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b180613904855f75c07bb31c977ee6034c2f8bf499aedd1fa98ed51e22659385
3
+ size 462476696
model.safetensors.index.json ADDED
The diff for this file is too large to render. See raw diff
 
special_tokens_map.json ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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": {
20
+ "content": "<pad>",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false
25
+ },
26
+ "unk_token": {
27
+ "content": "<unk>",
28
+ "lstrip": false,
29
+ "normalized": false,
30
+ "rstrip": false,
31
+ "single_word": false
32
+ }
33
+ }
tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4667f2089529e8e7657cfb6d1c19910ae71ff5f28aa7ab2ff2763330affad795
3
+ size 33384568
tokenizer.model ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1299c11d7cf632ef3b4e11937501358ada021bbdf7c47638d13c0ee982f2e79c
3
+ size 4689074
tokenizer_config.json ADDED
The diff for this file is too large to render. See raw diff
 
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b7bf9420411a51edd2de797f377ab25c216d62b2e8d5b778e4679622cbf313c8
3
+ size 8632