alwaysgood commited on
Commit
a74656f
·
verified ·
1 Parent(s): 9064868

Upload final model from parallel-cpt-gemma4-st2

Browse files
.gitattributes CHANGED
@@ -33,3 +33,6 @@ 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
+ checkpoint-90/tokenizer.json filter=lfs diff=lfs merge=lfs -text
37
+ tokenizer/tokenizer.json filter=lfs diff=lfs merge=lfs -text
38
+ tokenizer.json filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,59 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: alwaysgood/gemma4-st1
3
+ library_name: transformers
4
+ model_name: checkpoints
5
+ tags:
6
+ - generated_from_trainer
7
+ - unsloth
8
+ - sft
9
+ - trl
10
+ licence: license
11
+ ---
12
+
13
+ # Model Card for checkpoints
14
+
15
+ This model is a fine-tuned version of [alwaysgood/gemma4-st1](https://huggingface.co/alwaysgood/gemma4-st1).
16
+ It has been trained using [TRL](https://github.com/huggingface/trl).
17
+
18
+ ## Quick start
19
+
20
+ ```python
21
+ from transformers import pipeline
22
+
23
+ question = "If you had a time machine, but could only go to the past or the future once and never return, which would you choose and why?"
24
+ generator = pipeline("text-generation", model="None", device="cuda")
25
+ output = generator([{"role": "user", "content": question}], max_new_tokens=128, return_full_text=False)[0]
26
+ print(output["generated_text"])
27
+ ```
28
+
29
+ ## Training procedure
30
+
31
+ [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="150" height="24"/>](https://wandb.ai/hiloong/parallel-cpt/runs/scywgwy0)
32
+
33
+
34
+ This model was trained with SFT.
35
+
36
+ ### Framework versions
37
+
38
+ - TRL: 0.24.0
39
+ - Transformers: 5.5.4
40
+ - Pytorch: 2.9.0+cu128
41
+ - Datasets: 4.3.0
42
+ - Tokenizers: 0.22.2
43
+
44
+ ## Citations
45
+
46
+
47
+
48
+ Cite TRL as:
49
+
50
+ ```bibtex
51
+ @misc{vonwerra2022trl,
52
+ title = {{TRL: Transformer Reinforcement Learning}},
53
+ author = {Leandro von Werra and Younes Belkada and Lewis Tunstall and Edward Beeching and Tristan Thrush and Nathan Lambert and Shengyi Huang and Kashif Rasul and Quentin Gallou{\'e}dec},
54
+ year = 2020,
55
+ journal = {GitHub repository},
56
+ publisher = {GitHub},
57
+ howpublished = {\url{https://github.com/huggingface/trl}}
58
+ }
59
+ ```
all_results.json ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 1.0,
3
+ "eval_loss": 1.9723843336105347,
4
+ "eval_runtime": 2.0925,
5
+ "eval_samples_per_second": 34.408,
6
+ "eval_steps_per_second": 4.301,
7
+ "total_flos": 9299379063137280.0,
8
+ "train_loss": 2.1708432303534613,
9
+ "train_runtime": 176.9852,
10
+ "train_samples_per_second": 20.132,
11
+ "train_steps_per_second": 0.509
12
+ }
checkpoint-90/config.json ADDED
@@ -0,0 +1,193 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "Gemma4ForConditionalGeneration"
4
+ ],
5
+ "audio_config": {
6
+ "_name_or_path": "",
7
+ "architectures": null,
8
+ "attention_chunk_size": 12,
9
+ "attention_context_left": 13,
10
+ "attention_context_right": 0,
11
+ "attention_invalid_logits_value": -1000000000.0,
12
+ "attention_logit_cap": 50.0,
13
+ "chunk_size_feed_forward": 0,
14
+ "conv_kernel_size": 5,
15
+ "dtype": "bfloat16",
16
+ "gradient_clipping": 10000000000.0,
17
+ "hidden_act": "silu",
18
+ "hidden_size": 1024,
19
+ "id2label": {
20
+ "0": "LABEL_0",
21
+ "1": "LABEL_1"
22
+ },
23
+ "initializer_range": 0.02,
24
+ "is_encoder_decoder": false,
25
+ "label2id": {
26
+ "LABEL_0": 0,
27
+ "LABEL_1": 1
28
+ },
29
+ "model_type": "gemma4_audio",
30
+ "num_attention_heads": 8,
31
+ "num_hidden_layers": 12,
32
+ "output_attentions": false,
33
+ "output_hidden_states": false,
34
+ "output_proj_dims": 1536,
35
+ "problem_type": null,
36
+ "residual_weight": 0.5,
37
+ "return_dict": true,
38
+ "rms_norm_eps": 1e-06,
39
+ "subsampling_conv_channels": [
40
+ 128,
41
+ 32
42
+ ],
43
+ "use_clipped_linears": true
44
+ },
45
+ "audio_token_id": 258881,
46
+ "boa_token_id": 256000,
47
+ "boi_token_id": 255999,
48
+ "bos_token_id": 2,
49
+ "dtype": "bfloat16",
50
+ "eoa_token_id": 258883,
51
+ "eoa_token_index": 258883,
52
+ "eoi_token_id": 258882,
53
+ "eos_token_id": 1,
54
+ "image_token_id": 258880,
55
+ "initializer_range": 0.02,
56
+ "model_name": "alwaysgood/gemma4-st1",
57
+ "model_type": "gemma4",
58
+ "pad_token_id": 0,
59
+ "text_config": {
60
+ "attention_bias": false,
61
+ "attention_dropout": 0.0,
62
+ "attention_k_eq_v": false,
63
+ "bos_token_id": 2,
64
+ "dtype": "bfloat16",
65
+ "enable_moe_block": false,
66
+ "eos_token_id": 1,
67
+ "expert_intermediate_size": null,
68
+ "final_logit_softcapping": 30.0,
69
+ "global_head_dim": 512,
70
+ "head_dim": 256,
71
+ "hidden_activation": "gelu_pytorch_tanh",
72
+ "hidden_size": 1536,
73
+ "hidden_size_per_layer_input": 256,
74
+ "initializer_range": 0.02,
75
+ "intermediate_size": 6144,
76
+ "layer_types": [
77
+ "sliding_attention",
78
+ "sliding_attention",
79
+ "sliding_attention",
80
+ "sliding_attention",
81
+ "full_attention",
82
+ "sliding_attention",
83
+ "sliding_attention",
84
+ "sliding_attention",
85
+ "sliding_attention",
86
+ "full_attention",
87
+ "sliding_attention",
88
+ "sliding_attention",
89
+ "sliding_attention",
90
+ "sliding_attention",
91
+ "full_attention",
92
+ "sliding_attention",
93
+ "sliding_attention",
94
+ "sliding_attention",
95
+ "sliding_attention",
96
+ "full_attention",
97
+ "sliding_attention",
98
+ "sliding_attention",
99
+ "sliding_attention",
100
+ "sliding_attention",
101
+ "full_attention",
102
+ "sliding_attention",
103
+ "sliding_attention",
104
+ "sliding_attention",
105
+ "sliding_attention",
106
+ "full_attention",
107
+ "sliding_attention",
108
+ "sliding_attention",
109
+ "sliding_attention",
110
+ "sliding_attention",
111
+ "full_attention"
112
+ ],
113
+ "max_position_embeddings": 131072,
114
+ "model_type": "gemma4_text",
115
+ "moe_intermediate_size": null,
116
+ "num_attention_heads": 8,
117
+ "num_experts": null,
118
+ "num_global_key_value_heads": null,
119
+ "num_hidden_layers": 35,
120
+ "num_key_value_heads": 1,
121
+ "num_kv_shared_layers": 20,
122
+ "pad_token_id": 0,
123
+ "rms_norm_eps": 1e-06,
124
+ "rope_parameters": {
125
+ "full_attention": {
126
+ "partial_rotary_factor": 0.25,
127
+ "rope_theta": 1000000.0,
128
+ "rope_type": "proportional"
129
+ },
130
+ "sliding_attention": {
131
+ "rope_theta": 10000.0,
132
+ "rope_type": "default"
133
+ }
134
+ },
135
+ "sliding_window": 512,
136
+ "tie_word_embeddings": true,
137
+ "top_k_experts": null,
138
+ "use_bidirectional_attention": null,
139
+ "use_cache": true,
140
+ "use_double_wide_mlp": true,
141
+ "vocab_size": 262144,
142
+ "vocab_size_per_layer_input": 262144
143
+ },
144
+ "tie_word_embeddings": true,
145
+ "transformers_version": "5.5.4",
146
+ "unsloth_version": "2026.4.6",
147
+ "use_cache": false,
148
+ "video_token_id": 258884,
149
+ "vision_config": {
150
+ "_name_or_path": "",
151
+ "architectures": null,
152
+ "attention_bias": false,
153
+ "attention_dropout": 0.0,
154
+ "chunk_size_feed_forward": 0,
155
+ "default_output_length": 280,
156
+ "dtype": "bfloat16",
157
+ "global_head_dim": 64,
158
+ "head_dim": 64,
159
+ "hidden_activation": "gelu_pytorch_tanh",
160
+ "hidden_size": 768,
161
+ "id2label": {
162
+ "0": "LABEL_0",
163
+ "1": "LABEL_1"
164
+ },
165
+ "initializer_range": 0.02,
166
+ "intermediate_size": 3072,
167
+ "is_encoder_decoder": false,
168
+ "label2id": {
169
+ "LABEL_0": 0,
170
+ "LABEL_1": 1
171
+ },
172
+ "max_position_embeddings": 131072,
173
+ "model_type": "gemma4_vision",
174
+ "num_attention_heads": 12,
175
+ "num_hidden_layers": 16,
176
+ "num_key_value_heads": 12,
177
+ "output_attentions": false,
178
+ "output_hidden_states": false,
179
+ "patch_size": 16,
180
+ "pooling_kernel_size": 3,
181
+ "position_embedding_size": 10240,
182
+ "problem_type": null,
183
+ "return_dict": true,
184
+ "rms_norm_eps": 1e-06,
185
+ "rope_parameters": {
186
+ "rope_theta": 100.0,
187
+ "rope_type": "default"
188
+ },
189
+ "standardize": false,
190
+ "use_clipped_linears": true
191
+ },
192
+ "vision_soft_tokens_per_image": 280
193
+ }
checkpoint-90/generation_config.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token_id": 2,
3
+ "do_sample": true,
4
+ "eos_token_id": [
5
+ 1,
6
+ 1
7
+ ],
8
+ "pad_token_id": 0,
9
+ "temperature": 1.0,
10
+ "top_k": 64,
11
+ "top_p": 0.95,
12
+ "transformers_version": "5.5.4"
13
+ }
checkpoint-90/model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:edc6c3c7c734788c6d1bf85ae8a039adfea87a1aa161f182fcc16ae02d5fd74a
3
+ size 10208852910
checkpoint-90/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6b63f5d323c46a35c1b7847ac05a2dffaf288efcac41870a6a132499f5c8ec16
3
+ size 18514716902
checkpoint-90/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:098b29492211804ab324a36f37466821d948280bb74fce4ba895c03f13ecd878
3
+ size 14645
checkpoint-90/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bd4da37e1d2639c6a3c48eabc833153fc8f9bf81ee46a2bc40f3b4e879988db4
3
+ size 1465
checkpoint-90/tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cc8d3a0ce36466ccc1278bf987df5f71db1719b9ca6b4118264f45cb627bfe0f
3
+ size 32169626
checkpoint-90/tokenizer_config.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "audio_token": "<|audio|>",
3
+ "backend": "tokenizers",
4
+ "boa_token": "<|audio>",
5
+ "boi_token": "<|image>",
6
+ "bos_token": "<bos>",
7
+ "eoa_token": "<audio|>",
8
+ "eoc_token": "<channel|>",
9
+ "eoi_token": "<image|>",
10
+ "eos_token": "<eos>",
11
+ "eot_token": "<turn|>",
12
+ "escape_token": "<|\"|>",
13
+ "etc_token": "<tool_call|>",
14
+ "etd_token": "<tool|>",
15
+ "etr_token": "<tool_response|>",
16
+ "extra_special_tokens": [],
17
+ "image_token": "<|image|>",
18
+ "is_local": false,
19
+ "mask_token": "<mask>",
20
+ "model_max_length": 1000000000000000019884624838656,
21
+ "model_specific_special_tokens": {
22
+ "audio_token": "<|audio|>",
23
+ "boa_token": "<|audio>",
24
+ "boi_token": "<|image>",
25
+ "eoa_token": "<audio|>",
26
+ "eoc_token": "<channel|>",
27
+ "eoi_token": "<image|>",
28
+ "eot_token": "<turn|>",
29
+ "escape_token": "<|\"|>",
30
+ "etc_token": "<tool_call|>",
31
+ "etd_token": "<tool|>",
32
+ "etr_token": "<tool_response|>",
33
+ "image_token": "<|image|>",
34
+ "soc_token": "<|channel>",
35
+ "sot_token": "<|turn>",
36
+ "stc_token": "<|tool_call>",
37
+ "std_token": "<|tool>",
38
+ "str_token": "<|tool_response>",
39
+ "think_token": "<|think|>"
40
+ },
41
+ "pad_token": "<pad>",
42
+ "padding_side": "right",
43
+ "processor_class": "Gemma4Processor",
44
+ "soc_token": "<|channel>",
45
+ "sot_token": "<|turn>",
46
+ "stc_token": "<|tool_call>",
47
+ "std_token": "<|tool>",
48
+ "str_token": "<|tool_response>",
49
+ "think_token": "<|think|>",
50
+ "tokenizer_class": "GemmaTokenizer",
51
+ "unk_token": "<unk>"
52
+ }
checkpoint-90/trainer_state.json ADDED
@@ -0,0 +1,105 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 1.0,
6
+ "eval_steps": 500,
7
+ "global_step": 90,
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.11204481792717087,
14
+ "grad_norm": 40.25,
15
+ "learning_rate": 1e-05,
16
+ "loss": 2.609230613708496,
17
+ "step": 10
18
+ },
19
+ {
20
+ "epoch": 0.22408963585434175,
21
+ "grad_norm": 14.0625,
22
+ "learning_rate": 9.628619846344453e-06,
23
+ "loss": 2.171901512145996,
24
+ "step": 20
25
+ },
26
+ {
27
+ "epoch": 0.33613445378151263,
28
+ "grad_norm": 13.375,
29
+ "learning_rate": 8.569648672789496e-06,
30
+ "loss": 2.157339096069336,
31
+ "step": 30
32
+ },
33
+ {
34
+ "epoch": 0.4481792717086835,
35
+ "grad_norm": 37.0,
36
+ "learning_rate": 6.980398830195785e-06,
37
+ "loss": 2.139277458190918,
38
+ "step": 40
39
+ },
40
+ {
41
+ "epoch": 0.5602240896358543,
42
+ "grad_norm": 11.375,
43
+ "learning_rate": 5.096956658859122e-06,
44
+ "loss": 2.022007942199707,
45
+ "step": 50
46
+ },
47
+ {
48
+ "epoch": 0.6722689075630253,
49
+ "grad_norm": 11.5625,
50
+ "learning_rate": 3.1991113759764493e-06,
51
+ "loss": 2.084817314147949,
52
+ "step": 60
53
+ },
54
+ {
55
+ "epoch": 0.7843137254901961,
56
+ "grad_norm": 11.875,
57
+ "learning_rate": 1.5687918106563326e-06,
58
+ "loss": 2.074056053161621,
59
+ "step": 70
60
+ },
61
+ {
62
+ "epoch": 0.896358543417367,
63
+ "grad_norm": 12.5,
64
+ "learning_rate": 4.481852951692672e-07,
65
+ "loss": 2.1064525604248048,
66
+ "step": 80
67
+ },
68
+ {
69
+ "epoch": 1.0,
70
+ "grad_norm": 61.5,
71
+ "learning_rate": 3.760237478849793e-09,
72
+ "loss": 2.1725065231323244,
73
+ "step": 90
74
+ },
75
+ {
76
+ "epoch": 1.0,
77
+ "eval_loss": 1.973793625831604,
78
+ "eval_runtime": 3.3988,
79
+ "eval_samples_per_second": 21.184,
80
+ "eval_steps_per_second": 2.648,
81
+ "step": 90
82
+ }
83
+ ],
84
+ "logging_steps": 10,
85
+ "max_steps": 90,
86
+ "num_input_tokens_seen": 0,
87
+ "num_train_epochs": 1,
88
+ "save_steps": 200,
89
+ "stateful_callbacks": {
90
+ "TrainerControl": {
91
+ "args": {
92
+ "should_epoch_stop": false,
93
+ "should_evaluate": false,
94
+ "should_log": false,
95
+ "should_save": true,
96
+ "should_training_stop": true
97
+ },
98
+ "attributes": {}
99
+ }
100
+ },
101
+ "total_flos": 9299379063137280.0,
102
+ "train_batch_size": 10,
103
+ "trial_name": null,
104
+ "trial_params": null
105
+ }
checkpoint-90/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a29a74de87fb454420a9766a2719c1e8814f044926f712ba08991aa47c2b5a59
3
+ size 5777
config.json ADDED
@@ -0,0 +1,193 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "Gemma4ForConditionalGeneration"
4
+ ],
5
+ "audio_config": {
6
+ "_name_or_path": "",
7
+ "architectures": null,
8
+ "attention_chunk_size": 12,
9
+ "attention_context_left": 13,
10
+ "attention_context_right": 0,
11
+ "attention_invalid_logits_value": -1000000000.0,
12
+ "attention_logit_cap": 50.0,
13
+ "chunk_size_feed_forward": 0,
14
+ "conv_kernel_size": 5,
15
+ "dtype": "bfloat16",
16
+ "gradient_clipping": 10000000000.0,
17
+ "hidden_act": "silu",
18
+ "hidden_size": 1024,
19
+ "id2label": {
20
+ "0": "LABEL_0",
21
+ "1": "LABEL_1"
22
+ },
23
+ "initializer_range": 0.02,
24
+ "is_encoder_decoder": false,
25
+ "label2id": {
26
+ "LABEL_0": 0,
27
+ "LABEL_1": 1
28
+ },
29
+ "model_type": "gemma4_audio",
30
+ "num_attention_heads": 8,
31
+ "num_hidden_layers": 12,
32
+ "output_attentions": false,
33
+ "output_hidden_states": false,
34
+ "output_proj_dims": 1536,
35
+ "problem_type": null,
36
+ "residual_weight": 0.5,
37
+ "return_dict": true,
38
+ "rms_norm_eps": 1e-06,
39
+ "subsampling_conv_channels": [
40
+ 128,
41
+ 32
42
+ ],
43
+ "use_clipped_linears": true
44
+ },
45
+ "audio_token_id": 258881,
46
+ "boa_token_id": 256000,
47
+ "boi_token_id": 255999,
48
+ "bos_token_id": 2,
49
+ "dtype": "bfloat16",
50
+ "eoa_token_id": 258883,
51
+ "eoa_token_index": 258883,
52
+ "eoi_token_id": 258882,
53
+ "eos_token_id": 1,
54
+ "image_token_id": 258880,
55
+ "initializer_range": 0.02,
56
+ "model_name": "alwaysgood/gemma4-st1",
57
+ "model_type": "gemma4",
58
+ "pad_token_id": 0,
59
+ "text_config": {
60
+ "attention_bias": false,
61
+ "attention_dropout": 0.0,
62
+ "attention_k_eq_v": false,
63
+ "bos_token_id": 2,
64
+ "dtype": "bfloat16",
65
+ "enable_moe_block": false,
66
+ "eos_token_id": 1,
67
+ "expert_intermediate_size": null,
68
+ "final_logit_softcapping": 30.0,
69
+ "global_head_dim": 512,
70
+ "head_dim": 256,
71
+ "hidden_activation": "gelu_pytorch_tanh",
72
+ "hidden_size": 1536,
73
+ "hidden_size_per_layer_input": 256,
74
+ "initializer_range": 0.02,
75
+ "intermediate_size": 6144,
76
+ "layer_types": [
77
+ "sliding_attention",
78
+ "sliding_attention",
79
+ "sliding_attention",
80
+ "sliding_attention",
81
+ "full_attention",
82
+ "sliding_attention",
83
+ "sliding_attention",
84
+ "sliding_attention",
85
+ "sliding_attention",
86
+ "full_attention",
87
+ "sliding_attention",
88
+ "sliding_attention",
89
+ "sliding_attention",
90
+ "sliding_attention",
91
+ "full_attention",
92
+ "sliding_attention",
93
+ "sliding_attention",
94
+ "sliding_attention",
95
+ "sliding_attention",
96
+ "full_attention",
97
+ "sliding_attention",
98
+ "sliding_attention",
99
+ "sliding_attention",
100
+ "sliding_attention",
101
+ "full_attention",
102
+ "sliding_attention",
103
+ "sliding_attention",
104
+ "sliding_attention",
105
+ "sliding_attention",
106
+ "full_attention",
107
+ "sliding_attention",
108
+ "sliding_attention",
109
+ "sliding_attention",
110
+ "sliding_attention",
111
+ "full_attention"
112
+ ],
113
+ "max_position_embeddings": 131072,
114
+ "model_type": "gemma4_text",
115
+ "moe_intermediate_size": null,
116
+ "num_attention_heads": 8,
117
+ "num_experts": null,
118
+ "num_global_key_value_heads": null,
119
+ "num_hidden_layers": 35,
120
+ "num_key_value_heads": 1,
121
+ "num_kv_shared_layers": 20,
122
+ "pad_token_id": 0,
123
+ "rms_norm_eps": 1e-06,
124
+ "rope_parameters": {
125
+ "full_attention": {
126
+ "partial_rotary_factor": 0.25,
127
+ "rope_theta": 1000000.0,
128
+ "rope_type": "proportional"
129
+ },
130
+ "sliding_attention": {
131
+ "rope_theta": 10000.0,
132
+ "rope_type": "default"
133
+ }
134
+ },
135
+ "sliding_window": 512,
136
+ "tie_word_embeddings": true,
137
+ "top_k_experts": null,
138
+ "use_bidirectional_attention": null,
139
+ "use_cache": true,
140
+ "use_double_wide_mlp": true,
141
+ "vocab_size": 262144,
142
+ "vocab_size_per_layer_input": 262144
143
+ },
144
+ "tie_word_embeddings": true,
145
+ "transformers_version": "5.5.4",
146
+ "unsloth_version": "2026.4.6",
147
+ "use_cache": false,
148
+ "video_token_id": 258884,
149
+ "vision_config": {
150
+ "_name_or_path": "",
151
+ "architectures": null,
152
+ "attention_bias": false,
153
+ "attention_dropout": 0.0,
154
+ "chunk_size_feed_forward": 0,
155
+ "default_output_length": 280,
156
+ "dtype": "bfloat16",
157
+ "global_head_dim": 64,
158
+ "head_dim": 64,
159
+ "hidden_activation": "gelu_pytorch_tanh",
160
+ "hidden_size": 768,
161
+ "id2label": {
162
+ "0": "LABEL_0",
163
+ "1": "LABEL_1"
164
+ },
165
+ "initializer_range": 0.02,
166
+ "intermediate_size": 3072,
167
+ "is_encoder_decoder": false,
168
+ "label2id": {
169
+ "LABEL_0": 0,
170
+ "LABEL_1": 1
171
+ },
172
+ "max_position_embeddings": 131072,
173
+ "model_type": "gemma4_vision",
174
+ "num_attention_heads": 12,
175
+ "num_hidden_layers": 16,
176
+ "num_key_value_heads": 12,
177
+ "output_attentions": false,
178
+ "output_hidden_states": false,
179
+ "patch_size": 16,
180
+ "pooling_kernel_size": 3,
181
+ "position_embedding_size": 10240,
182
+ "problem_type": null,
183
+ "return_dict": true,
184
+ "rms_norm_eps": 1e-06,
185
+ "rope_parameters": {
186
+ "rope_theta": 100.0,
187
+ "rope_type": "default"
188
+ },
189
+ "standardize": false,
190
+ "use_clipped_linears": true
191
+ },
192
+ "vision_soft_tokens_per_image": 280
193
+ }
eval_results.json ADDED
@@ -0,0 +1,7 @@
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 1.0,
3
+ "eval_loss": 1.9723843336105347,
4
+ "eval_runtime": 2.0925,
5
+ "eval_samples_per_second": 34.408,
6
+ "eval_steps_per_second": 4.301
7
+ }
generation_config.json ADDED
@@ -0,0 +1,13 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token_id": 2,
3
+ "do_sample": true,
4
+ "eos_token_id": [
5
+ 1,
6
+ 1
7
+ ],
8
+ "pad_token_id": 0,
9
+ "temperature": 1.0,
10
+ "top_k": 64,
11
+ "top_p": 0.95,
12
+ "transformers_version": "5.5.4"
13
+ }
model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:edc6c3c7c734788c6d1bf85ae8a039adfea87a1aa161f182fcc16ae02d5fd74a
3
+ size 10208852910
tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cc8d3a0ce36466ccc1278bf987df5f71db1719b9ca6b4118264f45cb627bfe0f
3
+ size 32169626
tokenizer/tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cc8d3a0ce36466ccc1278bf987df5f71db1719b9ca6b4118264f45cb627bfe0f
3
+ size 32169626
tokenizer/tokenizer_config.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "audio_token": "<|audio|>",
3
+ "backend": "tokenizers",
4
+ "boa_token": "<|audio>",
5
+ "boi_token": "<|image>",
6
+ "bos_token": "<bos>",
7
+ "eoa_token": "<audio|>",
8
+ "eoc_token": "<channel|>",
9
+ "eoi_token": "<image|>",
10
+ "eos_token": "<eos>",
11
+ "eot_token": "<turn|>",
12
+ "escape_token": "<|\"|>",
13
+ "etc_token": "<tool_call|>",
14
+ "etd_token": "<tool|>",
15
+ "etr_token": "<tool_response|>",
16
+ "extra_special_tokens": [],
17
+ "image_token": "<|image|>",
18
+ "is_local": false,
19
+ "mask_token": "<mask>",
20
+ "model_max_length": 1000000000000000019884624838656,
21
+ "model_specific_special_tokens": {
22
+ "audio_token": "<|audio|>",
23
+ "boa_token": "<|audio>",
24
+ "boi_token": "<|image>",
25
+ "eoa_token": "<audio|>",
26
+ "eoc_token": "<channel|>",
27
+ "eoi_token": "<image|>",
28
+ "eot_token": "<turn|>",
29
+ "escape_token": "<|\"|>",
30
+ "etc_token": "<tool_call|>",
31
+ "etd_token": "<tool|>",
32
+ "etr_token": "<tool_response|>",
33
+ "image_token": "<|image|>",
34
+ "soc_token": "<|channel>",
35
+ "sot_token": "<|turn>",
36
+ "stc_token": "<|tool_call>",
37
+ "std_token": "<|tool>",
38
+ "str_token": "<|tool_response>",
39
+ "think_token": "<|think|>"
40
+ },
41
+ "pad_token": "<pad>",
42
+ "padding_side": "left",
43
+ "processor_class": "Gemma4Processor",
44
+ "soc_token": "<|channel>",
45
+ "sot_token": "<|turn>",
46
+ "stc_token": "<|tool_call>",
47
+ "std_token": "<|tool>",
48
+ "str_token": "<|tool_response>",
49
+ "think_token": "<|think|>",
50
+ "tokenizer_class": "GemmaTokenizer",
51
+ "unk_token": "<unk>"
52
+ }
tokenizer_config.json ADDED
@@ -0,0 +1,52 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "audio_token": "<|audio|>",
3
+ "backend": "tokenizers",
4
+ "boa_token": "<|audio>",
5
+ "boi_token": "<|image>",
6
+ "bos_token": "<bos>",
7
+ "eoa_token": "<audio|>",
8
+ "eoc_token": "<channel|>",
9
+ "eoi_token": "<image|>",
10
+ "eos_token": "<eos>",
11
+ "eot_token": "<turn|>",
12
+ "escape_token": "<|\"|>",
13
+ "etc_token": "<tool_call|>",
14
+ "etd_token": "<tool|>",
15
+ "etr_token": "<tool_response|>",
16
+ "extra_special_tokens": [],
17
+ "image_token": "<|image|>",
18
+ "is_local": false,
19
+ "mask_token": "<mask>",
20
+ "model_max_length": 1000000000000000019884624838656,
21
+ "model_specific_special_tokens": {
22
+ "audio_token": "<|audio|>",
23
+ "boa_token": "<|audio>",
24
+ "boi_token": "<|image>",
25
+ "eoa_token": "<audio|>",
26
+ "eoc_token": "<channel|>",
27
+ "eoi_token": "<image|>",
28
+ "eot_token": "<turn|>",
29
+ "escape_token": "<|\"|>",
30
+ "etc_token": "<tool_call|>",
31
+ "etd_token": "<tool|>",
32
+ "etr_token": "<tool_response|>",
33
+ "image_token": "<|image|>",
34
+ "soc_token": "<|channel>",
35
+ "sot_token": "<|turn>",
36
+ "stc_token": "<|tool_call>",
37
+ "std_token": "<|tool>",
38
+ "str_token": "<|tool_response>",
39
+ "think_token": "<|think|>"
40
+ },
41
+ "pad_token": "<pad>",
42
+ "padding_side": "left",
43
+ "processor_class": "Gemma4Processor",
44
+ "soc_token": "<|channel>",
45
+ "sot_token": "<|turn>",
46
+ "stc_token": "<|tool_call>",
47
+ "std_token": "<|tool>",
48
+ "str_token": "<|tool_response>",
49
+ "think_token": "<|think|>",
50
+ "tokenizer_class": "GemmaTokenizer",
51
+ "unk_token": "<unk>"
52
+ }
train_results.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 1.0,
3
+ "total_flos": 9299379063137280.0,
4
+ "train_loss": 2.1708432303534613,
5
+ "train_runtime": 176.9852,
6
+ "train_samples_per_second": 20.132,
7
+ "train_steps_per_second": 0.509
8
+ }
trainer_state.json ADDED
@@ -0,0 +1,114 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 1.0,
6
+ "eval_steps": 500,
7
+ "global_step": 90,
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.11204481792717087,
14
+ "grad_norm": 40.25,
15
+ "learning_rate": 1e-05,
16
+ "loss": 2.609230613708496,
17
+ "step": 10
18
+ },
19
+ {
20
+ "epoch": 0.22408963585434175,
21
+ "grad_norm": 14.0625,
22
+ "learning_rate": 9.628619846344453e-06,
23
+ "loss": 2.171901512145996,
24
+ "step": 20
25
+ },
26
+ {
27
+ "epoch": 0.33613445378151263,
28
+ "grad_norm": 13.375,
29
+ "learning_rate": 8.569648672789496e-06,
30
+ "loss": 2.157339096069336,
31
+ "step": 30
32
+ },
33
+ {
34
+ "epoch": 0.4481792717086835,
35
+ "grad_norm": 37.0,
36
+ "learning_rate": 6.980398830195785e-06,
37
+ "loss": 2.139277458190918,
38
+ "step": 40
39
+ },
40
+ {
41
+ "epoch": 0.5602240896358543,
42
+ "grad_norm": 11.375,
43
+ "learning_rate": 5.096956658859122e-06,
44
+ "loss": 2.022007942199707,
45
+ "step": 50
46
+ },
47
+ {
48
+ "epoch": 0.6722689075630253,
49
+ "grad_norm": 11.5625,
50
+ "learning_rate": 3.1991113759764493e-06,
51
+ "loss": 2.084817314147949,
52
+ "step": 60
53
+ },
54
+ {
55
+ "epoch": 0.7843137254901961,
56
+ "grad_norm": 11.875,
57
+ "learning_rate": 1.5687918106563326e-06,
58
+ "loss": 2.074056053161621,
59
+ "step": 70
60
+ },
61
+ {
62
+ "epoch": 0.896358543417367,
63
+ "grad_norm": 12.5,
64
+ "learning_rate": 4.481852951692672e-07,
65
+ "loss": 2.1064525604248048,
66
+ "step": 80
67
+ },
68
+ {
69
+ "epoch": 1.0,
70
+ "grad_norm": 61.5,
71
+ "learning_rate": 3.760237478849793e-09,
72
+ "loss": 2.1725065231323244,
73
+ "step": 90
74
+ },
75
+ {
76
+ "epoch": 1.0,
77
+ "eval_loss": 1.973793625831604,
78
+ "eval_runtime": 3.3988,
79
+ "eval_samples_per_second": 21.184,
80
+ "eval_steps_per_second": 2.648,
81
+ "step": 90
82
+ },
83
+ {
84
+ "epoch": 1.0,
85
+ "step": 90,
86
+ "total_flos": 9299379063137280.0,
87
+ "train_loss": 2.1708432303534613,
88
+ "train_runtime": 176.9852,
89
+ "train_samples_per_second": 20.132,
90
+ "train_steps_per_second": 0.509
91
+ }
92
+ ],
93
+ "logging_steps": 10,
94
+ "max_steps": 90,
95
+ "num_input_tokens_seen": 0,
96
+ "num_train_epochs": 1,
97
+ "save_steps": 200,
98
+ "stateful_callbacks": {
99
+ "TrainerControl": {
100
+ "args": {
101
+ "should_epoch_stop": false,
102
+ "should_evaluate": false,
103
+ "should_log": false,
104
+ "should_save": true,
105
+ "should_training_stop": true
106
+ },
107
+ "attributes": {}
108
+ }
109
+ },
110
+ "total_flos": 9299379063137280.0,
111
+ "train_batch_size": 10,
112
+ "trial_name": null,
113
+ "trial_params": null
114
+ }
training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a29a74de87fb454420a9766a2719c1e8814f044926f712ba08991aa47c2b5a59
3
+ size 5777