prog-love commited on
Commit
2650c3e
·
verified ·
1 Parent(s): c7b9151

Upload folder using huggingface_hub

Browse files
This view is limited to 50 files because it contains too many changes.   See raw diff
Files changed (50) hide show
  1. README.md +53 -11
  2. adapter_config.json +51 -0
  3. adapter_model.safetensors +3 -0
  4. checkpoint-120/README.md +210 -0
  5. checkpoint-120/adapter_config.json +50 -0
  6. checkpoint-120/adapter_model.safetensors +3 -0
  7. checkpoint-120/chat_template.jinja +351 -0
  8. checkpoint-120/optimizer.pt +3 -0
  9. checkpoint-120/processor_config.json +75 -0
  10. checkpoint-120/rng_state.pth +3 -0
  11. checkpoint-120/scaler.pt +3 -0
  12. checkpoint-120/scheduler.pt +3 -0
  13. checkpoint-120/tokenizer.json +3 -0
  14. checkpoint-120/tokenizer_config.json +289 -0
  15. checkpoint-120/trainer_state.json +1234 -0
  16. checkpoint-120/training_args.bin +3 -0
  17. checkpoint-150/README.md +210 -0
  18. checkpoint-150/adapter_config.json +50 -0
  19. checkpoint-150/adapter_model.safetensors +3 -0
  20. checkpoint-150/chat_template.jinja +351 -0
  21. checkpoint-150/optimizer.pt +3 -0
  22. checkpoint-150/processor_config.json +75 -0
  23. checkpoint-150/rng_state.pth +3 -0
  24. checkpoint-150/scaler.pt +3 -0
  25. checkpoint-150/scheduler.pt +3 -0
  26. checkpoint-150/tokenizer.json +3 -0
  27. checkpoint-150/tokenizer_config.json +289 -0
  28. checkpoint-150/trainer_state.json +1534 -0
  29. checkpoint-150/training_args.bin +3 -0
  30. checkpoint-180/README.md +210 -0
  31. checkpoint-180/adapter_config.json +50 -0
  32. checkpoint-180/adapter_model.safetensors +3 -0
  33. checkpoint-180/chat_template.jinja +351 -0
  34. checkpoint-180/optimizer.pt +3 -0
  35. checkpoint-180/processor_config.json +75 -0
  36. checkpoint-180/rng_state.pth +3 -0
  37. checkpoint-180/scaler.pt +3 -0
  38. checkpoint-180/scheduler.pt +3 -0
  39. checkpoint-180/tokenizer.json +3 -0
  40. checkpoint-180/tokenizer_config.json +289 -0
  41. checkpoint-180/trainer_state.json +1834 -0
  42. checkpoint-180/training_args.bin +3 -0
  43. checkpoint-190/README.md +210 -0
  44. checkpoint-190/adapter_config.json +50 -0
  45. checkpoint-190/adapter_model.safetensors +3 -0
  46. checkpoint-190/chat_template.jinja +351 -0
  47. checkpoint-190/optimizer.pt +3 -0
  48. checkpoint-190/processor_config.json +75 -0
  49. checkpoint-190/rng_state.pth +3 -0
  50. checkpoint-190/scaler.pt +3 -0
README.md CHANGED
@@ -1,21 +1,63 @@
1
  ---
2
  base_model: unsloth/gemma-4-e2b-it-unsloth-bnb-4bit
 
 
3
  tags:
4
- - text-generation-inference
 
 
5
  - transformers
 
6
  - unsloth
7
- - gemma4
8
- license: apache-2.0
9
- language:
10
- - en
11
  ---
12
 
13
- # Uploaded finetuned model
14
 
15
- - **Developed by:** prog-love
16
- - **License:** apache-2.0
17
- - **Finetuned from model :** unsloth/gemma-4-e2b-it-unsloth-bnb-4bit
18
 
19
- This gemma4 model was trained 2x faster with [Unsloth](https://github.com/unslothai/unsloth) and Huggingface's TRL library.
20
 
21
- [<img src="https://raw.githubusercontent.com/unslothai/unsloth/main/images/unsloth%20made%20with%20love.png" width="200"/>](https://github.com/unslothai/unsloth)
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  base_model: unsloth/gemma-4-e2b-it-unsloth-bnb-4bit
3
+ library_name: peft
4
+ model_name: unsloth_gemma-4-E2B-it_1780218767
5
  tags:
6
+ - base_model:adapter:unsloth/gemma-4-e2b-it-unsloth-bnb-4bit
7
+ - lora
8
+ - sft
9
  - transformers
10
+ - trl
11
  - unsloth
12
+ licence: license
13
+ pipeline_tag: text-generation
 
 
14
  ---
15
 
16
+ # Model Card for unsloth_gemma-4-E2B-it_1780218767
17
 
18
+ This model is a fine-tuned version of [unsloth/gemma-4-e2b-it-unsloth-bnb-4bit](https://huggingface.co/unsloth/gemma-4-e2b-it-unsloth-bnb-4bit).
19
+ It has been trained using [TRL](https://github.com/huggingface/trl).
 
20
 
21
+ ## Quick start
22
 
23
+ ```python
24
+ from transformers import pipeline
25
+
26
+ 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?"
27
+ generator = pipeline("text-generation", model="None", device="cuda")
28
+ output = generator([{"role": "user", "content": question}], max_new_tokens=128, return_full_text=False)[0]
29
+ print(output["generated_text"])
30
+ ```
31
+
32
+ ## Training procedure
33
+
34
+
35
+
36
+
37
+ This model was trained with SFT.
38
+
39
+ ### Framework versions
40
+
41
+ - PEFT 0.18.1
42
+ - TRL: 0.23.1
43
+ - Transformers: 5.5.0
44
+ - Pytorch: 2.10.0+cu128
45
+ - Datasets: 4.3.0
46
+ - Tokenizers: 0.22.2
47
+
48
+ ## Citations
49
+
50
+
51
+
52
+ Cite TRL as:
53
+
54
+ ```bibtex
55
+ @misc{vonwerra2022trl,
56
+ title = {{TRL: Transformer Reinforcement Learning}},
57
+ 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},
58
+ year = 2020,
59
+ journal = {GitHub repository},
60
+ publisher = {GitHub},
61
+ howpublished = {\url{https://github.com/huggingface/trl}}
62
+ }
63
+ ```
adapter_config.json ADDED
@@ -0,0 +1,51 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alora_invocation_tokens": null,
3
+ "alpha_pattern": {},
4
+ "arrow_config": null,
5
+ "auto_mapping": {
6
+ "base_model_class": "Gemma4ForConditionalGeneration",
7
+ "parent_library": "transformers.models.gemma4.modeling_gemma4",
8
+ "unsloth_fixed": true
9
+ },
10
+ "base_model_name_or_path": "unsloth/gemma-4-e2b-it-unsloth-bnb-4bit",
11
+ "bias": "none",
12
+ "corda_config": null,
13
+ "ensure_weight_tying": false,
14
+ "eva_config": null,
15
+ "exclude_modules": null,
16
+ "fan_in_fan_out": false,
17
+ "inference_mode": true,
18
+ "init_lora_weights": true,
19
+ "layer_replication": null,
20
+ "layers_pattern": null,
21
+ "layers_to_transform": null,
22
+ "loftq_config": {},
23
+ "lora_alpha": 32,
24
+ "lora_bias": false,
25
+ "lora_dropout": 0.0,
26
+ "megatron_config": null,
27
+ "megatron_core": "megatron.core",
28
+ "modules_to_save": null,
29
+ "peft_type": "LORA",
30
+ "peft_version": "0.18.1",
31
+ "qalora_group_size": 16,
32
+ "r": 16,
33
+ "rank_pattern": {},
34
+ "revision": null,
35
+ "target_modules": [
36
+ "q_proj",
37
+ "gate_proj",
38
+ "v_proj",
39
+ "k_proj",
40
+ "down_proj",
41
+ "o_proj",
42
+ "up_proj"
43
+ ],
44
+ "target_parameters": null,
45
+ "task_type": "CAUSAL_LM",
46
+ "trainable_token_indices": null,
47
+ "use_dora": false,
48
+ "use_qalora": false,
49
+ "use_rslora": false,
50
+ "unsloth_training_method": "qlora"
51
+ }
adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dae7c101be06c818341a1b3354a5715c93ad40fc4c4700a81f0c71c0e699b82b
3
+ size 124277728
checkpoint-120/README.md ADDED
@@ -0,0 +1,210 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: unsloth/gemma-4-e2b-it-unsloth-bnb-4bit
3
+ library_name: peft
4
+ pipeline_tag: text-generation
5
+ tags:
6
+ - base_model:adapter:unsloth/gemma-4-e2b-it-unsloth-bnb-4bit
7
+ - lora
8
+ - sft
9
+ - transformers
10
+ - trl
11
+ - unsloth
12
+ ---
13
+
14
+ # Model Card for Model ID
15
+
16
+ <!-- Provide a quick summary of what the model is/does. -->
17
+
18
+
19
+
20
+ ## Model Details
21
+
22
+ ### Model Description
23
+
24
+ <!-- Provide a longer summary of what this model is. -->
25
+
26
+
27
+
28
+ - **Developed by:** [More Information Needed]
29
+ - **Funded by [optional]:** [More Information Needed]
30
+ - **Shared by [optional]:** [More Information Needed]
31
+ - **Model type:** [More Information Needed]
32
+ - **Language(s) (NLP):** [More Information Needed]
33
+ - **License:** [More Information Needed]
34
+ - **Finetuned from model [optional]:** [More Information Needed]
35
+
36
+ ### Model Sources [optional]
37
+
38
+ <!-- Provide the basic links for the model. -->
39
+
40
+ - **Repository:** [More Information Needed]
41
+ - **Paper [optional]:** [More Information Needed]
42
+ - **Demo [optional]:** [More Information Needed]
43
+
44
+ ## Uses
45
+
46
+ <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
47
+
48
+ ### Direct Use
49
+
50
+ <!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
51
+
52
+ [More Information Needed]
53
+
54
+ ### Downstream Use [optional]
55
+
56
+ <!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
57
+
58
+ [More Information Needed]
59
+
60
+ ### Out-of-Scope Use
61
+
62
+ <!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
63
+
64
+ [More Information Needed]
65
+
66
+ ## Bias, Risks, and Limitations
67
+
68
+ <!-- This section is meant to convey both technical and sociotechnical limitations. -->
69
+
70
+ [More Information Needed]
71
+
72
+ ### Recommendations
73
+
74
+ <!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
75
+
76
+ Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
77
+
78
+ ## How to Get Started with the Model
79
+
80
+ Use the code below to get started with the model.
81
+
82
+ [More Information Needed]
83
+
84
+ ## Training Details
85
+
86
+ ### Training Data
87
+
88
+ <!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
89
+
90
+ [More Information Needed]
91
+
92
+ ### Training Procedure
93
+
94
+ <!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
95
+
96
+ #### Preprocessing [optional]
97
+
98
+ [More Information Needed]
99
+
100
+
101
+ #### Training Hyperparameters
102
+
103
+ - **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
104
+
105
+ #### Speeds, Sizes, Times [optional]
106
+
107
+ <!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
108
+
109
+ [More Information Needed]
110
+
111
+ ## Evaluation
112
+
113
+ <!-- This section describes the evaluation protocols and provides the results. -->
114
+
115
+ ### Testing Data, Factors & Metrics
116
+
117
+ #### Testing Data
118
+
119
+ <!-- This should link to a Dataset Card if possible. -->
120
+
121
+ [More Information Needed]
122
+
123
+ #### Factors
124
+
125
+ <!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
126
+
127
+ [More Information Needed]
128
+
129
+ #### Metrics
130
+
131
+ <!-- These are the evaluation metrics being used, ideally with a description of why. -->
132
+
133
+ [More Information Needed]
134
+
135
+ ### Results
136
+
137
+ [More Information Needed]
138
+
139
+ #### Summary
140
+
141
+
142
+
143
+ ## Model Examination [optional]
144
+
145
+ <!-- Relevant interpretability work for the model goes here -->
146
+
147
+ [More Information Needed]
148
+
149
+ ## Environmental Impact
150
+
151
+ <!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
152
+
153
+ Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
154
+
155
+ - **Hardware Type:** [More Information Needed]
156
+ - **Hours used:** [More Information Needed]
157
+ - **Cloud Provider:** [More Information Needed]
158
+ - **Compute Region:** [More Information Needed]
159
+ - **Carbon Emitted:** [More Information Needed]
160
+
161
+ ## Technical Specifications [optional]
162
+
163
+ ### Model Architecture and Objective
164
+
165
+ [More Information Needed]
166
+
167
+ ### Compute Infrastructure
168
+
169
+ [More Information Needed]
170
+
171
+ #### Hardware
172
+
173
+ [More Information Needed]
174
+
175
+ #### Software
176
+
177
+ [More Information Needed]
178
+
179
+ ## Citation [optional]
180
+
181
+ <!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
182
+
183
+ **BibTeX:**
184
+
185
+ [More Information Needed]
186
+
187
+ **APA:**
188
+
189
+ [More Information Needed]
190
+
191
+ ## Glossary [optional]
192
+
193
+ <!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
194
+
195
+ [More Information Needed]
196
+
197
+ ## More Information [optional]
198
+
199
+ [More Information Needed]
200
+
201
+ ## Model Card Authors [optional]
202
+
203
+ [More Information Needed]
204
+
205
+ ## Model Card Contact
206
+
207
+ [More Information Needed]
208
+ ### Framework versions
209
+
210
+ - PEFT 0.18.1
checkpoint-120/adapter_config.json ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alora_invocation_tokens": null,
3
+ "alpha_pattern": {},
4
+ "arrow_config": null,
5
+ "auto_mapping": {
6
+ "base_model_class": "Gemma4ForConditionalGeneration",
7
+ "parent_library": "transformers.models.gemma4.modeling_gemma4",
8
+ "unsloth_fixed": true
9
+ },
10
+ "base_model_name_or_path": "unsloth/gemma-4-e2b-it-unsloth-bnb-4bit",
11
+ "bias": "none",
12
+ "corda_config": null,
13
+ "ensure_weight_tying": false,
14
+ "eva_config": null,
15
+ "exclude_modules": null,
16
+ "fan_in_fan_out": false,
17
+ "inference_mode": true,
18
+ "init_lora_weights": true,
19
+ "layer_replication": null,
20
+ "layers_pattern": null,
21
+ "layers_to_transform": null,
22
+ "loftq_config": {},
23
+ "lora_alpha": 32,
24
+ "lora_bias": false,
25
+ "lora_dropout": 0.0,
26
+ "megatron_config": null,
27
+ "megatron_core": "megatron.core",
28
+ "modules_to_save": null,
29
+ "peft_type": "LORA",
30
+ "peft_version": "0.18.1",
31
+ "qalora_group_size": 16,
32
+ "r": 16,
33
+ "rank_pattern": {},
34
+ "revision": null,
35
+ "target_modules": [
36
+ "q_proj",
37
+ "gate_proj",
38
+ "v_proj",
39
+ "k_proj",
40
+ "down_proj",
41
+ "o_proj",
42
+ "up_proj"
43
+ ],
44
+ "target_parameters": null,
45
+ "task_type": "CAUSAL_LM",
46
+ "trainable_token_indices": null,
47
+ "use_dora": false,
48
+ "use_qalora": false,
49
+ "use_rslora": false
50
+ }
checkpoint-120/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a2fa7f51110b2c95c35000062cb6a564c25910d5c3d47a3ab9455def2a9d031c
3
+ size 124277728
checkpoint-120/chat_template.jinja ADDED
@@ -0,0 +1,351 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {%- macro format_parameters(properties, required, filter_keys=false) -%}
2
+ {%- set standard_keys = ['description', 'type', 'properties', 'required', 'nullable'] -%}
3
+ {%- set ns = namespace(found_first=false) -%}
4
+ {%- for key, value in properties | dictsort -%}
5
+ {%- set add_comma = false -%}
6
+ {%- if not filter_keys or key not in standard_keys -%}
7
+ {%- if ns.found_first %},{% endif -%}
8
+ {%- set ns.found_first = true -%}
9
+ {{ key }}:{
10
+ {%- if value['description'] -%}
11
+ description:<|"|>{{ value['description'] }}<|"|>
12
+ {%- set add_comma = true -%}
13
+ {%- endif -%}
14
+ {%- if value['type'] | upper == 'STRING' -%}
15
+ {%- if value['enum'] -%}
16
+ {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
17
+ enum:{{ format_argument(value['enum']) }}
18
+ {%- endif -%}
19
+ {%- elif value['type'] | upper == 'ARRAY' -%}
20
+ {%- if value['items'] is mapping and value['items'] -%}
21
+ {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
22
+ items:{
23
+ {%- set ns_items = namespace(found_first=false) -%}
24
+ {%- for item_key, item_value in value['items'] | dictsort -%}
25
+ {%- if item_value is not none -%}
26
+ {%- if ns_items.found_first %},{% endif -%}
27
+ {%- set ns_items.found_first = true -%}
28
+ {%- if item_key == 'properties' -%}
29
+ properties:{
30
+ {%- if item_value is mapping -%}
31
+ {{- format_parameters(item_value, value['items']['required'] | default([])) -}}
32
+ {%- endif -%}
33
+ }
34
+ {%- elif item_key == 'required' -%}
35
+ required:[
36
+ {%- for req_item in item_value -%}
37
+ <|"|>{{- req_item -}}<|"|>
38
+ {%- if not loop.last %},{% endif -%}
39
+ {%- endfor -%}
40
+ ]
41
+ {%- elif item_key == 'type' -%}
42
+ {%- if item_value is string -%}
43
+ type:{{ format_argument(item_value | upper) }}
44
+ {%- else -%}
45
+ type:{{ format_argument(item_value | map('upper') | list) }}
46
+ {%- endif -%}
47
+ {%- else -%}
48
+ {{ item_key }}:{{ format_argument(item_value) }}
49
+ {%- endif -%}
50
+ {%- endif -%}
51
+ {%- endfor -%}
52
+ }
53
+ {%- endif -%}
54
+ {%- endif -%}
55
+ {%- if value['nullable'] %}
56
+ {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
57
+ nullable:true
58
+ {%- endif -%}
59
+ {%- if value['type'] | upper == 'OBJECT' -%}
60
+ {%- if value['properties'] is defined and value['properties'] is mapping -%}
61
+ {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
62
+ properties:{
63
+ {{- format_parameters(value['properties'], value['required'] | default([])) -}}
64
+ }
65
+ {%- elif value is mapping -%}
66
+ {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
67
+ properties:{
68
+ {{- format_parameters(value, value['required'] | default([]), filter_keys=true) -}}
69
+ }
70
+ {%- endif -%}
71
+ {%- if value['required'] -%}
72
+ {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
73
+ required:[
74
+ {%- for item in value['required'] | default([]) -%}
75
+ <|"|>{{- item -}}<|"|>
76
+ {%- if not loop.last %},{% endif -%}
77
+ {%- endfor -%}
78
+ ]
79
+ {%- endif -%}
80
+ {%- endif -%}
81
+ {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
82
+ type:<|"|>{{ value['type'] | upper }}<|"|>}
83
+ {%- endif -%}
84
+ {%- endfor -%}
85
+ {%- endmacro -%}
86
+ {%- macro format_function_declaration(tool_data) -%}
87
+ declaration:{{- tool_data['function']['name'] -}}{description:<|"|>{{- tool_data['function']['description'] -}}<|"|>
88
+ {%- set params = tool_data['function']['parameters'] -%}
89
+ {%- if params -%}
90
+ ,parameters:{
91
+ {%- if params['properties'] -%}
92
+ properties:{ {{- format_parameters(params['properties'], params['required']) -}} },
93
+ {%- endif -%}
94
+ {%- if params['required'] -%}
95
+ required:[
96
+ {%- for item in params['required'] -%}
97
+ <|"|>{{- item -}}<|"|>
98
+ {{- ',' if not loop.last -}}
99
+ {%- endfor -%}
100
+ ],
101
+ {%- endif -%}
102
+ {%- if params['type'] -%}
103
+ type:<|"|>{{- params['type'] | upper -}}<|"|>}
104
+ {%- endif -%}
105
+ {%- endif -%}
106
+ {%- if 'response' in tool_data['function'] -%}
107
+ {%- set response_declaration = tool_data['function']['response'] -%}
108
+ ,response:{
109
+ {%- if response_declaration['description'] -%}
110
+ description:<|"|>{{- response_declaration['description'] -}}<|"|>,
111
+ {%- endif -%}
112
+ {%- if response_declaration['type'] | upper == 'OBJECT' -%}
113
+ type:<|"|>{{- response_declaration['type'] | upper -}}<|"|>}
114
+ {%- endif -%}
115
+ {%- endif -%}
116
+ }
117
+ {%- endmacro -%}
118
+ {%- macro format_argument(argument, escape_keys=True) -%}
119
+ {%- if argument is string -%}
120
+ {{- '<|"|>' + argument + '<|"|>' -}}
121
+ {%- elif argument is boolean -%}
122
+ {{- 'true' if argument else 'false' -}}
123
+ {%- elif argument is mapping -%}
124
+ {{- '{' -}}
125
+ {%- set ns = namespace(found_first=false) -%}
126
+ {%- for key, value in argument | dictsort -%}
127
+ {%- if ns.found_first %},{% endif -%}
128
+ {%- set ns.found_first = true -%}
129
+ {%- if escape_keys -%}
130
+ {{- '<|"|>' + key + '<|"|>' -}}
131
+ {%- else -%}
132
+ {{- key -}}
133
+ {%- endif -%}
134
+ :{{- format_argument(value, escape_keys=escape_keys) -}}
135
+ {%- endfor -%}
136
+ {{- '}' -}}
137
+ {%- elif argument is sequence -%}
138
+ {{- '[' -}}
139
+ {%- for item in argument -%}
140
+ {{- format_argument(item, escape_keys=escape_keys) -}}
141
+ {%- if not loop.last %},{% endif -%}
142
+ {%- endfor -%}
143
+ {{- ']' -}}
144
+ {%- else -%}
145
+ {{- argument -}}
146
+ {%- endif -%}
147
+ {%- endmacro -%}
148
+ {%- macro strip_thinking(text) -%}
149
+ {%- set ns = namespace(result='') -%}
150
+ {%- for part in text.split('<channel|>') -%}
151
+ {%- if '<|channel>' in part -%}
152
+ {%- set ns.result = ns.result + part.split('<|channel>')[0] -%}
153
+ {%- else -%}
154
+ {%- set ns.result = ns.result + part -%}
155
+ {%- endif -%}
156
+ {%- endfor -%}
157
+ {{- ns.result | trim -}}
158
+ {%- endmacro -%}
159
+
160
+ {%- macro format_tool_response_block(tool_name, response) -%}
161
+ {{- '<|tool_response>' -}}
162
+ {%- if response is mapping -%}
163
+ {{- 'response:' + tool_name + '{' -}}
164
+ {%- for key, value in response | dictsort -%}
165
+ {{- key -}}:{{- format_argument(value, escape_keys=False) -}}
166
+ {%- if not loop.last %},{% endif -%}
167
+ {%- endfor -%}
168
+ {{- '}' -}}
169
+ {%- else -%}
170
+ {{- 'response:' + tool_name + '{value:' + format_argument(response, escape_keys=False) + '}' -}}
171
+ {%- endif -%}
172
+ {{- '<tool_response|>' -}}
173
+ {%- endmacro -%}
174
+
175
+ {%- set ns = namespace(prev_message_type=None) -%}
176
+ {%- set loop_messages = messages -%}
177
+ {{- bos_token -}}
178
+ {#- Handle System/Tool Definitions Block -#}
179
+ {%- if (enable_thinking is defined and enable_thinking) or tools or messages[0]['role'] in ['system', 'developer'] -%}
180
+ {{- '<|turn>system\n' -}}
181
+ {#- Inject Thinking token at the very top of the FIRST system turn -#}
182
+ {%- if enable_thinking is defined and enable_thinking -%}
183
+ {{- '<|think|>\n' -}}
184
+ {%- set ns.prev_message_type = 'think' -%}
185
+ {%- endif -%}
186
+ {%- if messages[0]['role'] in ['system', 'developer'] -%}
187
+ {%- if messages[0]['content'] is string -%}
188
+ {{- messages[0]['content'] | trim -}}
189
+ {%- elif messages[0]['content'] is sequence -%}
190
+ {%- for item in messages[0]['content'] -%}
191
+ {{- item['text'] | trim + ' '-}}
192
+ {%- endfor -%}
193
+ {%- endif -%}
194
+ {%- set loop_messages = messages[1:] -%}
195
+ {%- endif -%}
196
+ {%- if tools -%}
197
+ {%- for tool in tools %}
198
+ {{- '<|tool>' -}}
199
+ {{- format_function_declaration(tool) | trim -}}
200
+ {{- '<tool|>' -}}
201
+ {%- endfor %}
202
+ {%- set ns.prev_message_type = 'tool' -%}
203
+ {%- endif -%}
204
+ {{- '<turn|>\n' -}}
205
+ {%- endif %}
206
+
207
+ {#- Pre-scan: find last user message index for reasoning guard -#}
208
+ {%- set ns_turn = namespace(last_user_idx=-1) -%}
209
+ {%- for i in range(loop_messages | length) -%}
210
+ {%- if loop_messages[i]['role'] == 'user' -%}
211
+ {%- set ns_turn.last_user_idx = i -%}
212
+ {%- endif -%}
213
+ {%- endfor -%}
214
+
215
+ {#- Loop through messages -#}
216
+ {%- for message in loop_messages -%}
217
+ {%- if message['role'] != 'tool' -%}
218
+ {%- set ns.prev_message_type = None -%}
219
+ {%- set role = 'model' if message['role'] == 'assistant' else message['role'] -%}
220
+ {#- Detect continuation: suppress duplicate <|turn>model when previous non-tool message was also assistant -#}
221
+ {%- set prev_nt = namespace(role=None, found=false) -%}
222
+ {%- if loop.index0 > 0 -%}
223
+ {%- for j in range(loop.index0 - 1, -1, -1) -%}
224
+ {%- if not prev_nt.found -%}
225
+ {%- if loop_messages[j]['role'] != 'tool' -%}
226
+ {%- set prev_nt.role = loop_messages[j]['role'] -%}
227
+ {%- set prev_nt.found = true -%}
228
+ {%- endif -%}
229
+ {%- endif -%}
230
+ {%- endfor -%}
231
+ {%- endif -%}
232
+ {%- set continue_same_model_turn = (role == 'model' and prev_nt.role == 'assistant') -%}
233
+ {%- if not continue_same_model_turn -%}
234
+ {{- '<|turn>' + role + '\n' }}
235
+ {%- endif -%}
236
+
237
+ {#- Render reasoning/reasoning_content as thinking channel -#}
238
+ {%- set thinking_text = message.get('reasoning') or message.get('reasoning_content') -%}
239
+ {%- if thinking_text and loop.index0 > ns_turn.last_user_idx and message.get('tool_calls') -%}
240
+ {{- '<|channel>thought\n' + thinking_text + '\n<channel|>' -}}
241
+ {%- endif -%}
242
+
243
+ {%- if message['tool_calls'] -%}
244
+ {%- for tool_call in message['tool_calls'] -%}
245
+ {%- set function = tool_call['function'] -%}
246
+ {{- '<|tool_call>call:' + function['name'] + '{' -}}
247
+ {%- if function['arguments'] is mapping -%}
248
+ {%- set ns_args = namespace(found_first=false) -%}
249
+ {%- for key, value in function['arguments'] | dictsort -%}
250
+ {%- if ns_args.found_first %},{% endif -%}
251
+ {%- set ns_args.found_first = true -%}
252
+ {{- key -}}:{{- format_argument(value, escape_keys=False) -}}
253
+ {%- endfor -%}
254
+ {%- elif function['arguments'] is string -%}
255
+ {{- function['arguments'] -}}
256
+ {%- endif -%}
257
+ {{- '}<tool_call|>' -}}
258
+ {%- endfor -%}
259
+ {%- set ns.prev_message_type = 'tool_call' -%}
260
+ {%- endif -%}
261
+
262
+ {%- set ns_tr_out = namespace(flag=false) -%}
263
+ {%- if message.get('tool_responses') -%}
264
+ {#- Legacy: tool_responses embedded on the assistant message (Google/Gemma native) -#}
265
+ {%- for tool_response in message['tool_responses'] -%}
266
+ {{- format_tool_response_block(tool_response['name'] | default('unknown'), tool_response['response']) -}}
267
+ {%- set ns_tr_out.flag = true -%}
268
+ {%- set ns.prev_message_type = 'tool_response' -%}
269
+ {%- endfor -%}
270
+ {%- elif message.get('tool_calls') -%}
271
+ {#- OpenAI Chat Completions: forward-scan consecutive role:tool messages -#}
272
+ {%- set ns_tool_scan = namespace(stopped=false) -%}
273
+ {%- for k in range(loop.index0 + 1, loop_messages | length) -%}
274
+ {%- if ns_tool_scan.stopped -%}
275
+ {%- elif loop_messages[k]['role'] != 'tool' -%}
276
+ {%- set ns_tool_scan.stopped = true -%}
277
+ {%- else -%}
278
+ {%- set follow = loop_messages[k] -%}
279
+ {#- Resolve tool_call_id to function name -#}
280
+ {%- set ns_tname = namespace(name=follow.get('name') | default('unknown')) -%}
281
+ {%- for tc in message['tool_calls'] -%}
282
+ {%- if tc.get('id') == follow.get('tool_call_id') -%}
283
+ {%- set ns_tname.name = tc['function']['name'] -%}
284
+ {%- endif -%}
285
+ {%- endfor -%}
286
+ {#- Handle content as string or content-parts array -#}
287
+ {%- set tool_body = follow.get('content') -%}
288
+ {%- if tool_body is string -%}
289
+ {{- format_tool_response_block(ns_tname.name, tool_body) -}}
290
+ {%- elif tool_body is sequence and tool_body is not string -%}
291
+ {%- set ns_txt = namespace(s='') -%}
292
+ {%- for part in tool_body -%}
293
+ {%- if part.get('type') == 'text' -%}
294
+ {%- set ns_txt.s = ns_txt.s + (part.get('text') | default('')) -%}
295
+ {%- endif -%}
296
+ {%- endfor -%}
297
+ {{- format_tool_response_block(ns_tname.name, ns_txt.s) -}}
298
+ {%- else -%}
299
+ {{- format_tool_response_block(ns_tname.name, tool_body) -}}
300
+ {%- endif -%}
301
+ {%- set ns_tr_out.flag = true -%}
302
+ {%- set ns.prev_message_type = 'tool_response' -%}
303
+ {%- endif -%}
304
+ {%- endfor -%}
305
+ {%- endif -%}
306
+
307
+ {%- set captured_content -%}
308
+ {%- if message['content'] is string -%}
309
+ {%- if role == 'model' -%}
310
+ {{- strip_thinking(message['content']) -}}
311
+ {%- else -%}
312
+ {{- message['content'] | trim -}}
313
+ {%- endif -%}
314
+ {%- elif message['content'] is sequence -%}
315
+ {%- for item in message['content'] -%}
316
+ {%- if item['type'] == 'text' -%}
317
+ {%- if role == 'model' -%}
318
+ {{- strip_thinking(item['text']) -}}
319
+ {%- else -%}
320
+ {{- item['text'] | trim -}}
321
+ {%- endif -%}
322
+ {%- elif item['type'] == 'image' -%}
323
+ {{- '<|image|>' -}}
324
+ {%- set ns.prev_message_type = 'image' -%}
325
+ {%- elif item['type'] == 'audio' -%}
326
+ {{- '<|audio|>' -}}
327
+ {%- set ns.prev_message_type = 'audio' -%}
328
+ {%- elif item['type'] == 'video' -%}
329
+ {{- '<|video|>' -}}
330
+ {%- set ns.prev_message_type = 'video' -%}
331
+ {%- endif -%}
332
+ {%- endfor -%}
333
+ {%- endif -%}
334
+ {%- endset -%}
335
+
336
+ {{- captured_content -}}
337
+ {%- set has_content = captured_content | trim | length > 0 -%}
338
+
339
+ {%- if ns.prev_message_type == 'tool_call' and not ns_tr_out.flag -%}
340
+ {{- '<|tool_response>' -}}
341
+ {%- elif not (ns_tr_out.flag and not has_content) -%}
342
+ {{- '<turn|>\n' -}}
343
+ {%- endif -%}
344
+ {%- endif -%}
345
+ {%- endfor -%}
346
+
347
+ {%- if add_generation_prompt -%}
348
+ {%- if ns.prev_message_type != 'tool_response' and ns.prev_message_type != 'tool_call' -%}
349
+ {{- '<|turn>model\n' -}}
350
+ {%- endif -%}
351
+ {%- endif -%}
checkpoint-120/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:095e1fcdf14af0658b82cfeea97af6ef9916881ed1b2df6af4546d0f79623b85
3
+ size 52047117
checkpoint-120/processor_config.json ADDED
@@ -0,0 +1,75 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "audio_ms_per_token": 40,
3
+ "audio_seq_length": 750,
4
+ "feature_extractor": {
5
+ "dither": 0.0,
6
+ "feature_extractor_type": "Gemma4AudioFeatureExtractor",
7
+ "feature_size": 128,
8
+ "fft_length": 512,
9
+ "fft_overdrive": false,
10
+ "frame_length": 320,
11
+ "hop_length": 160,
12
+ "input_scale_factor": 1.0,
13
+ "max_frequency": 8000.0,
14
+ "mel_floor": 0.001,
15
+ "min_frequency": 0.0,
16
+ "padding_side": "left",
17
+ "padding_value": 0.0,
18
+ "per_bin_mean": null,
19
+ "per_bin_stddev": null,
20
+ "preemphasis": 0.0,
21
+ "preemphasis_htk_flavor": true,
22
+ "return_attention_mask": true,
23
+ "sampling_rate": 16000
24
+ },
25
+ "image_processor": {
26
+ "do_convert_rgb": true,
27
+ "do_normalize": false,
28
+ "do_rescale": true,
29
+ "do_resize": true,
30
+ "image_mean": [
31
+ 0.0,
32
+ 0.0,
33
+ 0.0
34
+ ],
35
+ "image_processor_type": "Gemma4ImageProcessor",
36
+ "image_seq_length": 280,
37
+ "image_std": [
38
+ 1.0,
39
+ 1.0,
40
+ 1.0
41
+ ],
42
+ "max_soft_tokens": 280,
43
+ "patch_size": 16,
44
+ "pooling_kernel_size": 3,
45
+ "resample": 3,
46
+ "rescale_factor": 0.00392156862745098
47
+ },
48
+ "image_seq_length": 280,
49
+ "processor_class": "Gemma4Processor",
50
+ "video_processor": {
51
+ "do_convert_rgb": true,
52
+ "do_normalize": true,
53
+ "do_rescale": true,
54
+ "do_resize": true,
55
+ "do_sample_frames": true,
56
+ "image_mean": [
57
+ 0.0,
58
+ 0.0,
59
+ 0.0
60
+ ],
61
+ "image_std": [
62
+ 1.0,
63
+ 1.0,
64
+ 1.0
65
+ ],
66
+ "max_soft_tokens": 70,
67
+ "num_frames": 32,
68
+ "patch_size": 16,
69
+ "pooling_kernel_size": 3,
70
+ "resample": 3,
71
+ "rescale_factor": 0.00392156862745098,
72
+ "return_metadata": false,
73
+ "video_processor_type": "Gemma4VideoProcessor"
74
+ }
75
+ }
checkpoint-120/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f1d565802a8e26c4e8a31328752b7a7fdc186d9401aa008e65697d0ad8c22e33
3
+ size 14645
checkpoint-120/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aa76f9fdc8f6bd88ee74f70ad294ae36bf2fe178b9965d6d01f072e8fa9198a8
3
+ size 1383
checkpoint-120/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2d142e1b585c4fbc6a90a9d8170d8738b8abf95d0d40cf8046076286563427f1
3
+ size 1465
checkpoint-120/tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cc8d3a0ce36466ccc1278bf987df5f71db1719b9ca6b4118264f45cb627bfe0f
3
+ size 32169626
checkpoint-120/tokenizer_config.json ADDED
@@ -0,0 +1,289 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ "<|video|>"
18
+ ],
19
+ "image_token": "<|image|>",
20
+ "is_local": false,
21
+ "mask_token": "<mask>",
22
+ "model_max_length": 131072,
23
+ "model_specific_special_tokens": {
24
+ "audio_token": "<|audio|>",
25
+ "boa_token": "<|audio>",
26
+ "boi_token": "<|image>",
27
+ "eoa_token": "<audio|>",
28
+ "eoc_token": "<channel|>",
29
+ "eoi_token": "<image|>",
30
+ "eot_token": "<turn|>",
31
+ "escape_token": "<|\"|>",
32
+ "etc_token": "<tool_call|>",
33
+ "etd_token": "<tool|>",
34
+ "etr_token": "<tool_response|>",
35
+ "image_token": "<|image|>",
36
+ "soc_token": "<|channel>",
37
+ "sot_token": "<|turn>",
38
+ "stc_token": "<|tool_call>",
39
+ "std_token": "<|tool>",
40
+ "str_token": "<|tool_response>",
41
+ "think_token": "<|think|>"
42
+ },
43
+ "pad_token": "<pad>",
44
+ "padding_side": "right",
45
+ "processor_class": "Gemma4Processor",
46
+ "response_schema": {
47
+ "properties": {
48
+ "content": {
49
+ "type": "string"
50
+ },
51
+ "role": {
52
+ "const": "assistant"
53
+ },
54
+ "thinking": {
55
+ "type": "string"
56
+ },
57
+ "tool_calls": {
58
+ "items": {
59
+ "properties": {
60
+ "function": {
61
+ "properties": {
62
+ "arguments": {
63
+ "additionalProperties": {},
64
+ "type": "object",
65
+ "x-parser": "gemma4-tool-call"
66
+ },
67
+ "name": {
68
+ "type": "string"
69
+ }
70
+ },
71
+ "type": "object",
72
+ "x-regex": "call\\:(?P<name>\\w+)(?P<arguments>\\{.*\\})"
73
+ },
74
+ "type": {
75
+ "const": "function"
76
+ }
77
+ },
78
+ "type": "object"
79
+ },
80
+ "type": "array",
81
+ "x-regex-iterator": "<\\|tool_call>(.*?)<tool_call\\|>"
82
+ }
83
+ },
84
+ "type": "object",
85
+ "x-regex": "(\\<\\|channel\\>thought\\n(?P<thinking>.*?)\\<channel\\|\\>)?(?P<tool_calls>\\<\\|tool_call\\>.*\\<tool_call\\|\\>)?(?P<content>(?:(?!\\<turn\\|\\>)(?!\\<\\|tool_response\\>).)+)?(?:\\<turn\\|\\>|\\<\\|tool_response\\>)?"
86
+ },
87
+ "soc_token": "<|channel>",
88
+ "sot_token": "<|turn>",
89
+ "stc_token": "<|tool_call>",
90
+ "std_token": "<|tool>",
91
+ "str_token": "<|tool_response>",
92
+ "think_token": "<|think|>",
93
+ "tokenizer_class": "GemmaTokenizer",
94
+ "unk_token": "<unk>",
95
+ "added_tokens_decoder": {
96
+ "0": {
97
+ "content": "<pad>",
98
+ "single_word": false,
99
+ "lstrip": false,
100
+ "rstrip": false,
101
+ "normalized": false,
102
+ "special": true
103
+ },
104
+ "1": {
105
+ "content": "<eos>",
106
+ "single_word": false,
107
+ "lstrip": false,
108
+ "rstrip": false,
109
+ "normalized": false,
110
+ "special": true
111
+ },
112
+ "2": {
113
+ "content": "<bos>",
114
+ "single_word": false,
115
+ "lstrip": false,
116
+ "rstrip": false,
117
+ "normalized": false,
118
+ "special": true
119
+ },
120
+ "3": {
121
+ "content": "<unk>",
122
+ "single_word": false,
123
+ "lstrip": false,
124
+ "rstrip": false,
125
+ "normalized": false,
126
+ "special": true
127
+ },
128
+ "4": {
129
+ "content": "<mask>",
130
+ "single_word": false,
131
+ "lstrip": false,
132
+ "rstrip": false,
133
+ "normalized": false,
134
+ "special": true
135
+ },
136
+ "46": {
137
+ "content": "<|tool>",
138
+ "single_word": false,
139
+ "lstrip": false,
140
+ "rstrip": false,
141
+ "normalized": false,
142
+ "special": true
143
+ },
144
+ "47": {
145
+ "content": "<tool|>",
146
+ "single_word": false,
147
+ "lstrip": false,
148
+ "rstrip": false,
149
+ "normalized": false,
150
+ "special": true
151
+ },
152
+ "48": {
153
+ "content": "<|tool_call>",
154
+ "single_word": false,
155
+ "lstrip": false,
156
+ "rstrip": false,
157
+ "normalized": false,
158
+ "special": true
159
+ },
160
+ "49": {
161
+ "content": "<tool_call|>",
162
+ "single_word": false,
163
+ "lstrip": false,
164
+ "rstrip": false,
165
+ "normalized": false,
166
+ "special": true
167
+ },
168
+ "50": {
169
+ "content": "<|tool_response>",
170
+ "single_word": false,
171
+ "lstrip": false,
172
+ "rstrip": false,
173
+ "normalized": false,
174
+ "special": true
175
+ },
176
+ "51": {
177
+ "content": "<tool_response|>",
178
+ "single_word": false,
179
+ "lstrip": false,
180
+ "rstrip": false,
181
+ "normalized": false,
182
+ "special": true
183
+ },
184
+ "52": {
185
+ "content": "<|\"|>",
186
+ "single_word": false,
187
+ "lstrip": false,
188
+ "rstrip": false,
189
+ "normalized": false,
190
+ "special": true
191
+ },
192
+ "98": {
193
+ "content": "<|think|>",
194
+ "single_word": false,
195
+ "lstrip": false,
196
+ "rstrip": false,
197
+ "normalized": false,
198
+ "special": true
199
+ },
200
+ "100": {
201
+ "content": "<|channel>",
202
+ "single_word": false,
203
+ "lstrip": false,
204
+ "rstrip": false,
205
+ "normalized": false,
206
+ "special": true
207
+ },
208
+ "101": {
209
+ "content": "<channel|>",
210
+ "single_word": false,
211
+ "lstrip": false,
212
+ "rstrip": false,
213
+ "normalized": false,
214
+ "special": true
215
+ },
216
+ "105": {
217
+ "content": "<|turn>",
218
+ "single_word": false,
219
+ "lstrip": false,
220
+ "rstrip": false,
221
+ "normalized": false,
222
+ "special": true
223
+ },
224
+ "106": {
225
+ "content": "<turn|>",
226
+ "single_word": false,
227
+ "lstrip": false,
228
+ "rstrip": false,
229
+ "normalized": false,
230
+ "special": true
231
+ },
232
+ "255999": {
233
+ "content": "<|image>",
234
+ "single_word": false,
235
+ "lstrip": false,
236
+ "rstrip": false,
237
+ "normalized": false,
238
+ "special": true
239
+ },
240
+ "256000": {
241
+ "content": "<|audio>",
242
+ "single_word": false,
243
+ "lstrip": false,
244
+ "rstrip": false,
245
+ "normalized": false,
246
+ "special": true
247
+ },
248
+ "258880": {
249
+ "content": "<|image|>",
250
+ "single_word": false,
251
+ "lstrip": false,
252
+ "rstrip": false,
253
+ "normalized": false,
254
+ "special": true
255
+ },
256
+ "258881": {
257
+ "content": "<|audio|>",
258
+ "single_word": false,
259
+ "lstrip": false,
260
+ "rstrip": false,
261
+ "normalized": false,
262
+ "special": true
263
+ },
264
+ "258882": {
265
+ "content": "<image|>",
266
+ "single_word": false,
267
+ "lstrip": false,
268
+ "rstrip": false,
269
+ "normalized": false,
270
+ "special": true
271
+ },
272
+ "258883": {
273
+ "content": "<audio|>",
274
+ "single_word": false,
275
+ "lstrip": false,
276
+ "rstrip": false,
277
+ "normalized": false,
278
+ "special": true
279
+ },
280
+ "258884": {
281
+ "content": "<|video|>",
282
+ "single_word": false,
283
+ "lstrip": false,
284
+ "rstrip": false,
285
+ "normalized": false,
286
+ "special": true
287
+ }
288
+ }
289
+ }
checkpoint-120/trainer_state.json ADDED
@@ -0,0 +1,1234 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 0.028402366863905324,
6
+ "eval_steps": 500,
7
+ "global_step": 120,
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.00023668639053254438,
14
+ "grad_norm": NaN,
15
+ "learning_rate": 0.0,
16
+ "loss": 4.347962379455566,
17
+ "num_input_tokens_seen": 842,
18
+ "step": 1,
19
+ "train_runtime": 63.0289,
20
+ "train_tokens_per_second": 13.359
21
+ },
22
+ {
23
+ "epoch": 0.00047337278106508875,
24
+ "grad_norm": NaN,
25
+ "learning_rate": 0.0,
26
+ "loss": 4.634688854217529,
27
+ "num_input_tokens_seen": 1820,
28
+ "step": 2,
29
+ "train_runtime": 67.7033,
30
+ "train_tokens_per_second": 26.882
31
+ },
32
+ {
33
+ "epoch": 0.0007100591715976331,
34
+ "grad_norm": 13.944212913513184,
35
+ "learning_rate": 0.0,
36
+ "loss": 4.42241096496582,
37
+ "num_input_tokens_seen": 2754,
38
+ "step": 3,
39
+ "train_runtime": 72.324,
40
+ "train_tokens_per_second": 38.079
41
+ },
42
+ {
43
+ "epoch": 0.0009467455621301775,
44
+ "grad_norm": 13.445413589477539,
45
+ "learning_rate": 4e-05,
46
+ "loss": 4.682804107666016,
47
+ "num_input_tokens_seen": 3700,
48
+ "step": 4,
49
+ "train_runtime": 75.3925,
50
+ "train_tokens_per_second": 49.077
51
+ },
52
+ {
53
+ "epoch": 0.001183431952662722,
54
+ "grad_norm": 10.576727867126465,
55
+ "learning_rate": 8e-05,
56
+ "loss": 4.166159629821777,
57
+ "num_input_tokens_seen": 4714,
58
+ "step": 5,
59
+ "train_runtime": 78.474,
60
+ "train_tokens_per_second": 60.071
61
+ },
62
+ {
63
+ "epoch": 0.0014201183431952662,
64
+ "grad_norm": 8.077850341796875,
65
+ "learning_rate": 0.00012,
66
+ "loss": 3.2006261348724365,
67
+ "num_input_tokens_seen": 5674,
68
+ "step": 6,
69
+ "train_runtime": 81.6139,
70
+ "train_tokens_per_second": 69.522
71
+ },
72
+ {
73
+ "epoch": 0.0016568047337278107,
74
+ "grad_norm": 6.60789680480957,
75
+ "learning_rate": 0.00016,
76
+ "loss": 2.7067415714263916,
77
+ "num_input_tokens_seen": 6646,
78
+ "step": 7,
79
+ "train_runtime": 84.786,
80
+ "train_tokens_per_second": 78.386
81
+ },
82
+ {
83
+ "epoch": 0.001893491124260355,
84
+ "grad_norm": 9.175209045410156,
85
+ "learning_rate": 0.0002,
86
+ "loss": 2.367570638656616,
87
+ "num_input_tokens_seen": 7518,
88
+ "step": 8,
89
+ "train_runtime": 87.8178,
90
+ "train_tokens_per_second": 85.609
91
+ },
92
+ {
93
+ "epoch": 0.0021301775147928993,
94
+ "grad_norm": 3.438948631286621,
95
+ "learning_rate": 0.00019891891891891895,
96
+ "loss": 2.1013119220733643,
97
+ "num_input_tokens_seen": 8470,
98
+ "step": 9,
99
+ "train_runtime": 90.9458,
100
+ "train_tokens_per_second": 93.132
101
+ },
102
+ {
103
+ "epoch": 0.002366863905325444,
104
+ "grad_norm": 3.459472894668579,
105
+ "learning_rate": 0.00019783783783783784,
106
+ "loss": 1.8991940021514893,
107
+ "num_input_tokens_seen": 9458,
108
+ "step": 10,
109
+ "train_runtime": 94.0577,
110
+ "train_tokens_per_second": 100.555
111
+ },
112
+ {
113
+ "epoch": 0.0026035502958579883,
114
+ "grad_norm": 2.169750928878784,
115
+ "learning_rate": 0.00019675675675675678,
116
+ "loss": 1.7395055294036865,
117
+ "num_input_tokens_seen": 10456,
118
+ "step": 11,
119
+ "train_runtime": 97.2068,
120
+ "train_tokens_per_second": 107.565
121
+ },
122
+ {
123
+ "epoch": 0.0028402366863905324,
124
+ "grad_norm": 1.8638927936553955,
125
+ "learning_rate": 0.00019567567567567567,
126
+ "loss": 1.5622950792312622,
127
+ "num_input_tokens_seen": 11410,
128
+ "step": 12,
129
+ "train_runtime": 100.2911,
130
+ "train_tokens_per_second": 113.769
131
+ },
132
+ {
133
+ "epoch": 0.003076923076923077,
134
+ "grad_norm": 1.4741116762161255,
135
+ "learning_rate": 0.00019459459459459462,
136
+ "loss": 1.392699956893921,
137
+ "num_input_tokens_seen": 12316,
138
+ "step": 13,
139
+ "train_runtime": 103.3664,
140
+ "train_tokens_per_second": 119.149
141
+ },
142
+ {
143
+ "epoch": 0.0033136094674556214,
144
+ "grad_norm": 1.3238699436187744,
145
+ "learning_rate": 0.00019351351351351353,
146
+ "loss": 1.2535074949264526,
147
+ "num_input_tokens_seen": 13256,
148
+ "step": 14,
149
+ "train_runtime": 106.4031,
150
+ "train_tokens_per_second": 124.583
151
+ },
152
+ {
153
+ "epoch": 0.0035502958579881655,
154
+ "grad_norm": 1.276484489440918,
155
+ "learning_rate": 0.00019243243243243245,
156
+ "loss": 1.135048508644104,
157
+ "num_input_tokens_seen": 14228,
158
+ "step": 15,
159
+ "train_runtime": 109.4909,
160
+ "train_tokens_per_second": 129.947
161
+ },
162
+ {
163
+ "epoch": 0.00378698224852071,
164
+ "grad_norm": 1.1560865640640259,
165
+ "learning_rate": 0.00019135135135135137,
166
+ "loss": 1.0317097902297974,
167
+ "num_input_tokens_seen": 15192,
168
+ "step": 16,
169
+ "train_runtime": 112.5741,
170
+ "train_tokens_per_second": 134.951
171
+ },
172
+ {
173
+ "epoch": 0.004023668639053254,
174
+ "grad_norm": 1.1331353187561035,
175
+ "learning_rate": 0.00019027027027027028,
176
+ "loss": 0.9967219233512878,
177
+ "num_input_tokens_seen": 16130,
178
+ "step": 17,
179
+ "train_runtime": 115.6495,
180
+ "train_tokens_per_second": 139.473
181
+ },
182
+ {
183
+ "epoch": 0.004260355029585799,
184
+ "grad_norm": 1.1567702293395996,
185
+ "learning_rate": 0.0001891891891891892,
186
+ "loss": 0.905368447303772,
187
+ "num_input_tokens_seen": 17078,
188
+ "step": 18,
189
+ "train_runtime": 118.6796,
190
+ "train_tokens_per_second": 143.9
191
+ },
192
+ {
193
+ "epoch": 0.004497041420118343,
194
+ "grad_norm": 0.9157600998878479,
195
+ "learning_rate": 0.00018810810810810814,
196
+ "loss": 0.8075690865516663,
197
+ "num_input_tokens_seen": 18062,
198
+ "step": 19,
199
+ "train_runtime": 121.8602,
200
+ "train_tokens_per_second": 148.219
201
+ },
202
+ {
203
+ "epoch": 0.004733727810650888,
204
+ "grad_norm": 0.8306605815887451,
205
+ "learning_rate": 0.00018702702702702703,
206
+ "loss": 0.8007175922393799,
207
+ "num_input_tokens_seen": 19078,
208
+ "step": 20,
209
+ "train_runtime": 125.0019,
210
+ "train_tokens_per_second": 152.622
211
+ },
212
+ {
213
+ "epoch": 0.004970414201183432,
214
+ "grad_norm": 0.690018355846405,
215
+ "learning_rate": 0.00018594594594594597,
216
+ "loss": 0.7810791730880737,
217
+ "num_input_tokens_seen": 20068,
218
+ "step": 21,
219
+ "train_runtime": 128.0932,
220
+ "train_tokens_per_second": 156.667
221
+ },
222
+ {
223
+ "epoch": 0.005207100591715977,
224
+ "grad_norm": 1.4768867492675781,
225
+ "learning_rate": 0.00018486486486486486,
226
+ "loss": 0.7172120809555054,
227
+ "num_input_tokens_seen": 20972,
228
+ "step": 22,
229
+ "train_runtime": 131.0901,
230
+ "train_tokens_per_second": 159.982
231
+ },
232
+ {
233
+ "epoch": 0.00544378698224852,
234
+ "grad_norm": 0.7999809980392456,
235
+ "learning_rate": 0.0001837837837837838,
236
+ "loss": 0.6333726048469543,
237
+ "num_input_tokens_seen": 21900,
238
+ "step": 23,
239
+ "train_runtime": 134.212,
240
+ "train_tokens_per_second": 163.175
241
+ },
242
+ {
243
+ "epoch": 0.005680473372781065,
244
+ "grad_norm": 0.9285996556282043,
245
+ "learning_rate": 0.00018270270270270272,
246
+ "loss": 0.6750729084014893,
247
+ "num_input_tokens_seen": 22770,
248
+ "step": 24,
249
+ "train_runtime": 137.3193,
250
+ "train_tokens_per_second": 165.818
251
+ },
252
+ {
253
+ "epoch": 0.005917159763313609,
254
+ "grad_norm": 0.7733820080757141,
255
+ "learning_rate": 0.00018162162162162164,
256
+ "loss": 0.672032356262207,
257
+ "num_input_tokens_seen": 23746,
258
+ "step": 25,
259
+ "train_runtime": 140.3175,
260
+ "train_tokens_per_second": 169.23
261
+ },
262
+ {
263
+ "epoch": 0.006153846153846154,
264
+ "grad_norm": 0.73944091796875,
265
+ "learning_rate": 0.00018054054054054055,
266
+ "loss": 0.6716101169586182,
267
+ "num_input_tokens_seen": 24716,
268
+ "step": 26,
269
+ "train_runtime": 143.3683,
270
+ "train_tokens_per_second": 172.395
271
+ },
272
+ {
273
+ "epoch": 0.006390532544378698,
274
+ "grad_norm": 0.7659372091293335,
275
+ "learning_rate": 0.00017945945945945947,
276
+ "loss": 0.6067175269126892,
277
+ "num_input_tokens_seen": 25724,
278
+ "step": 27,
279
+ "train_runtime": 146.5377,
280
+ "train_tokens_per_second": 175.545
281
+ },
282
+ {
283
+ "epoch": 0.006627218934911243,
284
+ "grad_norm": 0.5428617000579834,
285
+ "learning_rate": 0.00017837837837837839,
286
+ "loss": 0.6083395481109619,
287
+ "num_input_tokens_seen": 26712,
288
+ "step": 28,
289
+ "train_runtime": 149.6429,
290
+ "train_tokens_per_second": 178.505
291
+ },
292
+ {
293
+ "epoch": 0.006863905325443787,
294
+ "grad_norm": 0.5915114283561707,
295
+ "learning_rate": 0.00017729729729729733,
296
+ "loss": 0.6463742256164551,
297
+ "num_input_tokens_seen": 27704,
298
+ "step": 29,
299
+ "train_runtime": 152.7501,
300
+ "train_tokens_per_second": 181.368
301
+ },
302
+ {
303
+ "epoch": 0.007100591715976331,
304
+ "grad_norm": 1.3350757360458374,
305
+ "learning_rate": 0.00017621621621621622,
306
+ "loss": 0.5705698132514954,
307
+ "num_input_tokens_seen": 28648,
308
+ "step": 30,
309
+ "train_runtime": 155.8918,
310
+ "train_tokens_per_second": 183.768
311
+ },
312
+ {
313
+ "epoch": 0.0073372781065088755,
314
+ "grad_norm": 0.68421471118927,
315
+ "learning_rate": 0.00017513513513513516,
316
+ "loss": 0.5660746097564697,
317
+ "num_input_tokens_seen": 29622,
318
+ "step": 31,
319
+ "train_runtime": 163.3209,
320
+ "train_tokens_per_second": 181.373
321
+ },
322
+ {
323
+ "epoch": 0.00757396449704142,
324
+ "grad_norm": 0.5762604475021362,
325
+ "learning_rate": 0.00017405405405405405,
326
+ "loss": 0.5483309030532837,
327
+ "num_input_tokens_seen": 30534,
328
+ "step": 32,
329
+ "train_runtime": 166.5185,
330
+ "train_tokens_per_second": 183.367
331
+ },
332
+ {
333
+ "epoch": 0.0078106508875739646,
334
+ "grad_norm": 0.5751784443855286,
335
+ "learning_rate": 0.000172972972972973,
336
+ "loss": 0.5207005739212036,
337
+ "num_input_tokens_seen": 31458,
338
+ "step": 33,
339
+ "train_runtime": 169.7055,
340
+ "train_tokens_per_second": 185.368
341
+ },
342
+ {
343
+ "epoch": 0.008047337278106508,
344
+ "grad_norm": 0.5752039551734924,
345
+ "learning_rate": 0.0001718918918918919,
346
+ "loss": 0.5534584522247314,
347
+ "num_input_tokens_seen": 32424,
348
+ "step": 34,
349
+ "train_runtime": 172.9184,
350
+ "train_tokens_per_second": 187.51
351
+ },
352
+ {
353
+ "epoch": 0.008284023668639054,
354
+ "grad_norm": 0.5918363332748413,
355
+ "learning_rate": 0.00017081081081081083,
356
+ "loss": 0.482683002948761,
357
+ "num_input_tokens_seen": 33398,
358
+ "step": 35,
359
+ "train_runtime": 176.2158,
360
+ "train_tokens_per_second": 189.529
361
+ },
362
+ {
363
+ "epoch": 0.008520710059171597,
364
+ "grad_norm": 0.6921846270561218,
365
+ "learning_rate": 0.00016972972972972974,
366
+ "loss": 0.46112507581710815,
367
+ "num_input_tokens_seen": 34324,
368
+ "step": 36,
369
+ "train_runtime": 179.3173,
370
+ "train_tokens_per_second": 191.415
371
+ },
372
+ {
373
+ "epoch": 0.008757396449704143,
374
+ "grad_norm": 0.6429721117019653,
375
+ "learning_rate": 0.00016864864864864866,
376
+ "loss": 0.45629197359085083,
377
+ "num_input_tokens_seen": 35270,
378
+ "step": 37,
379
+ "train_runtime": 182.3558,
380
+ "train_tokens_per_second": 193.413
381
+ },
382
+ {
383
+ "epoch": 0.008994082840236686,
384
+ "grad_norm": 0.5477020740509033,
385
+ "learning_rate": 0.00016756756756756757,
386
+ "loss": 0.4547877907752991,
387
+ "num_input_tokens_seen": 36206,
388
+ "step": 38,
389
+ "train_runtime": 185.4483,
390
+ "train_tokens_per_second": 195.235
391
+ },
392
+ {
393
+ "epoch": 0.009230769230769232,
394
+ "grad_norm": 0.6158634424209595,
395
+ "learning_rate": 0.00016648648648648652,
396
+ "loss": 0.367482990026474,
397
+ "num_input_tokens_seen": 37080,
398
+ "step": 39,
399
+ "train_runtime": 188.5114,
400
+ "train_tokens_per_second": 196.699
401
+ },
402
+ {
403
+ "epoch": 0.009467455621301775,
404
+ "grad_norm": 0.5405091643333435,
405
+ "learning_rate": 0.0001654054054054054,
406
+ "loss": 0.42049410939216614,
407
+ "num_input_tokens_seen": 38030,
408
+ "step": 40,
409
+ "train_runtime": 191.5986,
410
+ "train_tokens_per_second": 198.488
411
+ },
412
+ {
413
+ "epoch": 0.009704142011834319,
414
+ "grad_norm": 0.6389183402061462,
415
+ "learning_rate": 0.00016432432432432435,
416
+ "loss": 0.3935014307498932,
417
+ "num_input_tokens_seen": 38942,
418
+ "step": 41,
419
+ "train_runtime": 194.8426,
420
+ "train_tokens_per_second": 199.864
421
+ },
422
+ {
423
+ "epoch": 0.009940828402366864,
424
+ "grad_norm": 0.6742956042289734,
425
+ "learning_rate": 0.00016324324324324324,
426
+ "loss": 0.4373692274093628,
427
+ "num_input_tokens_seen": 39902,
428
+ "step": 42,
429
+ "train_runtime": 197.8938,
430
+ "train_tokens_per_second": 201.633
431
+ },
432
+ {
433
+ "epoch": 0.010177514792899408,
434
+ "grad_norm": 0.6083794236183167,
435
+ "learning_rate": 0.00016216216216216218,
436
+ "loss": 0.3776765465736389,
437
+ "num_input_tokens_seen": 40858,
438
+ "step": 43,
439
+ "train_runtime": 200.9708,
440
+ "train_tokens_per_second": 203.303
441
+ },
442
+ {
443
+ "epoch": 0.010414201183431953,
444
+ "grad_norm": 0.5469992160797119,
445
+ "learning_rate": 0.0001610810810810811,
446
+ "loss": 0.31133151054382324,
447
+ "num_input_tokens_seen": 41760,
448
+ "step": 44,
449
+ "train_runtime": 204.1304,
450
+ "train_tokens_per_second": 204.575
451
+ },
452
+ {
453
+ "epoch": 0.010650887573964497,
454
+ "grad_norm": 0.6452498435974121,
455
+ "learning_rate": 0.00016,
456
+ "loss": 0.42721521854400635,
457
+ "num_input_tokens_seen": 42740,
458
+ "step": 45,
459
+ "train_runtime": 207.1845,
460
+ "train_tokens_per_second": 206.29
461
+ },
462
+ {
463
+ "epoch": 0.01088757396449704,
464
+ "grad_norm": 0.5988945960998535,
465
+ "learning_rate": 0.00015891891891891893,
466
+ "loss": 0.41375258564949036,
467
+ "num_input_tokens_seen": 43750,
468
+ "step": 46,
469
+ "train_runtime": 210.2912,
470
+ "train_tokens_per_second": 208.045
471
+ },
472
+ {
473
+ "epoch": 0.011124260355029586,
474
+ "grad_norm": 0.6452038288116455,
475
+ "learning_rate": 0.00015783783783783785,
476
+ "loss": 0.3485276699066162,
477
+ "num_input_tokens_seen": 44690,
478
+ "step": 47,
479
+ "train_runtime": 213.3684,
480
+ "train_tokens_per_second": 209.45
481
+ },
482
+ {
483
+ "epoch": 0.01136094674556213,
484
+ "grad_norm": 0.7041454911231995,
485
+ "learning_rate": 0.00015675675675675676,
486
+ "loss": 0.3296637535095215,
487
+ "num_input_tokens_seen": 45666,
488
+ "step": 48,
489
+ "train_runtime": 216.4459,
490
+ "train_tokens_per_second": 210.981
491
+ },
492
+ {
493
+ "epoch": 0.011597633136094675,
494
+ "grad_norm": 0.6726324558258057,
495
+ "learning_rate": 0.0001556756756756757,
496
+ "loss": 0.3179457187652588,
497
+ "num_input_tokens_seen": 46560,
498
+ "step": 49,
499
+ "train_runtime": 219.4697,
500
+ "train_tokens_per_second": 212.148
501
+ },
502
+ {
503
+ "epoch": 0.011834319526627219,
504
+ "grad_norm": 0.7513703107833862,
505
+ "learning_rate": 0.0001545945945945946,
506
+ "loss": 0.3722391426563263,
507
+ "num_input_tokens_seen": 47528,
508
+ "step": 50,
509
+ "train_runtime": 222.596,
510
+ "train_tokens_per_second": 213.517
511
+ },
512
+ {
513
+ "epoch": 0.012071005917159764,
514
+ "grad_norm": 0.6454160213470459,
515
+ "learning_rate": 0.00015351351351351354,
516
+ "loss": 0.283376544713974,
517
+ "num_input_tokens_seen": 48468,
518
+ "step": 51,
519
+ "train_runtime": 225.7262,
520
+ "train_tokens_per_second": 214.72
521
+ },
522
+ {
523
+ "epoch": 0.012307692307692308,
524
+ "grad_norm": 0.9116244316101074,
525
+ "learning_rate": 0.00015243243243243243,
526
+ "loss": 0.4286752939224243,
527
+ "num_input_tokens_seen": 49444,
528
+ "step": 52,
529
+ "train_runtime": 228.803,
530
+ "train_tokens_per_second": 216.099
531
+ },
532
+ {
533
+ "epoch": 0.012544378698224851,
534
+ "grad_norm": 0.6157099604606628,
535
+ "learning_rate": 0.00015135135135135137,
536
+ "loss": 0.27355721592903137,
537
+ "num_input_tokens_seen": 50424,
538
+ "step": 53,
539
+ "train_runtime": 231.8413,
540
+ "train_tokens_per_second": 217.494
541
+ },
542
+ {
543
+ "epoch": 0.012781065088757397,
544
+ "grad_norm": 0.6335808038711548,
545
+ "learning_rate": 0.00015027027027027028,
546
+ "loss": 0.37197017669677734,
547
+ "num_input_tokens_seen": 51328,
548
+ "step": 54,
549
+ "train_runtime": 234.9265,
550
+ "train_tokens_per_second": 218.485
551
+ },
552
+ {
553
+ "epoch": 0.01301775147928994,
554
+ "grad_norm": 0.6838424205780029,
555
+ "learning_rate": 0.0001491891891891892,
556
+ "loss": 0.3504602909088135,
557
+ "num_input_tokens_seen": 52274,
558
+ "step": 55,
559
+ "train_runtime": 238.0114,
560
+ "train_tokens_per_second": 219.628
561
+ },
562
+ {
563
+ "epoch": 0.013254437869822486,
564
+ "grad_norm": 0.6566418409347534,
565
+ "learning_rate": 0.00014810810810810812,
566
+ "loss": 0.3189927935600281,
567
+ "num_input_tokens_seen": 53214,
568
+ "step": 56,
569
+ "train_runtime": 241.0892,
570
+ "train_tokens_per_second": 220.723
571
+ },
572
+ {
573
+ "epoch": 0.01349112426035503,
574
+ "grad_norm": 0.6112971901893616,
575
+ "learning_rate": 0.00014702702702702703,
576
+ "loss": 0.3080369830131531,
577
+ "num_input_tokens_seen": 54162,
578
+ "step": 57,
579
+ "train_runtime": 244.2461,
580
+ "train_tokens_per_second": 221.752
581
+ },
582
+ {
583
+ "epoch": 0.013727810650887575,
584
+ "grad_norm": 0.6469757556915283,
585
+ "learning_rate": 0.00014594594594594595,
586
+ "loss": 0.24279898405075073,
587
+ "num_input_tokens_seen": 55158,
588
+ "step": 58,
589
+ "train_runtime": 247.4064,
590
+ "train_tokens_per_second": 222.945
591
+ },
592
+ {
593
+ "epoch": 0.013964497041420118,
594
+ "grad_norm": 0.7099384665489197,
595
+ "learning_rate": 0.0001448648648648649,
596
+ "loss": 0.31188705563545227,
597
+ "num_input_tokens_seen": 56096,
598
+ "step": 59,
599
+ "train_runtime": 250.4708,
600
+ "train_tokens_per_second": 223.962
601
+ },
602
+ {
603
+ "epoch": 0.014201183431952662,
604
+ "grad_norm": 0.5927040576934814,
605
+ "learning_rate": 0.00014378378378378378,
606
+ "loss": 0.25696688890457153,
607
+ "num_input_tokens_seen": 57012,
608
+ "step": 60,
609
+ "train_runtime": 253.7131,
610
+ "train_tokens_per_second": 224.711
611
+ },
612
+ {
613
+ "epoch": 0.014437869822485207,
614
+ "grad_norm": 0.5483338236808777,
615
+ "learning_rate": 0.00014270270270270272,
616
+ "loss": 0.2725352644920349,
617
+ "num_input_tokens_seen": 58036,
618
+ "step": 61,
619
+ "train_runtime": 260.2844,
620
+ "train_tokens_per_second": 222.971
621
+ },
622
+ {
623
+ "epoch": 0.014674556213017751,
624
+ "grad_norm": 0.6806847453117371,
625
+ "learning_rate": 0.00014162162162162161,
626
+ "loss": 0.31959983706474304,
627
+ "num_input_tokens_seen": 59026,
628
+ "step": 62,
629
+ "train_runtime": 263.3802,
630
+ "train_tokens_per_second": 224.109
631
+ },
632
+ {
633
+ "epoch": 0.014911242603550296,
634
+ "grad_norm": 0.6916660070419312,
635
+ "learning_rate": 0.00014054054054054056,
636
+ "loss": 0.312305748462677,
637
+ "num_input_tokens_seen": 59982,
638
+ "step": 63,
639
+ "train_runtime": 266.3794,
640
+ "train_tokens_per_second": 225.175
641
+ },
642
+ {
643
+ "epoch": 0.01514792899408284,
644
+ "grad_norm": 0.5516266226768494,
645
+ "learning_rate": 0.00013945945945945947,
646
+ "loss": 0.2653953433036804,
647
+ "num_input_tokens_seen": 60926,
648
+ "step": 64,
649
+ "train_runtime": 269.3893,
650
+ "train_tokens_per_second": 226.163
651
+ },
652
+ {
653
+ "epoch": 0.015384615384615385,
654
+ "grad_norm": 0.5579270720481873,
655
+ "learning_rate": 0.0001383783783783784,
656
+ "loss": 0.20462818443775177,
657
+ "num_input_tokens_seen": 61844,
658
+ "step": 65,
659
+ "train_runtime": 272.3982,
660
+ "train_tokens_per_second": 227.035
661
+ },
662
+ {
663
+ "epoch": 0.015621301775147929,
664
+ "grad_norm": 0.7645660638809204,
665
+ "learning_rate": 0.0001372972972972973,
666
+ "loss": 0.2650893032550812,
667
+ "num_input_tokens_seen": 62766,
668
+ "step": 66,
669
+ "train_runtime": 275.404,
670
+ "train_tokens_per_second": 227.905
671
+ },
672
+ {
673
+ "epoch": 0.015857988165680473,
674
+ "grad_norm": 0.5974738001823425,
675
+ "learning_rate": 0.00013621621621621622,
676
+ "loss": 0.21247778832912445,
677
+ "num_input_tokens_seen": 63690,
678
+ "step": 67,
679
+ "train_runtime": 278.4021,
680
+ "train_tokens_per_second": 228.77
681
+ },
682
+ {
683
+ "epoch": 0.016094674556213016,
684
+ "grad_norm": 0.5831193327903748,
685
+ "learning_rate": 0.00013513513513513514,
686
+ "loss": 0.2024172842502594,
687
+ "num_input_tokens_seen": 64496,
688
+ "step": 68,
689
+ "train_runtime": 281.4313,
690
+ "train_tokens_per_second": 229.171
691
+ },
692
+ {
693
+ "epoch": 0.016331360946745564,
694
+ "grad_norm": 0.8678795695304871,
695
+ "learning_rate": 0.00013405405405405408,
696
+ "loss": 0.31490638852119446,
697
+ "num_input_tokens_seen": 65460,
698
+ "step": 69,
699
+ "train_runtime": 284.6137,
700
+ "train_tokens_per_second": 229.996
701
+ },
702
+ {
703
+ "epoch": 0.016568047337278107,
704
+ "grad_norm": 0.7611344456672668,
705
+ "learning_rate": 0.00013297297297297297,
706
+ "loss": 0.24661925435066223,
707
+ "num_input_tokens_seen": 66374,
708
+ "step": 70,
709
+ "train_runtime": 287.7742,
710
+ "train_tokens_per_second": 230.646
711
+ },
712
+ {
713
+ "epoch": 0.01680473372781065,
714
+ "grad_norm": 0.6685308814048767,
715
+ "learning_rate": 0.0001318918918918919,
716
+ "loss": 0.1843535602092743,
717
+ "num_input_tokens_seen": 67112,
718
+ "step": 71,
719
+ "train_runtime": 290.9321,
720
+ "train_tokens_per_second": 230.679
721
+ },
722
+ {
723
+ "epoch": 0.017041420118343194,
724
+ "grad_norm": 0.6150883436203003,
725
+ "learning_rate": 0.0001308108108108108,
726
+ "loss": 0.23372295498847961,
727
+ "num_input_tokens_seen": 68154,
728
+ "step": 72,
729
+ "train_runtime": 294.0809,
730
+ "train_tokens_per_second": 231.753
731
+ },
732
+ {
733
+ "epoch": 0.017278106508875738,
734
+ "grad_norm": 0.6938812732696533,
735
+ "learning_rate": 0.00012972972972972974,
736
+ "loss": 0.22619186341762543,
737
+ "num_input_tokens_seen": 69102,
738
+ "step": 73,
739
+ "train_runtime": 297.1843,
740
+ "train_tokens_per_second": 232.522
741
+ },
742
+ {
743
+ "epoch": 0.017514792899408285,
744
+ "grad_norm": 0.617787778377533,
745
+ "learning_rate": 0.00012864864864864866,
746
+ "loss": 0.2535328269004822,
747
+ "num_input_tokens_seen": 70010,
748
+ "step": 74,
749
+ "train_runtime": 300.2851,
750
+ "train_tokens_per_second": 233.145
751
+ },
752
+ {
753
+ "epoch": 0.01775147928994083,
754
+ "grad_norm": 0.6884900331497192,
755
+ "learning_rate": 0.00012756756756756758,
756
+ "loss": 0.21632739901542664,
757
+ "num_input_tokens_seen": 70940,
758
+ "step": 75,
759
+ "train_runtime": 303.3533,
760
+ "train_tokens_per_second": 233.853
761
+ },
762
+ {
763
+ "epoch": 0.017988165680473372,
764
+ "grad_norm": 0.5782189965248108,
765
+ "learning_rate": 0.0001264864864864865,
766
+ "loss": 0.20167675614356995,
767
+ "num_input_tokens_seen": 71862,
768
+ "step": 76,
769
+ "train_runtime": 306.4503,
770
+ "train_tokens_per_second": 234.498
771
+ },
772
+ {
773
+ "epoch": 0.018224852071005916,
774
+ "grad_norm": 0.6541483402252197,
775
+ "learning_rate": 0.0001254054054054054,
776
+ "loss": 0.23231221735477448,
777
+ "num_input_tokens_seen": 72812,
778
+ "step": 77,
779
+ "train_runtime": 309.5743,
780
+ "train_tokens_per_second": 235.2
781
+ },
782
+ {
783
+ "epoch": 0.018461538461538463,
784
+ "grad_norm": 0.7524886727333069,
785
+ "learning_rate": 0.00012432432432432433,
786
+ "loss": 0.27632662653923035,
787
+ "num_input_tokens_seen": 73774,
788
+ "step": 78,
789
+ "train_runtime": 312.6963,
790
+ "train_tokens_per_second": 235.929
791
+ },
792
+ {
793
+ "epoch": 0.018698224852071007,
794
+ "grad_norm": 0.716282308101654,
795
+ "learning_rate": 0.00012324324324324327,
796
+ "loss": 0.2734939754009247,
797
+ "num_input_tokens_seen": 74716,
798
+ "step": 79,
799
+ "train_runtime": 315.7612,
800
+ "train_tokens_per_second": 236.622
801
+ },
802
+ {
803
+ "epoch": 0.01893491124260355,
804
+ "grad_norm": 0.7010631561279297,
805
+ "learning_rate": 0.00012216216216216216,
806
+ "loss": 0.2774362564086914,
807
+ "num_input_tokens_seen": 75692,
808
+ "step": 80,
809
+ "train_runtime": 318.7983,
810
+ "train_tokens_per_second": 237.429
811
+ },
812
+ {
813
+ "epoch": 0.019171597633136094,
814
+ "grad_norm": 0.6561240553855896,
815
+ "learning_rate": 0.00012108108108108109,
816
+ "loss": 0.21640686690807343,
817
+ "num_input_tokens_seen": 76558,
818
+ "step": 81,
819
+ "train_runtime": 321.889,
820
+ "train_tokens_per_second": 237.84
821
+ },
822
+ {
823
+ "epoch": 0.019408284023668638,
824
+ "grad_norm": 0.5552831888198853,
825
+ "learning_rate": 0.00012,
826
+ "loss": 0.21232956647872925,
827
+ "num_input_tokens_seen": 77408,
828
+ "step": 82,
829
+ "train_runtime": 325.0469,
830
+ "train_tokens_per_second": 238.144
831
+ },
832
+ {
833
+ "epoch": 0.019644970414201185,
834
+ "grad_norm": 0.5905120372772217,
835
+ "learning_rate": 0.00011891891891891893,
836
+ "loss": 0.22443550825119019,
837
+ "num_input_tokens_seen": 78278,
838
+ "step": 83,
839
+ "train_runtime": 328.099,
840
+ "train_tokens_per_second": 238.58
841
+ },
842
+ {
843
+ "epoch": 0.01988165680473373,
844
+ "grad_norm": 0.6606366634368896,
845
+ "learning_rate": 0.00011783783783783784,
846
+ "loss": 0.2400747388601303,
847
+ "num_input_tokens_seen": 79262,
848
+ "step": 84,
849
+ "train_runtime": 331.1361,
850
+ "train_tokens_per_second": 239.364
851
+ },
852
+ {
853
+ "epoch": 0.020118343195266272,
854
+ "grad_norm": 0.6508055329322815,
855
+ "learning_rate": 0.00011675675675675676,
856
+ "loss": 0.19072499871253967,
857
+ "num_input_tokens_seen": 80144,
858
+ "step": 85,
859
+ "train_runtime": 334.2158,
860
+ "train_tokens_per_second": 239.797
861
+ },
862
+ {
863
+ "epoch": 0.020355029585798816,
864
+ "grad_norm": 0.606562614440918,
865
+ "learning_rate": 0.00011567567567567568,
866
+ "loss": 0.20951862633228302,
867
+ "num_input_tokens_seen": 81108,
868
+ "step": 86,
869
+ "train_runtime": 337.3105,
870
+ "train_tokens_per_second": 240.455
871
+ },
872
+ {
873
+ "epoch": 0.02059171597633136,
874
+ "grad_norm": 0.6757521629333496,
875
+ "learning_rate": 0.00011459459459459461,
876
+ "loss": 0.17614495754241943,
877
+ "num_input_tokens_seen": 82032,
878
+ "step": 87,
879
+ "train_runtime": 340.3438,
880
+ "train_tokens_per_second": 241.027
881
+ },
882
+ {
883
+ "epoch": 0.020828402366863907,
884
+ "grad_norm": 0.671481728553772,
885
+ "learning_rate": 0.00011351351351351351,
886
+ "loss": 0.1901710331439972,
887
+ "num_input_tokens_seen": 82954,
888
+ "step": 88,
889
+ "train_runtime": 343.4059,
890
+ "train_tokens_per_second": 241.563
891
+ },
892
+ {
893
+ "epoch": 0.02106508875739645,
894
+ "grad_norm": 0.5919027924537659,
895
+ "learning_rate": 0.00011243243243243244,
896
+ "loss": 0.1563737988471985,
897
+ "num_input_tokens_seen": 83880,
898
+ "step": 89,
899
+ "train_runtime": 346.5125,
900
+ "train_tokens_per_second": 242.069
901
+ },
902
+ {
903
+ "epoch": 0.021301775147928994,
904
+ "grad_norm": 0.7108309268951416,
905
+ "learning_rate": 0.00011135135135135135,
906
+ "loss": 0.17256894707679749,
907
+ "num_input_tokens_seen": 84852,
908
+ "step": 90,
909
+ "train_runtime": 349.6508,
910
+ "train_tokens_per_second": 242.676
911
+ },
912
+ {
913
+ "epoch": 0.021538461538461538,
914
+ "grad_norm": 0.8608420491218567,
915
+ "learning_rate": 0.00011027027027027029,
916
+ "loss": 0.1845530867576599,
917
+ "num_input_tokens_seen": 85820,
918
+ "step": 91,
919
+ "train_runtime": 356.267,
920
+ "train_tokens_per_second": 240.887
921
+ },
922
+ {
923
+ "epoch": 0.02177514792899408,
924
+ "grad_norm": 0.9677010774612427,
925
+ "learning_rate": 0.00010918918918918919,
926
+ "loss": 0.24358105659484863,
927
+ "num_input_tokens_seen": 86770,
928
+ "step": 92,
929
+ "train_runtime": 359.3722,
930
+ "train_tokens_per_second": 241.449
931
+ },
932
+ {
933
+ "epoch": 0.02201183431952663,
934
+ "grad_norm": 0.6302966475486755,
935
+ "learning_rate": 0.00010810810810810812,
936
+ "loss": 0.18155133724212646,
937
+ "num_input_tokens_seen": 87736,
938
+ "step": 93,
939
+ "train_runtime": 362.5874,
940
+ "train_tokens_per_second": 241.972
941
+ },
942
+ {
943
+ "epoch": 0.022248520710059172,
944
+ "grad_norm": 0.7171812653541565,
945
+ "learning_rate": 0.00010702702702702702,
946
+ "loss": 0.15349116921424866,
947
+ "num_input_tokens_seen": 88626,
948
+ "step": 94,
949
+ "train_runtime": 365.7055,
950
+ "train_tokens_per_second": 242.343
951
+ },
952
+ {
953
+ "epoch": 0.022485207100591716,
954
+ "grad_norm": 0.7326887845993042,
955
+ "learning_rate": 0.00010594594594594595,
956
+ "loss": 0.18243758380413055,
957
+ "num_input_tokens_seen": 89548,
958
+ "step": 95,
959
+ "train_runtime": 368.799,
960
+ "train_tokens_per_second": 242.81
961
+ },
962
+ {
963
+ "epoch": 0.02272189349112426,
964
+ "grad_norm": 0.6686019897460938,
965
+ "learning_rate": 0.00010486486486486487,
966
+ "loss": 0.15622465312480927,
967
+ "num_input_tokens_seen": 90504,
968
+ "step": 96,
969
+ "train_runtime": 371.8404,
970
+ "train_tokens_per_second": 243.395
971
+ },
972
+ {
973
+ "epoch": 0.022958579881656806,
974
+ "grad_norm": 0.6306503415107727,
975
+ "learning_rate": 0.0001037837837837838,
976
+ "loss": 0.1272117793560028,
977
+ "num_input_tokens_seen": 91454,
978
+ "step": 97,
979
+ "train_runtime": 375.0199,
980
+ "train_tokens_per_second": 243.864
981
+ },
982
+ {
983
+ "epoch": 0.02319526627218935,
984
+ "grad_norm": 0.6068493127822876,
985
+ "learning_rate": 0.0001027027027027027,
986
+ "loss": 0.1577155441045761,
987
+ "num_input_tokens_seen": 92466,
988
+ "step": 98,
989
+ "train_runtime": 378.0978,
990
+ "train_tokens_per_second": 244.556
991
+ },
992
+ {
993
+ "epoch": 0.023431952662721894,
994
+ "grad_norm": 0.7263635993003845,
995
+ "learning_rate": 0.00010162162162162163,
996
+ "loss": 0.1845340132713318,
997
+ "num_input_tokens_seen": 93398,
998
+ "step": 99,
999
+ "train_runtime": 381.1358,
1000
+ "train_tokens_per_second": 245.052
1001
+ },
1002
+ {
1003
+ "epoch": 0.023668639053254437,
1004
+ "grad_norm": 0.6466794013977051,
1005
+ "learning_rate": 0.00010054054054054053,
1006
+ "loss": 0.14443959295749664,
1007
+ "num_input_tokens_seen": 94230,
1008
+ "step": 100,
1009
+ "train_runtime": 384.2906,
1010
+ "train_tokens_per_second": 245.205
1011
+ },
1012
+ {
1013
+ "epoch": 0.02390532544378698,
1014
+ "grad_norm": 0.6924043297767639,
1015
+ "learning_rate": 9.945945945945948e-05,
1016
+ "loss": 0.18473166227340698,
1017
+ "num_input_tokens_seen": 95132,
1018
+ "step": 101,
1019
+ "train_runtime": 387.5005,
1020
+ "train_tokens_per_second": 245.502
1021
+ },
1022
+ {
1023
+ "epoch": 0.024142011834319528,
1024
+ "grad_norm": 0.732337236404419,
1025
+ "learning_rate": 9.837837837837839e-05,
1026
+ "loss": 0.16353662312030792,
1027
+ "num_input_tokens_seen": 96080,
1028
+ "step": 102,
1029
+ "train_runtime": 390.584,
1030
+ "train_tokens_per_second": 245.991
1031
+ },
1032
+ {
1033
+ "epoch": 0.02437869822485207,
1034
+ "grad_norm": 0.5956932902336121,
1035
+ "learning_rate": 9.729729729729731e-05,
1036
+ "loss": 0.16359847784042358,
1037
+ "num_input_tokens_seen": 97048,
1038
+ "step": 103,
1039
+ "train_runtime": 393.6474,
1040
+ "train_tokens_per_second": 246.535
1041
+ },
1042
+ {
1043
+ "epoch": 0.024615384615384615,
1044
+ "grad_norm": 0.6137388348579407,
1045
+ "learning_rate": 9.621621621621622e-05,
1046
+ "loss": 0.15779417753219604,
1047
+ "num_input_tokens_seen": 97986,
1048
+ "step": 104,
1049
+ "train_runtime": 396.7042,
1050
+ "train_tokens_per_second": 247.0
1051
+ },
1052
+ {
1053
+ "epoch": 0.02485207100591716,
1054
+ "grad_norm": 0.7936599254608154,
1055
+ "learning_rate": 9.513513513513514e-05,
1056
+ "loss": 0.21357379853725433,
1057
+ "num_input_tokens_seen": 98894,
1058
+ "step": 105,
1059
+ "train_runtime": 399.8228,
1060
+ "train_tokens_per_second": 247.345
1061
+ },
1062
+ {
1063
+ "epoch": 0.025088757396449703,
1064
+ "grad_norm": 0.6574507355690002,
1065
+ "learning_rate": 9.405405405405407e-05,
1066
+ "loss": 0.1685573309659958,
1067
+ "num_input_tokens_seen": 99856,
1068
+ "step": 106,
1069
+ "train_runtime": 402.8867,
1070
+ "train_tokens_per_second": 247.851
1071
+ },
1072
+ {
1073
+ "epoch": 0.02532544378698225,
1074
+ "grad_norm": 0.5891634821891785,
1075
+ "learning_rate": 9.297297297297299e-05,
1076
+ "loss": 0.12082140147686005,
1077
+ "num_input_tokens_seen": 100810,
1078
+ "step": 107,
1079
+ "train_runtime": 405.9752,
1080
+ "train_tokens_per_second": 248.316
1081
+ },
1082
+ {
1083
+ "epoch": 0.025562130177514793,
1084
+ "grad_norm": 0.6624342799186707,
1085
+ "learning_rate": 9.18918918918919e-05,
1086
+ "loss": 0.1473795771598816,
1087
+ "num_input_tokens_seen": 101744,
1088
+ "step": 108,
1089
+ "train_runtime": 409.0084,
1090
+ "train_tokens_per_second": 248.758
1091
+ },
1092
+ {
1093
+ "epoch": 0.025798816568047337,
1094
+ "grad_norm": 0.5968036651611328,
1095
+ "learning_rate": 9.081081081081082e-05,
1096
+ "loss": 0.12923751771450043,
1097
+ "num_input_tokens_seen": 102702,
1098
+ "step": 109,
1099
+ "train_runtime": 412.199,
1100
+ "train_tokens_per_second": 249.156
1101
+ },
1102
+ {
1103
+ "epoch": 0.02603550295857988,
1104
+ "grad_norm": 0.5954580307006836,
1105
+ "learning_rate": 8.972972972972973e-05,
1106
+ "loss": 0.1161399856209755,
1107
+ "num_input_tokens_seen": 103672,
1108
+ "step": 110,
1109
+ "train_runtime": 415.3736,
1110
+ "train_tokens_per_second": 249.587
1111
+ },
1112
+ {
1113
+ "epoch": 0.026272189349112424,
1114
+ "grad_norm": 0.6139379739761353,
1115
+ "learning_rate": 8.864864864864866e-05,
1116
+ "loss": 0.12352899461984634,
1117
+ "num_input_tokens_seen": 104598,
1118
+ "step": 111,
1119
+ "train_runtime": 418.3271,
1120
+ "train_tokens_per_second": 250.039
1121
+ },
1122
+ {
1123
+ "epoch": 0.02650887573964497,
1124
+ "grad_norm": 0.7933095097541809,
1125
+ "learning_rate": 8.756756756756758e-05,
1126
+ "loss": 0.1709994524717331,
1127
+ "num_input_tokens_seen": 105580,
1128
+ "step": 112,
1129
+ "train_runtime": 421.3565,
1130
+ "train_tokens_per_second": 250.572
1131
+ },
1132
+ {
1133
+ "epoch": 0.026745562130177515,
1134
+ "grad_norm": 0.7641370892524719,
1135
+ "learning_rate": 8.64864864864865e-05,
1136
+ "loss": 0.13086311519145966,
1137
+ "num_input_tokens_seen": 106444,
1138
+ "step": 113,
1139
+ "train_runtime": 424.4668,
1140
+ "train_tokens_per_second": 250.771
1141
+ },
1142
+ {
1143
+ "epoch": 0.02698224852071006,
1144
+ "grad_norm": 0.7611938118934631,
1145
+ "learning_rate": 8.540540540540541e-05,
1146
+ "loss": 0.16537640988826752,
1147
+ "num_input_tokens_seen": 107454,
1148
+ "step": 114,
1149
+ "train_runtime": 427.5,
1150
+ "train_tokens_per_second": 251.354
1151
+ },
1152
+ {
1153
+ "epoch": 0.027218934911242602,
1154
+ "grad_norm": 0.7495304346084595,
1155
+ "learning_rate": 8.432432432432433e-05,
1156
+ "loss": 0.14228317141532898,
1157
+ "num_input_tokens_seen": 108426,
1158
+ "step": 115,
1159
+ "train_runtime": 430.5035,
1160
+ "train_tokens_per_second": 251.859
1161
+ },
1162
+ {
1163
+ "epoch": 0.02745562130177515,
1164
+ "grad_norm": 1.2078282833099365,
1165
+ "learning_rate": 8.324324324324326e-05,
1166
+ "loss": 0.10530930012464523,
1167
+ "num_input_tokens_seen": 109446,
1168
+ "step": 116,
1169
+ "train_runtime": 433.4964,
1170
+ "train_tokens_per_second": 252.473
1171
+ },
1172
+ {
1173
+ "epoch": 0.027692307692307693,
1174
+ "grad_norm": 0.6765572428703308,
1175
+ "learning_rate": 8.216216216216217e-05,
1176
+ "loss": 0.10717278718948364,
1177
+ "num_input_tokens_seen": 110374,
1178
+ "step": 117,
1179
+ "train_runtime": 436.5901,
1180
+ "train_tokens_per_second": 252.809
1181
+ },
1182
+ {
1183
+ "epoch": 0.027928994082840237,
1184
+ "grad_norm": 0.8501551151275635,
1185
+ "learning_rate": 8.108108108108109e-05,
1186
+ "loss": 0.17784273624420166,
1187
+ "num_input_tokens_seen": 111340,
1188
+ "step": 118,
1189
+ "train_runtime": 439.6526,
1190
+ "train_tokens_per_second": 253.245
1191
+ },
1192
+ {
1193
+ "epoch": 0.02816568047337278,
1194
+ "grad_norm": 0.7583926320075989,
1195
+ "learning_rate": 8e-05,
1196
+ "loss": 0.16588062047958374,
1197
+ "num_input_tokens_seen": 112288,
1198
+ "step": 119,
1199
+ "train_runtime": 442.8046,
1200
+ "train_tokens_per_second": 253.584
1201
+ },
1202
+ {
1203
+ "epoch": 0.028402366863905324,
1204
+ "grad_norm": 0.6620173454284668,
1205
+ "learning_rate": 7.891891891891892e-05,
1206
+ "loss": 0.11676295101642609,
1207
+ "num_input_tokens_seen": 113206,
1208
+ "step": 120,
1209
+ "train_runtime": 445.9682,
1210
+ "train_tokens_per_second": 253.843
1211
+ }
1212
+ ],
1213
+ "logging_steps": 1,
1214
+ "max_steps": 190,
1215
+ "num_input_tokens_seen": 113206,
1216
+ "num_train_epochs": 1,
1217
+ "save_steps": 30,
1218
+ "stateful_callbacks": {
1219
+ "TrainerControl": {
1220
+ "args": {
1221
+ "should_epoch_stop": false,
1222
+ "should_evaluate": false,
1223
+ "should_log": false,
1224
+ "should_save": true,
1225
+ "should_training_stop": false
1226
+ },
1227
+ "attributes": {}
1228
+ }
1229
+ },
1230
+ "total_flos": 1632037070282880.0,
1231
+ "train_batch_size": 2,
1232
+ "trial_name": null,
1233
+ "trial_params": null
1234
+ }
checkpoint-120/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:34c4ca6d50f04301fb38588516a91421e14a8ad1ffddf12afdfd9ba35cd4b64e
3
+ size 5777
checkpoint-150/README.md ADDED
@@ -0,0 +1,210 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: unsloth/gemma-4-e2b-it-unsloth-bnb-4bit
3
+ library_name: peft
4
+ pipeline_tag: text-generation
5
+ tags:
6
+ - base_model:adapter:unsloth/gemma-4-e2b-it-unsloth-bnb-4bit
7
+ - lora
8
+ - sft
9
+ - transformers
10
+ - trl
11
+ - unsloth
12
+ ---
13
+
14
+ # Model Card for Model ID
15
+
16
+ <!-- Provide a quick summary of what the model is/does. -->
17
+
18
+
19
+
20
+ ## Model Details
21
+
22
+ ### Model Description
23
+
24
+ <!-- Provide a longer summary of what this model is. -->
25
+
26
+
27
+
28
+ - **Developed by:** [More Information Needed]
29
+ - **Funded by [optional]:** [More Information Needed]
30
+ - **Shared by [optional]:** [More Information Needed]
31
+ - **Model type:** [More Information Needed]
32
+ - **Language(s) (NLP):** [More Information Needed]
33
+ - **License:** [More Information Needed]
34
+ - **Finetuned from model [optional]:** [More Information Needed]
35
+
36
+ ### Model Sources [optional]
37
+
38
+ <!-- Provide the basic links for the model. -->
39
+
40
+ - **Repository:** [More Information Needed]
41
+ - **Paper [optional]:** [More Information Needed]
42
+ - **Demo [optional]:** [More Information Needed]
43
+
44
+ ## Uses
45
+
46
+ <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
47
+
48
+ ### Direct Use
49
+
50
+ <!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
51
+
52
+ [More Information Needed]
53
+
54
+ ### Downstream Use [optional]
55
+
56
+ <!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
57
+
58
+ [More Information Needed]
59
+
60
+ ### Out-of-Scope Use
61
+
62
+ <!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
63
+
64
+ [More Information Needed]
65
+
66
+ ## Bias, Risks, and Limitations
67
+
68
+ <!-- This section is meant to convey both technical and sociotechnical limitations. -->
69
+
70
+ [More Information Needed]
71
+
72
+ ### Recommendations
73
+
74
+ <!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
75
+
76
+ Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
77
+
78
+ ## How to Get Started with the Model
79
+
80
+ Use the code below to get started with the model.
81
+
82
+ [More Information Needed]
83
+
84
+ ## Training Details
85
+
86
+ ### Training Data
87
+
88
+ <!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
89
+
90
+ [More Information Needed]
91
+
92
+ ### Training Procedure
93
+
94
+ <!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
95
+
96
+ #### Preprocessing [optional]
97
+
98
+ [More Information Needed]
99
+
100
+
101
+ #### Training Hyperparameters
102
+
103
+ - **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
104
+
105
+ #### Speeds, Sizes, Times [optional]
106
+
107
+ <!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
108
+
109
+ [More Information Needed]
110
+
111
+ ## Evaluation
112
+
113
+ <!-- This section describes the evaluation protocols and provides the results. -->
114
+
115
+ ### Testing Data, Factors & Metrics
116
+
117
+ #### Testing Data
118
+
119
+ <!-- This should link to a Dataset Card if possible. -->
120
+
121
+ [More Information Needed]
122
+
123
+ #### Factors
124
+
125
+ <!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
126
+
127
+ [More Information Needed]
128
+
129
+ #### Metrics
130
+
131
+ <!-- These are the evaluation metrics being used, ideally with a description of why. -->
132
+
133
+ [More Information Needed]
134
+
135
+ ### Results
136
+
137
+ [More Information Needed]
138
+
139
+ #### Summary
140
+
141
+
142
+
143
+ ## Model Examination [optional]
144
+
145
+ <!-- Relevant interpretability work for the model goes here -->
146
+
147
+ [More Information Needed]
148
+
149
+ ## Environmental Impact
150
+
151
+ <!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
152
+
153
+ Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
154
+
155
+ - **Hardware Type:** [More Information Needed]
156
+ - **Hours used:** [More Information Needed]
157
+ - **Cloud Provider:** [More Information Needed]
158
+ - **Compute Region:** [More Information Needed]
159
+ - **Carbon Emitted:** [More Information Needed]
160
+
161
+ ## Technical Specifications [optional]
162
+
163
+ ### Model Architecture and Objective
164
+
165
+ [More Information Needed]
166
+
167
+ ### Compute Infrastructure
168
+
169
+ [More Information Needed]
170
+
171
+ #### Hardware
172
+
173
+ [More Information Needed]
174
+
175
+ #### Software
176
+
177
+ [More Information Needed]
178
+
179
+ ## Citation [optional]
180
+
181
+ <!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
182
+
183
+ **BibTeX:**
184
+
185
+ [More Information Needed]
186
+
187
+ **APA:**
188
+
189
+ [More Information Needed]
190
+
191
+ ## Glossary [optional]
192
+
193
+ <!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
194
+
195
+ [More Information Needed]
196
+
197
+ ## More Information [optional]
198
+
199
+ [More Information Needed]
200
+
201
+ ## Model Card Authors [optional]
202
+
203
+ [More Information Needed]
204
+
205
+ ## Model Card Contact
206
+
207
+ [More Information Needed]
208
+ ### Framework versions
209
+
210
+ - PEFT 0.18.1
checkpoint-150/adapter_config.json ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alora_invocation_tokens": null,
3
+ "alpha_pattern": {},
4
+ "arrow_config": null,
5
+ "auto_mapping": {
6
+ "base_model_class": "Gemma4ForConditionalGeneration",
7
+ "parent_library": "transformers.models.gemma4.modeling_gemma4",
8
+ "unsloth_fixed": true
9
+ },
10
+ "base_model_name_or_path": "unsloth/gemma-4-e2b-it-unsloth-bnb-4bit",
11
+ "bias": "none",
12
+ "corda_config": null,
13
+ "ensure_weight_tying": false,
14
+ "eva_config": null,
15
+ "exclude_modules": null,
16
+ "fan_in_fan_out": false,
17
+ "inference_mode": true,
18
+ "init_lora_weights": true,
19
+ "layer_replication": null,
20
+ "layers_pattern": null,
21
+ "layers_to_transform": null,
22
+ "loftq_config": {},
23
+ "lora_alpha": 32,
24
+ "lora_bias": false,
25
+ "lora_dropout": 0.0,
26
+ "megatron_config": null,
27
+ "megatron_core": "megatron.core",
28
+ "modules_to_save": null,
29
+ "peft_type": "LORA",
30
+ "peft_version": "0.18.1",
31
+ "qalora_group_size": 16,
32
+ "r": 16,
33
+ "rank_pattern": {},
34
+ "revision": null,
35
+ "target_modules": [
36
+ "q_proj",
37
+ "gate_proj",
38
+ "v_proj",
39
+ "k_proj",
40
+ "down_proj",
41
+ "o_proj",
42
+ "up_proj"
43
+ ],
44
+ "target_parameters": null,
45
+ "task_type": "CAUSAL_LM",
46
+ "trainable_token_indices": null,
47
+ "use_dora": false,
48
+ "use_qalora": false,
49
+ "use_rslora": false
50
+ }
checkpoint-150/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:79025ee3acf1d35d8aa435821bda964bc6b3cb6801665ea77bdff85aaa1067c2
3
+ size 124277728
checkpoint-150/chat_template.jinja ADDED
@@ -0,0 +1,351 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {%- macro format_parameters(properties, required, filter_keys=false) -%}
2
+ {%- set standard_keys = ['description', 'type', 'properties', 'required', 'nullable'] -%}
3
+ {%- set ns = namespace(found_first=false) -%}
4
+ {%- for key, value in properties | dictsort -%}
5
+ {%- set add_comma = false -%}
6
+ {%- if not filter_keys or key not in standard_keys -%}
7
+ {%- if ns.found_first %},{% endif -%}
8
+ {%- set ns.found_first = true -%}
9
+ {{ key }}:{
10
+ {%- if value['description'] -%}
11
+ description:<|"|>{{ value['description'] }}<|"|>
12
+ {%- set add_comma = true -%}
13
+ {%- endif -%}
14
+ {%- if value['type'] | upper == 'STRING' -%}
15
+ {%- if value['enum'] -%}
16
+ {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
17
+ enum:{{ format_argument(value['enum']) }}
18
+ {%- endif -%}
19
+ {%- elif value['type'] | upper == 'ARRAY' -%}
20
+ {%- if value['items'] is mapping and value['items'] -%}
21
+ {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
22
+ items:{
23
+ {%- set ns_items = namespace(found_first=false) -%}
24
+ {%- for item_key, item_value in value['items'] | dictsort -%}
25
+ {%- if item_value is not none -%}
26
+ {%- if ns_items.found_first %},{% endif -%}
27
+ {%- set ns_items.found_first = true -%}
28
+ {%- if item_key == 'properties' -%}
29
+ properties:{
30
+ {%- if item_value is mapping -%}
31
+ {{- format_parameters(item_value, value['items']['required'] | default([])) -}}
32
+ {%- endif -%}
33
+ }
34
+ {%- elif item_key == 'required' -%}
35
+ required:[
36
+ {%- for req_item in item_value -%}
37
+ <|"|>{{- req_item -}}<|"|>
38
+ {%- if not loop.last %},{% endif -%}
39
+ {%- endfor -%}
40
+ ]
41
+ {%- elif item_key == 'type' -%}
42
+ {%- if item_value is string -%}
43
+ type:{{ format_argument(item_value | upper) }}
44
+ {%- else -%}
45
+ type:{{ format_argument(item_value | map('upper') | list) }}
46
+ {%- endif -%}
47
+ {%- else -%}
48
+ {{ item_key }}:{{ format_argument(item_value) }}
49
+ {%- endif -%}
50
+ {%- endif -%}
51
+ {%- endfor -%}
52
+ }
53
+ {%- endif -%}
54
+ {%- endif -%}
55
+ {%- if value['nullable'] %}
56
+ {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
57
+ nullable:true
58
+ {%- endif -%}
59
+ {%- if value['type'] | upper == 'OBJECT' -%}
60
+ {%- if value['properties'] is defined and value['properties'] is mapping -%}
61
+ {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
62
+ properties:{
63
+ {{- format_parameters(value['properties'], value['required'] | default([])) -}}
64
+ }
65
+ {%- elif value is mapping -%}
66
+ {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
67
+ properties:{
68
+ {{- format_parameters(value, value['required'] | default([]), filter_keys=true) -}}
69
+ }
70
+ {%- endif -%}
71
+ {%- if value['required'] -%}
72
+ {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
73
+ required:[
74
+ {%- for item in value['required'] | default([]) -%}
75
+ <|"|>{{- item -}}<|"|>
76
+ {%- if not loop.last %},{% endif -%}
77
+ {%- endfor -%}
78
+ ]
79
+ {%- endif -%}
80
+ {%- endif -%}
81
+ {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
82
+ type:<|"|>{{ value['type'] | upper }}<|"|>}
83
+ {%- endif -%}
84
+ {%- endfor -%}
85
+ {%- endmacro -%}
86
+ {%- macro format_function_declaration(tool_data) -%}
87
+ declaration:{{- tool_data['function']['name'] -}}{description:<|"|>{{- tool_data['function']['description'] -}}<|"|>
88
+ {%- set params = tool_data['function']['parameters'] -%}
89
+ {%- if params -%}
90
+ ,parameters:{
91
+ {%- if params['properties'] -%}
92
+ properties:{ {{- format_parameters(params['properties'], params['required']) -}} },
93
+ {%- endif -%}
94
+ {%- if params['required'] -%}
95
+ required:[
96
+ {%- for item in params['required'] -%}
97
+ <|"|>{{- item -}}<|"|>
98
+ {{- ',' if not loop.last -}}
99
+ {%- endfor -%}
100
+ ],
101
+ {%- endif -%}
102
+ {%- if params['type'] -%}
103
+ type:<|"|>{{- params['type'] | upper -}}<|"|>}
104
+ {%- endif -%}
105
+ {%- endif -%}
106
+ {%- if 'response' in tool_data['function'] -%}
107
+ {%- set response_declaration = tool_data['function']['response'] -%}
108
+ ,response:{
109
+ {%- if response_declaration['description'] -%}
110
+ description:<|"|>{{- response_declaration['description'] -}}<|"|>,
111
+ {%- endif -%}
112
+ {%- if response_declaration['type'] | upper == 'OBJECT' -%}
113
+ type:<|"|>{{- response_declaration['type'] | upper -}}<|"|>}
114
+ {%- endif -%}
115
+ {%- endif -%}
116
+ }
117
+ {%- endmacro -%}
118
+ {%- macro format_argument(argument, escape_keys=True) -%}
119
+ {%- if argument is string -%}
120
+ {{- '<|"|>' + argument + '<|"|>' -}}
121
+ {%- elif argument is boolean -%}
122
+ {{- 'true' if argument else 'false' -}}
123
+ {%- elif argument is mapping -%}
124
+ {{- '{' -}}
125
+ {%- set ns = namespace(found_first=false) -%}
126
+ {%- for key, value in argument | dictsort -%}
127
+ {%- if ns.found_first %},{% endif -%}
128
+ {%- set ns.found_first = true -%}
129
+ {%- if escape_keys -%}
130
+ {{- '<|"|>' + key + '<|"|>' -}}
131
+ {%- else -%}
132
+ {{- key -}}
133
+ {%- endif -%}
134
+ :{{- format_argument(value, escape_keys=escape_keys) -}}
135
+ {%- endfor -%}
136
+ {{- '}' -}}
137
+ {%- elif argument is sequence -%}
138
+ {{- '[' -}}
139
+ {%- for item in argument -%}
140
+ {{- format_argument(item, escape_keys=escape_keys) -}}
141
+ {%- if not loop.last %},{% endif -%}
142
+ {%- endfor -%}
143
+ {{- ']' -}}
144
+ {%- else -%}
145
+ {{- argument -}}
146
+ {%- endif -%}
147
+ {%- endmacro -%}
148
+ {%- macro strip_thinking(text) -%}
149
+ {%- set ns = namespace(result='') -%}
150
+ {%- for part in text.split('<channel|>') -%}
151
+ {%- if '<|channel>' in part -%}
152
+ {%- set ns.result = ns.result + part.split('<|channel>')[0] -%}
153
+ {%- else -%}
154
+ {%- set ns.result = ns.result + part -%}
155
+ {%- endif -%}
156
+ {%- endfor -%}
157
+ {{- ns.result | trim -}}
158
+ {%- endmacro -%}
159
+
160
+ {%- macro format_tool_response_block(tool_name, response) -%}
161
+ {{- '<|tool_response>' -}}
162
+ {%- if response is mapping -%}
163
+ {{- 'response:' + tool_name + '{' -}}
164
+ {%- for key, value in response | dictsort -%}
165
+ {{- key -}}:{{- format_argument(value, escape_keys=False) -}}
166
+ {%- if not loop.last %},{% endif -%}
167
+ {%- endfor -%}
168
+ {{- '}' -}}
169
+ {%- else -%}
170
+ {{- 'response:' + tool_name + '{value:' + format_argument(response, escape_keys=False) + '}' -}}
171
+ {%- endif -%}
172
+ {{- '<tool_response|>' -}}
173
+ {%- endmacro -%}
174
+
175
+ {%- set ns = namespace(prev_message_type=None) -%}
176
+ {%- set loop_messages = messages -%}
177
+ {{- bos_token -}}
178
+ {#- Handle System/Tool Definitions Block -#}
179
+ {%- if (enable_thinking is defined and enable_thinking) or tools or messages[0]['role'] in ['system', 'developer'] -%}
180
+ {{- '<|turn>system\n' -}}
181
+ {#- Inject Thinking token at the very top of the FIRST system turn -#}
182
+ {%- if enable_thinking is defined and enable_thinking -%}
183
+ {{- '<|think|>\n' -}}
184
+ {%- set ns.prev_message_type = 'think' -%}
185
+ {%- endif -%}
186
+ {%- if messages[0]['role'] in ['system', 'developer'] -%}
187
+ {%- if messages[0]['content'] is string -%}
188
+ {{- messages[0]['content'] | trim -}}
189
+ {%- elif messages[0]['content'] is sequence -%}
190
+ {%- for item in messages[0]['content'] -%}
191
+ {{- item['text'] | trim + ' '-}}
192
+ {%- endfor -%}
193
+ {%- endif -%}
194
+ {%- set loop_messages = messages[1:] -%}
195
+ {%- endif -%}
196
+ {%- if tools -%}
197
+ {%- for tool in tools %}
198
+ {{- '<|tool>' -}}
199
+ {{- format_function_declaration(tool) | trim -}}
200
+ {{- '<tool|>' -}}
201
+ {%- endfor %}
202
+ {%- set ns.prev_message_type = 'tool' -%}
203
+ {%- endif -%}
204
+ {{- '<turn|>\n' -}}
205
+ {%- endif %}
206
+
207
+ {#- Pre-scan: find last user message index for reasoning guard -#}
208
+ {%- set ns_turn = namespace(last_user_idx=-1) -%}
209
+ {%- for i in range(loop_messages | length) -%}
210
+ {%- if loop_messages[i]['role'] == 'user' -%}
211
+ {%- set ns_turn.last_user_idx = i -%}
212
+ {%- endif -%}
213
+ {%- endfor -%}
214
+
215
+ {#- Loop through messages -#}
216
+ {%- for message in loop_messages -%}
217
+ {%- if message['role'] != 'tool' -%}
218
+ {%- set ns.prev_message_type = None -%}
219
+ {%- set role = 'model' if message['role'] == 'assistant' else message['role'] -%}
220
+ {#- Detect continuation: suppress duplicate <|turn>model when previous non-tool message was also assistant -#}
221
+ {%- set prev_nt = namespace(role=None, found=false) -%}
222
+ {%- if loop.index0 > 0 -%}
223
+ {%- for j in range(loop.index0 - 1, -1, -1) -%}
224
+ {%- if not prev_nt.found -%}
225
+ {%- if loop_messages[j]['role'] != 'tool' -%}
226
+ {%- set prev_nt.role = loop_messages[j]['role'] -%}
227
+ {%- set prev_nt.found = true -%}
228
+ {%- endif -%}
229
+ {%- endif -%}
230
+ {%- endfor -%}
231
+ {%- endif -%}
232
+ {%- set continue_same_model_turn = (role == 'model' and prev_nt.role == 'assistant') -%}
233
+ {%- if not continue_same_model_turn -%}
234
+ {{- '<|turn>' + role + '\n' }}
235
+ {%- endif -%}
236
+
237
+ {#- Render reasoning/reasoning_content as thinking channel -#}
238
+ {%- set thinking_text = message.get('reasoning') or message.get('reasoning_content') -%}
239
+ {%- if thinking_text and loop.index0 > ns_turn.last_user_idx and message.get('tool_calls') -%}
240
+ {{- '<|channel>thought\n' + thinking_text + '\n<channel|>' -}}
241
+ {%- endif -%}
242
+
243
+ {%- if message['tool_calls'] -%}
244
+ {%- for tool_call in message['tool_calls'] -%}
245
+ {%- set function = tool_call['function'] -%}
246
+ {{- '<|tool_call>call:' + function['name'] + '{' -}}
247
+ {%- if function['arguments'] is mapping -%}
248
+ {%- set ns_args = namespace(found_first=false) -%}
249
+ {%- for key, value in function['arguments'] | dictsort -%}
250
+ {%- if ns_args.found_first %},{% endif -%}
251
+ {%- set ns_args.found_first = true -%}
252
+ {{- key -}}:{{- format_argument(value, escape_keys=False) -}}
253
+ {%- endfor -%}
254
+ {%- elif function['arguments'] is string -%}
255
+ {{- function['arguments'] -}}
256
+ {%- endif -%}
257
+ {{- '}<tool_call|>' -}}
258
+ {%- endfor -%}
259
+ {%- set ns.prev_message_type = 'tool_call' -%}
260
+ {%- endif -%}
261
+
262
+ {%- set ns_tr_out = namespace(flag=false) -%}
263
+ {%- if message.get('tool_responses') -%}
264
+ {#- Legacy: tool_responses embedded on the assistant message (Google/Gemma native) -#}
265
+ {%- for tool_response in message['tool_responses'] -%}
266
+ {{- format_tool_response_block(tool_response['name'] | default('unknown'), tool_response['response']) -}}
267
+ {%- set ns_tr_out.flag = true -%}
268
+ {%- set ns.prev_message_type = 'tool_response' -%}
269
+ {%- endfor -%}
270
+ {%- elif message.get('tool_calls') -%}
271
+ {#- OpenAI Chat Completions: forward-scan consecutive role:tool messages -#}
272
+ {%- set ns_tool_scan = namespace(stopped=false) -%}
273
+ {%- for k in range(loop.index0 + 1, loop_messages | length) -%}
274
+ {%- if ns_tool_scan.stopped -%}
275
+ {%- elif loop_messages[k]['role'] != 'tool' -%}
276
+ {%- set ns_tool_scan.stopped = true -%}
277
+ {%- else -%}
278
+ {%- set follow = loop_messages[k] -%}
279
+ {#- Resolve tool_call_id to function name -#}
280
+ {%- set ns_tname = namespace(name=follow.get('name') | default('unknown')) -%}
281
+ {%- for tc in message['tool_calls'] -%}
282
+ {%- if tc.get('id') == follow.get('tool_call_id') -%}
283
+ {%- set ns_tname.name = tc['function']['name'] -%}
284
+ {%- endif -%}
285
+ {%- endfor -%}
286
+ {#- Handle content as string or content-parts array -#}
287
+ {%- set tool_body = follow.get('content') -%}
288
+ {%- if tool_body is string -%}
289
+ {{- format_tool_response_block(ns_tname.name, tool_body) -}}
290
+ {%- elif tool_body is sequence and tool_body is not string -%}
291
+ {%- set ns_txt = namespace(s='') -%}
292
+ {%- for part in tool_body -%}
293
+ {%- if part.get('type') == 'text' -%}
294
+ {%- set ns_txt.s = ns_txt.s + (part.get('text') | default('')) -%}
295
+ {%- endif -%}
296
+ {%- endfor -%}
297
+ {{- format_tool_response_block(ns_tname.name, ns_txt.s) -}}
298
+ {%- else -%}
299
+ {{- format_tool_response_block(ns_tname.name, tool_body) -}}
300
+ {%- endif -%}
301
+ {%- set ns_tr_out.flag = true -%}
302
+ {%- set ns.prev_message_type = 'tool_response' -%}
303
+ {%- endif -%}
304
+ {%- endfor -%}
305
+ {%- endif -%}
306
+
307
+ {%- set captured_content -%}
308
+ {%- if message['content'] is string -%}
309
+ {%- if role == 'model' -%}
310
+ {{- strip_thinking(message['content']) -}}
311
+ {%- else -%}
312
+ {{- message['content'] | trim -}}
313
+ {%- endif -%}
314
+ {%- elif message['content'] is sequence -%}
315
+ {%- for item in message['content'] -%}
316
+ {%- if item['type'] == 'text' -%}
317
+ {%- if role == 'model' -%}
318
+ {{- strip_thinking(item['text']) -}}
319
+ {%- else -%}
320
+ {{- item['text'] | trim -}}
321
+ {%- endif -%}
322
+ {%- elif item['type'] == 'image' -%}
323
+ {{- '<|image|>' -}}
324
+ {%- set ns.prev_message_type = 'image' -%}
325
+ {%- elif item['type'] == 'audio' -%}
326
+ {{- '<|audio|>' -}}
327
+ {%- set ns.prev_message_type = 'audio' -%}
328
+ {%- elif item['type'] == 'video' -%}
329
+ {{- '<|video|>' -}}
330
+ {%- set ns.prev_message_type = 'video' -%}
331
+ {%- endif -%}
332
+ {%- endfor -%}
333
+ {%- endif -%}
334
+ {%- endset -%}
335
+
336
+ {{- captured_content -}}
337
+ {%- set has_content = captured_content | trim | length > 0 -%}
338
+
339
+ {%- if ns.prev_message_type == 'tool_call' and not ns_tr_out.flag -%}
340
+ {{- '<|tool_response>' -}}
341
+ {%- elif not (ns_tr_out.flag and not has_content) -%}
342
+ {{- '<turn|>\n' -}}
343
+ {%- endif -%}
344
+ {%- endif -%}
345
+ {%- endfor -%}
346
+
347
+ {%- if add_generation_prompt -%}
348
+ {%- if ns.prev_message_type != 'tool_response' and ns.prev_message_type != 'tool_call' -%}
349
+ {{- '<|turn>model\n' -}}
350
+ {%- endif -%}
351
+ {%- endif -%}
checkpoint-150/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:676e7ff5f1af2f689ece698e6985cdaf684ab2bafbeeaa8681977a6453dffe22
3
+ size 52047117
checkpoint-150/processor_config.json ADDED
@@ -0,0 +1,75 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "audio_ms_per_token": 40,
3
+ "audio_seq_length": 750,
4
+ "feature_extractor": {
5
+ "dither": 0.0,
6
+ "feature_extractor_type": "Gemma4AudioFeatureExtractor",
7
+ "feature_size": 128,
8
+ "fft_length": 512,
9
+ "fft_overdrive": false,
10
+ "frame_length": 320,
11
+ "hop_length": 160,
12
+ "input_scale_factor": 1.0,
13
+ "max_frequency": 8000.0,
14
+ "mel_floor": 0.001,
15
+ "min_frequency": 0.0,
16
+ "padding_side": "left",
17
+ "padding_value": 0.0,
18
+ "per_bin_mean": null,
19
+ "per_bin_stddev": null,
20
+ "preemphasis": 0.0,
21
+ "preemphasis_htk_flavor": true,
22
+ "return_attention_mask": true,
23
+ "sampling_rate": 16000
24
+ },
25
+ "image_processor": {
26
+ "do_convert_rgb": true,
27
+ "do_normalize": false,
28
+ "do_rescale": true,
29
+ "do_resize": true,
30
+ "image_mean": [
31
+ 0.0,
32
+ 0.0,
33
+ 0.0
34
+ ],
35
+ "image_processor_type": "Gemma4ImageProcessor",
36
+ "image_seq_length": 280,
37
+ "image_std": [
38
+ 1.0,
39
+ 1.0,
40
+ 1.0
41
+ ],
42
+ "max_soft_tokens": 280,
43
+ "patch_size": 16,
44
+ "pooling_kernel_size": 3,
45
+ "resample": 3,
46
+ "rescale_factor": 0.00392156862745098
47
+ },
48
+ "image_seq_length": 280,
49
+ "processor_class": "Gemma4Processor",
50
+ "video_processor": {
51
+ "do_convert_rgb": true,
52
+ "do_normalize": true,
53
+ "do_rescale": true,
54
+ "do_resize": true,
55
+ "do_sample_frames": true,
56
+ "image_mean": [
57
+ 0.0,
58
+ 0.0,
59
+ 0.0
60
+ ],
61
+ "image_std": [
62
+ 1.0,
63
+ 1.0,
64
+ 1.0
65
+ ],
66
+ "max_soft_tokens": 70,
67
+ "num_frames": 32,
68
+ "patch_size": 16,
69
+ "pooling_kernel_size": 3,
70
+ "resample": 3,
71
+ "rescale_factor": 0.00392156862745098,
72
+ "return_metadata": false,
73
+ "video_processor_type": "Gemma4VideoProcessor"
74
+ }
75
+ }
checkpoint-150/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f1d565802a8e26c4e8a31328752b7a7fdc186d9401aa008e65697d0ad8c22e33
3
+ size 14645
checkpoint-150/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:41ce6f17608f83ea4a19655ee2f08447977bad8cf4b831826d8c03f86b5176e9
3
+ size 1383
checkpoint-150/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a02bbb7a0dfd9017bf6ee57d7f39ecba548a46e11e466823a59cfab4c980712f
3
+ size 1465
checkpoint-150/tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cc8d3a0ce36466ccc1278bf987df5f71db1719b9ca6b4118264f45cb627bfe0f
3
+ size 32169626
checkpoint-150/tokenizer_config.json ADDED
@@ -0,0 +1,289 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ "<|video|>"
18
+ ],
19
+ "image_token": "<|image|>",
20
+ "is_local": false,
21
+ "mask_token": "<mask>",
22
+ "model_max_length": 131072,
23
+ "model_specific_special_tokens": {
24
+ "audio_token": "<|audio|>",
25
+ "boa_token": "<|audio>",
26
+ "boi_token": "<|image>",
27
+ "eoa_token": "<audio|>",
28
+ "eoc_token": "<channel|>",
29
+ "eoi_token": "<image|>",
30
+ "eot_token": "<turn|>",
31
+ "escape_token": "<|\"|>",
32
+ "etc_token": "<tool_call|>",
33
+ "etd_token": "<tool|>",
34
+ "etr_token": "<tool_response|>",
35
+ "image_token": "<|image|>",
36
+ "soc_token": "<|channel>",
37
+ "sot_token": "<|turn>",
38
+ "stc_token": "<|tool_call>",
39
+ "std_token": "<|tool>",
40
+ "str_token": "<|tool_response>",
41
+ "think_token": "<|think|>"
42
+ },
43
+ "pad_token": "<pad>",
44
+ "padding_side": "right",
45
+ "processor_class": "Gemma4Processor",
46
+ "response_schema": {
47
+ "properties": {
48
+ "content": {
49
+ "type": "string"
50
+ },
51
+ "role": {
52
+ "const": "assistant"
53
+ },
54
+ "thinking": {
55
+ "type": "string"
56
+ },
57
+ "tool_calls": {
58
+ "items": {
59
+ "properties": {
60
+ "function": {
61
+ "properties": {
62
+ "arguments": {
63
+ "additionalProperties": {},
64
+ "type": "object",
65
+ "x-parser": "gemma4-tool-call"
66
+ },
67
+ "name": {
68
+ "type": "string"
69
+ }
70
+ },
71
+ "type": "object",
72
+ "x-regex": "call\\:(?P<name>\\w+)(?P<arguments>\\{.*\\})"
73
+ },
74
+ "type": {
75
+ "const": "function"
76
+ }
77
+ },
78
+ "type": "object"
79
+ },
80
+ "type": "array",
81
+ "x-regex-iterator": "<\\|tool_call>(.*?)<tool_call\\|>"
82
+ }
83
+ },
84
+ "type": "object",
85
+ "x-regex": "(\\<\\|channel\\>thought\\n(?P<thinking>.*?)\\<channel\\|\\>)?(?P<tool_calls>\\<\\|tool_call\\>.*\\<tool_call\\|\\>)?(?P<content>(?:(?!\\<turn\\|\\>)(?!\\<\\|tool_response\\>).)+)?(?:\\<turn\\|\\>|\\<\\|tool_response\\>)?"
86
+ },
87
+ "soc_token": "<|channel>",
88
+ "sot_token": "<|turn>",
89
+ "stc_token": "<|tool_call>",
90
+ "std_token": "<|tool>",
91
+ "str_token": "<|tool_response>",
92
+ "think_token": "<|think|>",
93
+ "tokenizer_class": "GemmaTokenizer",
94
+ "unk_token": "<unk>",
95
+ "added_tokens_decoder": {
96
+ "0": {
97
+ "content": "<pad>",
98
+ "single_word": false,
99
+ "lstrip": false,
100
+ "rstrip": false,
101
+ "normalized": false,
102
+ "special": true
103
+ },
104
+ "1": {
105
+ "content": "<eos>",
106
+ "single_word": false,
107
+ "lstrip": false,
108
+ "rstrip": false,
109
+ "normalized": false,
110
+ "special": true
111
+ },
112
+ "2": {
113
+ "content": "<bos>",
114
+ "single_word": false,
115
+ "lstrip": false,
116
+ "rstrip": false,
117
+ "normalized": false,
118
+ "special": true
119
+ },
120
+ "3": {
121
+ "content": "<unk>",
122
+ "single_word": false,
123
+ "lstrip": false,
124
+ "rstrip": false,
125
+ "normalized": false,
126
+ "special": true
127
+ },
128
+ "4": {
129
+ "content": "<mask>",
130
+ "single_word": false,
131
+ "lstrip": false,
132
+ "rstrip": false,
133
+ "normalized": false,
134
+ "special": true
135
+ },
136
+ "46": {
137
+ "content": "<|tool>",
138
+ "single_word": false,
139
+ "lstrip": false,
140
+ "rstrip": false,
141
+ "normalized": false,
142
+ "special": true
143
+ },
144
+ "47": {
145
+ "content": "<tool|>",
146
+ "single_word": false,
147
+ "lstrip": false,
148
+ "rstrip": false,
149
+ "normalized": false,
150
+ "special": true
151
+ },
152
+ "48": {
153
+ "content": "<|tool_call>",
154
+ "single_word": false,
155
+ "lstrip": false,
156
+ "rstrip": false,
157
+ "normalized": false,
158
+ "special": true
159
+ },
160
+ "49": {
161
+ "content": "<tool_call|>",
162
+ "single_word": false,
163
+ "lstrip": false,
164
+ "rstrip": false,
165
+ "normalized": false,
166
+ "special": true
167
+ },
168
+ "50": {
169
+ "content": "<|tool_response>",
170
+ "single_word": false,
171
+ "lstrip": false,
172
+ "rstrip": false,
173
+ "normalized": false,
174
+ "special": true
175
+ },
176
+ "51": {
177
+ "content": "<tool_response|>",
178
+ "single_word": false,
179
+ "lstrip": false,
180
+ "rstrip": false,
181
+ "normalized": false,
182
+ "special": true
183
+ },
184
+ "52": {
185
+ "content": "<|\"|>",
186
+ "single_word": false,
187
+ "lstrip": false,
188
+ "rstrip": false,
189
+ "normalized": false,
190
+ "special": true
191
+ },
192
+ "98": {
193
+ "content": "<|think|>",
194
+ "single_word": false,
195
+ "lstrip": false,
196
+ "rstrip": false,
197
+ "normalized": false,
198
+ "special": true
199
+ },
200
+ "100": {
201
+ "content": "<|channel>",
202
+ "single_word": false,
203
+ "lstrip": false,
204
+ "rstrip": false,
205
+ "normalized": false,
206
+ "special": true
207
+ },
208
+ "101": {
209
+ "content": "<channel|>",
210
+ "single_word": false,
211
+ "lstrip": false,
212
+ "rstrip": false,
213
+ "normalized": false,
214
+ "special": true
215
+ },
216
+ "105": {
217
+ "content": "<|turn>",
218
+ "single_word": false,
219
+ "lstrip": false,
220
+ "rstrip": false,
221
+ "normalized": false,
222
+ "special": true
223
+ },
224
+ "106": {
225
+ "content": "<turn|>",
226
+ "single_word": false,
227
+ "lstrip": false,
228
+ "rstrip": false,
229
+ "normalized": false,
230
+ "special": true
231
+ },
232
+ "255999": {
233
+ "content": "<|image>",
234
+ "single_word": false,
235
+ "lstrip": false,
236
+ "rstrip": false,
237
+ "normalized": false,
238
+ "special": true
239
+ },
240
+ "256000": {
241
+ "content": "<|audio>",
242
+ "single_word": false,
243
+ "lstrip": false,
244
+ "rstrip": false,
245
+ "normalized": false,
246
+ "special": true
247
+ },
248
+ "258880": {
249
+ "content": "<|image|>",
250
+ "single_word": false,
251
+ "lstrip": false,
252
+ "rstrip": false,
253
+ "normalized": false,
254
+ "special": true
255
+ },
256
+ "258881": {
257
+ "content": "<|audio|>",
258
+ "single_word": false,
259
+ "lstrip": false,
260
+ "rstrip": false,
261
+ "normalized": false,
262
+ "special": true
263
+ },
264
+ "258882": {
265
+ "content": "<image|>",
266
+ "single_word": false,
267
+ "lstrip": false,
268
+ "rstrip": false,
269
+ "normalized": false,
270
+ "special": true
271
+ },
272
+ "258883": {
273
+ "content": "<audio|>",
274
+ "single_word": false,
275
+ "lstrip": false,
276
+ "rstrip": false,
277
+ "normalized": false,
278
+ "special": true
279
+ },
280
+ "258884": {
281
+ "content": "<|video|>",
282
+ "single_word": false,
283
+ "lstrip": false,
284
+ "rstrip": false,
285
+ "normalized": false,
286
+ "special": true
287
+ }
288
+ }
289
+ }
checkpoint-150/trainer_state.json ADDED
@@ -0,0 +1,1534 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 0.03550295857988166,
6
+ "eval_steps": 500,
7
+ "global_step": 150,
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.00023668639053254438,
14
+ "grad_norm": NaN,
15
+ "learning_rate": 0.0,
16
+ "loss": 4.347962379455566,
17
+ "num_input_tokens_seen": 842,
18
+ "step": 1,
19
+ "train_runtime": 63.0289,
20
+ "train_tokens_per_second": 13.359
21
+ },
22
+ {
23
+ "epoch": 0.00047337278106508875,
24
+ "grad_norm": NaN,
25
+ "learning_rate": 0.0,
26
+ "loss": 4.634688854217529,
27
+ "num_input_tokens_seen": 1820,
28
+ "step": 2,
29
+ "train_runtime": 67.7033,
30
+ "train_tokens_per_second": 26.882
31
+ },
32
+ {
33
+ "epoch": 0.0007100591715976331,
34
+ "grad_norm": 13.944212913513184,
35
+ "learning_rate": 0.0,
36
+ "loss": 4.42241096496582,
37
+ "num_input_tokens_seen": 2754,
38
+ "step": 3,
39
+ "train_runtime": 72.324,
40
+ "train_tokens_per_second": 38.079
41
+ },
42
+ {
43
+ "epoch": 0.0009467455621301775,
44
+ "grad_norm": 13.445413589477539,
45
+ "learning_rate": 4e-05,
46
+ "loss": 4.682804107666016,
47
+ "num_input_tokens_seen": 3700,
48
+ "step": 4,
49
+ "train_runtime": 75.3925,
50
+ "train_tokens_per_second": 49.077
51
+ },
52
+ {
53
+ "epoch": 0.001183431952662722,
54
+ "grad_norm": 10.576727867126465,
55
+ "learning_rate": 8e-05,
56
+ "loss": 4.166159629821777,
57
+ "num_input_tokens_seen": 4714,
58
+ "step": 5,
59
+ "train_runtime": 78.474,
60
+ "train_tokens_per_second": 60.071
61
+ },
62
+ {
63
+ "epoch": 0.0014201183431952662,
64
+ "grad_norm": 8.077850341796875,
65
+ "learning_rate": 0.00012,
66
+ "loss": 3.2006261348724365,
67
+ "num_input_tokens_seen": 5674,
68
+ "step": 6,
69
+ "train_runtime": 81.6139,
70
+ "train_tokens_per_second": 69.522
71
+ },
72
+ {
73
+ "epoch": 0.0016568047337278107,
74
+ "grad_norm": 6.60789680480957,
75
+ "learning_rate": 0.00016,
76
+ "loss": 2.7067415714263916,
77
+ "num_input_tokens_seen": 6646,
78
+ "step": 7,
79
+ "train_runtime": 84.786,
80
+ "train_tokens_per_second": 78.386
81
+ },
82
+ {
83
+ "epoch": 0.001893491124260355,
84
+ "grad_norm": 9.175209045410156,
85
+ "learning_rate": 0.0002,
86
+ "loss": 2.367570638656616,
87
+ "num_input_tokens_seen": 7518,
88
+ "step": 8,
89
+ "train_runtime": 87.8178,
90
+ "train_tokens_per_second": 85.609
91
+ },
92
+ {
93
+ "epoch": 0.0021301775147928993,
94
+ "grad_norm": 3.438948631286621,
95
+ "learning_rate": 0.00019891891891891895,
96
+ "loss": 2.1013119220733643,
97
+ "num_input_tokens_seen": 8470,
98
+ "step": 9,
99
+ "train_runtime": 90.9458,
100
+ "train_tokens_per_second": 93.132
101
+ },
102
+ {
103
+ "epoch": 0.002366863905325444,
104
+ "grad_norm": 3.459472894668579,
105
+ "learning_rate": 0.00019783783783783784,
106
+ "loss": 1.8991940021514893,
107
+ "num_input_tokens_seen": 9458,
108
+ "step": 10,
109
+ "train_runtime": 94.0577,
110
+ "train_tokens_per_second": 100.555
111
+ },
112
+ {
113
+ "epoch": 0.0026035502958579883,
114
+ "grad_norm": 2.169750928878784,
115
+ "learning_rate": 0.00019675675675675678,
116
+ "loss": 1.7395055294036865,
117
+ "num_input_tokens_seen": 10456,
118
+ "step": 11,
119
+ "train_runtime": 97.2068,
120
+ "train_tokens_per_second": 107.565
121
+ },
122
+ {
123
+ "epoch": 0.0028402366863905324,
124
+ "grad_norm": 1.8638927936553955,
125
+ "learning_rate": 0.00019567567567567567,
126
+ "loss": 1.5622950792312622,
127
+ "num_input_tokens_seen": 11410,
128
+ "step": 12,
129
+ "train_runtime": 100.2911,
130
+ "train_tokens_per_second": 113.769
131
+ },
132
+ {
133
+ "epoch": 0.003076923076923077,
134
+ "grad_norm": 1.4741116762161255,
135
+ "learning_rate": 0.00019459459459459462,
136
+ "loss": 1.392699956893921,
137
+ "num_input_tokens_seen": 12316,
138
+ "step": 13,
139
+ "train_runtime": 103.3664,
140
+ "train_tokens_per_second": 119.149
141
+ },
142
+ {
143
+ "epoch": 0.0033136094674556214,
144
+ "grad_norm": 1.3238699436187744,
145
+ "learning_rate": 0.00019351351351351353,
146
+ "loss": 1.2535074949264526,
147
+ "num_input_tokens_seen": 13256,
148
+ "step": 14,
149
+ "train_runtime": 106.4031,
150
+ "train_tokens_per_second": 124.583
151
+ },
152
+ {
153
+ "epoch": 0.0035502958579881655,
154
+ "grad_norm": 1.276484489440918,
155
+ "learning_rate": 0.00019243243243243245,
156
+ "loss": 1.135048508644104,
157
+ "num_input_tokens_seen": 14228,
158
+ "step": 15,
159
+ "train_runtime": 109.4909,
160
+ "train_tokens_per_second": 129.947
161
+ },
162
+ {
163
+ "epoch": 0.00378698224852071,
164
+ "grad_norm": 1.1560865640640259,
165
+ "learning_rate": 0.00019135135135135137,
166
+ "loss": 1.0317097902297974,
167
+ "num_input_tokens_seen": 15192,
168
+ "step": 16,
169
+ "train_runtime": 112.5741,
170
+ "train_tokens_per_second": 134.951
171
+ },
172
+ {
173
+ "epoch": 0.004023668639053254,
174
+ "grad_norm": 1.1331353187561035,
175
+ "learning_rate": 0.00019027027027027028,
176
+ "loss": 0.9967219233512878,
177
+ "num_input_tokens_seen": 16130,
178
+ "step": 17,
179
+ "train_runtime": 115.6495,
180
+ "train_tokens_per_second": 139.473
181
+ },
182
+ {
183
+ "epoch": 0.004260355029585799,
184
+ "grad_norm": 1.1567702293395996,
185
+ "learning_rate": 0.0001891891891891892,
186
+ "loss": 0.905368447303772,
187
+ "num_input_tokens_seen": 17078,
188
+ "step": 18,
189
+ "train_runtime": 118.6796,
190
+ "train_tokens_per_second": 143.9
191
+ },
192
+ {
193
+ "epoch": 0.004497041420118343,
194
+ "grad_norm": 0.9157600998878479,
195
+ "learning_rate": 0.00018810810810810814,
196
+ "loss": 0.8075690865516663,
197
+ "num_input_tokens_seen": 18062,
198
+ "step": 19,
199
+ "train_runtime": 121.8602,
200
+ "train_tokens_per_second": 148.219
201
+ },
202
+ {
203
+ "epoch": 0.004733727810650888,
204
+ "grad_norm": 0.8306605815887451,
205
+ "learning_rate": 0.00018702702702702703,
206
+ "loss": 0.8007175922393799,
207
+ "num_input_tokens_seen": 19078,
208
+ "step": 20,
209
+ "train_runtime": 125.0019,
210
+ "train_tokens_per_second": 152.622
211
+ },
212
+ {
213
+ "epoch": 0.004970414201183432,
214
+ "grad_norm": 0.690018355846405,
215
+ "learning_rate": 0.00018594594594594597,
216
+ "loss": 0.7810791730880737,
217
+ "num_input_tokens_seen": 20068,
218
+ "step": 21,
219
+ "train_runtime": 128.0932,
220
+ "train_tokens_per_second": 156.667
221
+ },
222
+ {
223
+ "epoch": 0.005207100591715977,
224
+ "grad_norm": 1.4768867492675781,
225
+ "learning_rate": 0.00018486486486486486,
226
+ "loss": 0.7172120809555054,
227
+ "num_input_tokens_seen": 20972,
228
+ "step": 22,
229
+ "train_runtime": 131.0901,
230
+ "train_tokens_per_second": 159.982
231
+ },
232
+ {
233
+ "epoch": 0.00544378698224852,
234
+ "grad_norm": 0.7999809980392456,
235
+ "learning_rate": 0.0001837837837837838,
236
+ "loss": 0.6333726048469543,
237
+ "num_input_tokens_seen": 21900,
238
+ "step": 23,
239
+ "train_runtime": 134.212,
240
+ "train_tokens_per_second": 163.175
241
+ },
242
+ {
243
+ "epoch": 0.005680473372781065,
244
+ "grad_norm": 0.9285996556282043,
245
+ "learning_rate": 0.00018270270270270272,
246
+ "loss": 0.6750729084014893,
247
+ "num_input_tokens_seen": 22770,
248
+ "step": 24,
249
+ "train_runtime": 137.3193,
250
+ "train_tokens_per_second": 165.818
251
+ },
252
+ {
253
+ "epoch": 0.005917159763313609,
254
+ "grad_norm": 0.7733820080757141,
255
+ "learning_rate": 0.00018162162162162164,
256
+ "loss": 0.672032356262207,
257
+ "num_input_tokens_seen": 23746,
258
+ "step": 25,
259
+ "train_runtime": 140.3175,
260
+ "train_tokens_per_second": 169.23
261
+ },
262
+ {
263
+ "epoch": 0.006153846153846154,
264
+ "grad_norm": 0.73944091796875,
265
+ "learning_rate": 0.00018054054054054055,
266
+ "loss": 0.6716101169586182,
267
+ "num_input_tokens_seen": 24716,
268
+ "step": 26,
269
+ "train_runtime": 143.3683,
270
+ "train_tokens_per_second": 172.395
271
+ },
272
+ {
273
+ "epoch": 0.006390532544378698,
274
+ "grad_norm": 0.7659372091293335,
275
+ "learning_rate": 0.00017945945945945947,
276
+ "loss": 0.6067175269126892,
277
+ "num_input_tokens_seen": 25724,
278
+ "step": 27,
279
+ "train_runtime": 146.5377,
280
+ "train_tokens_per_second": 175.545
281
+ },
282
+ {
283
+ "epoch": 0.006627218934911243,
284
+ "grad_norm": 0.5428617000579834,
285
+ "learning_rate": 0.00017837837837837839,
286
+ "loss": 0.6083395481109619,
287
+ "num_input_tokens_seen": 26712,
288
+ "step": 28,
289
+ "train_runtime": 149.6429,
290
+ "train_tokens_per_second": 178.505
291
+ },
292
+ {
293
+ "epoch": 0.006863905325443787,
294
+ "grad_norm": 0.5915114283561707,
295
+ "learning_rate": 0.00017729729729729733,
296
+ "loss": 0.6463742256164551,
297
+ "num_input_tokens_seen": 27704,
298
+ "step": 29,
299
+ "train_runtime": 152.7501,
300
+ "train_tokens_per_second": 181.368
301
+ },
302
+ {
303
+ "epoch": 0.007100591715976331,
304
+ "grad_norm": 1.3350757360458374,
305
+ "learning_rate": 0.00017621621621621622,
306
+ "loss": 0.5705698132514954,
307
+ "num_input_tokens_seen": 28648,
308
+ "step": 30,
309
+ "train_runtime": 155.8918,
310
+ "train_tokens_per_second": 183.768
311
+ },
312
+ {
313
+ "epoch": 0.0073372781065088755,
314
+ "grad_norm": 0.68421471118927,
315
+ "learning_rate": 0.00017513513513513516,
316
+ "loss": 0.5660746097564697,
317
+ "num_input_tokens_seen": 29622,
318
+ "step": 31,
319
+ "train_runtime": 163.3209,
320
+ "train_tokens_per_second": 181.373
321
+ },
322
+ {
323
+ "epoch": 0.00757396449704142,
324
+ "grad_norm": 0.5762604475021362,
325
+ "learning_rate": 0.00017405405405405405,
326
+ "loss": 0.5483309030532837,
327
+ "num_input_tokens_seen": 30534,
328
+ "step": 32,
329
+ "train_runtime": 166.5185,
330
+ "train_tokens_per_second": 183.367
331
+ },
332
+ {
333
+ "epoch": 0.0078106508875739646,
334
+ "grad_norm": 0.5751784443855286,
335
+ "learning_rate": 0.000172972972972973,
336
+ "loss": 0.5207005739212036,
337
+ "num_input_tokens_seen": 31458,
338
+ "step": 33,
339
+ "train_runtime": 169.7055,
340
+ "train_tokens_per_second": 185.368
341
+ },
342
+ {
343
+ "epoch": 0.008047337278106508,
344
+ "grad_norm": 0.5752039551734924,
345
+ "learning_rate": 0.0001718918918918919,
346
+ "loss": 0.5534584522247314,
347
+ "num_input_tokens_seen": 32424,
348
+ "step": 34,
349
+ "train_runtime": 172.9184,
350
+ "train_tokens_per_second": 187.51
351
+ },
352
+ {
353
+ "epoch": 0.008284023668639054,
354
+ "grad_norm": 0.5918363332748413,
355
+ "learning_rate": 0.00017081081081081083,
356
+ "loss": 0.482683002948761,
357
+ "num_input_tokens_seen": 33398,
358
+ "step": 35,
359
+ "train_runtime": 176.2158,
360
+ "train_tokens_per_second": 189.529
361
+ },
362
+ {
363
+ "epoch": 0.008520710059171597,
364
+ "grad_norm": 0.6921846270561218,
365
+ "learning_rate": 0.00016972972972972974,
366
+ "loss": 0.46112507581710815,
367
+ "num_input_tokens_seen": 34324,
368
+ "step": 36,
369
+ "train_runtime": 179.3173,
370
+ "train_tokens_per_second": 191.415
371
+ },
372
+ {
373
+ "epoch": 0.008757396449704143,
374
+ "grad_norm": 0.6429721117019653,
375
+ "learning_rate": 0.00016864864864864866,
376
+ "loss": 0.45629197359085083,
377
+ "num_input_tokens_seen": 35270,
378
+ "step": 37,
379
+ "train_runtime": 182.3558,
380
+ "train_tokens_per_second": 193.413
381
+ },
382
+ {
383
+ "epoch": 0.008994082840236686,
384
+ "grad_norm": 0.5477020740509033,
385
+ "learning_rate": 0.00016756756756756757,
386
+ "loss": 0.4547877907752991,
387
+ "num_input_tokens_seen": 36206,
388
+ "step": 38,
389
+ "train_runtime": 185.4483,
390
+ "train_tokens_per_second": 195.235
391
+ },
392
+ {
393
+ "epoch": 0.009230769230769232,
394
+ "grad_norm": 0.6158634424209595,
395
+ "learning_rate": 0.00016648648648648652,
396
+ "loss": 0.367482990026474,
397
+ "num_input_tokens_seen": 37080,
398
+ "step": 39,
399
+ "train_runtime": 188.5114,
400
+ "train_tokens_per_second": 196.699
401
+ },
402
+ {
403
+ "epoch": 0.009467455621301775,
404
+ "grad_norm": 0.5405091643333435,
405
+ "learning_rate": 0.0001654054054054054,
406
+ "loss": 0.42049410939216614,
407
+ "num_input_tokens_seen": 38030,
408
+ "step": 40,
409
+ "train_runtime": 191.5986,
410
+ "train_tokens_per_second": 198.488
411
+ },
412
+ {
413
+ "epoch": 0.009704142011834319,
414
+ "grad_norm": 0.6389183402061462,
415
+ "learning_rate": 0.00016432432432432435,
416
+ "loss": 0.3935014307498932,
417
+ "num_input_tokens_seen": 38942,
418
+ "step": 41,
419
+ "train_runtime": 194.8426,
420
+ "train_tokens_per_second": 199.864
421
+ },
422
+ {
423
+ "epoch": 0.009940828402366864,
424
+ "grad_norm": 0.6742956042289734,
425
+ "learning_rate": 0.00016324324324324324,
426
+ "loss": 0.4373692274093628,
427
+ "num_input_tokens_seen": 39902,
428
+ "step": 42,
429
+ "train_runtime": 197.8938,
430
+ "train_tokens_per_second": 201.633
431
+ },
432
+ {
433
+ "epoch": 0.010177514792899408,
434
+ "grad_norm": 0.6083794236183167,
435
+ "learning_rate": 0.00016216216216216218,
436
+ "loss": 0.3776765465736389,
437
+ "num_input_tokens_seen": 40858,
438
+ "step": 43,
439
+ "train_runtime": 200.9708,
440
+ "train_tokens_per_second": 203.303
441
+ },
442
+ {
443
+ "epoch": 0.010414201183431953,
444
+ "grad_norm": 0.5469992160797119,
445
+ "learning_rate": 0.0001610810810810811,
446
+ "loss": 0.31133151054382324,
447
+ "num_input_tokens_seen": 41760,
448
+ "step": 44,
449
+ "train_runtime": 204.1304,
450
+ "train_tokens_per_second": 204.575
451
+ },
452
+ {
453
+ "epoch": 0.010650887573964497,
454
+ "grad_norm": 0.6452498435974121,
455
+ "learning_rate": 0.00016,
456
+ "loss": 0.42721521854400635,
457
+ "num_input_tokens_seen": 42740,
458
+ "step": 45,
459
+ "train_runtime": 207.1845,
460
+ "train_tokens_per_second": 206.29
461
+ },
462
+ {
463
+ "epoch": 0.01088757396449704,
464
+ "grad_norm": 0.5988945960998535,
465
+ "learning_rate": 0.00015891891891891893,
466
+ "loss": 0.41375258564949036,
467
+ "num_input_tokens_seen": 43750,
468
+ "step": 46,
469
+ "train_runtime": 210.2912,
470
+ "train_tokens_per_second": 208.045
471
+ },
472
+ {
473
+ "epoch": 0.011124260355029586,
474
+ "grad_norm": 0.6452038288116455,
475
+ "learning_rate": 0.00015783783783783785,
476
+ "loss": 0.3485276699066162,
477
+ "num_input_tokens_seen": 44690,
478
+ "step": 47,
479
+ "train_runtime": 213.3684,
480
+ "train_tokens_per_second": 209.45
481
+ },
482
+ {
483
+ "epoch": 0.01136094674556213,
484
+ "grad_norm": 0.7041454911231995,
485
+ "learning_rate": 0.00015675675675675676,
486
+ "loss": 0.3296637535095215,
487
+ "num_input_tokens_seen": 45666,
488
+ "step": 48,
489
+ "train_runtime": 216.4459,
490
+ "train_tokens_per_second": 210.981
491
+ },
492
+ {
493
+ "epoch": 0.011597633136094675,
494
+ "grad_norm": 0.6726324558258057,
495
+ "learning_rate": 0.0001556756756756757,
496
+ "loss": 0.3179457187652588,
497
+ "num_input_tokens_seen": 46560,
498
+ "step": 49,
499
+ "train_runtime": 219.4697,
500
+ "train_tokens_per_second": 212.148
501
+ },
502
+ {
503
+ "epoch": 0.011834319526627219,
504
+ "grad_norm": 0.7513703107833862,
505
+ "learning_rate": 0.0001545945945945946,
506
+ "loss": 0.3722391426563263,
507
+ "num_input_tokens_seen": 47528,
508
+ "step": 50,
509
+ "train_runtime": 222.596,
510
+ "train_tokens_per_second": 213.517
511
+ },
512
+ {
513
+ "epoch": 0.012071005917159764,
514
+ "grad_norm": 0.6454160213470459,
515
+ "learning_rate": 0.00015351351351351354,
516
+ "loss": 0.283376544713974,
517
+ "num_input_tokens_seen": 48468,
518
+ "step": 51,
519
+ "train_runtime": 225.7262,
520
+ "train_tokens_per_second": 214.72
521
+ },
522
+ {
523
+ "epoch": 0.012307692307692308,
524
+ "grad_norm": 0.9116244316101074,
525
+ "learning_rate": 0.00015243243243243243,
526
+ "loss": 0.4286752939224243,
527
+ "num_input_tokens_seen": 49444,
528
+ "step": 52,
529
+ "train_runtime": 228.803,
530
+ "train_tokens_per_second": 216.099
531
+ },
532
+ {
533
+ "epoch": 0.012544378698224851,
534
+ "grad_norm": 0.6157099604606628,
535
+ "learning_rate": 0.00015135135135135137,
536
+ "loss": 0.27355721592903137,
537
+ "num_input_tokens_seen": 50424,
538
+ "step": 53,
539
+ "train_runtime": 231.8413,
540
+ "train_tokens_per_second": 217.494
541
+ },
542
+ {
543
+ "epoch": 0.012781065088757397,
544
+ "grad_norm": 0.6335808038711548,
545
+ "learning_rate": 0.00015027027027027028,
546
+ "loss": 0.37197017669677734,
547
+ "num_input_tokens_seen": 51328,
548
+ "step": 54,
549
+ "train_runtime": 234.9265,
550
+ "train_tokens_per_second": 218.485
551
+ },
552
+ {
553
+ "epoch": 0.01301775147928994,
554
+ "grad_norm": 0.6838424205780029,
555
+ "learning_rate": 0.0001491891891891892,
556
+ "loss": 0.3504602909088135,
557
+ "num_input_tokens_seen": 52274,
558
+ "step": 55,
559
+ "train_runtime": 238.0114,
560
+ "train_tokens_per_second": 219.628
561
+ },
562
+ {
563
+ "epoch": 0.013254437869822486,
564
+ "grad_norm": 0.6566418409347534,
565
+ "learning_rate": 0.00014810810810810812,
566
+ "loss": 0.3189927935600281,
567
+ "num_input_tokens_seen": 53214,
568
+ "step": 56,
569
+ "train_runtime": 241.0892,
570
+ "train_tokens_per_second": 220.723
571
+ },
572
+ {
573
+ "epoch": 0.01349112426035503,
574
+ "grad_norm": 0.6112971901893616,
575
+ "learning_rate": 0.00014702702702702703,
576
+ "loss": 0.3080369830131531,
577
+ "num_input_tokens_seen": 54162,
578
+ "step": 57,
579
+ "train_runtime": 244.2461,
580
+ "train_tokens_per_second": 221.752
581
+ },
582
+ {
583
+ "epoch": 0.013727810650887575,
584
+ "grad_norm": 0.6469757556915283,
585
+ "learning_rate": 0.00014594594594594595,
586
+ "loss": 0.24279898405075073,
587
+ "num_input_tokens_seen": 55158,
588
+ "step": 58,
589
+ "train_runtime": 247.4064,
590
+ "train_tokens_per_second": 222.945
591
+ },
592
+ {
593
+ "epoch": 0.013964497041420118,
594
+ "grad_norm": 0.7099384665489197,
595
+ "learning_rate": 0.0001448648648648649,
596
+ "loss": 0.31188705563545227,
597
+ "num_input_tokens_seen": 56096,
598
+ "step": 59,
599
+ "train_runtime": 250.4708,
600
+ "train_tokens_per_second": 223.962
601
+ },
602
+ {
603
+ "epoch": 0.014201183431952662,
604
+ "grad_norm": 0.5927040576934814,
605
+ "learning_rate": 0.00014378378378378378,
606
+ "loss": 0.25696688890457153,
607
+ "num_input_tokens_seen": 57012,
608
+ "step": 60,
609
+ "train_runtime": 253.7131,
610
+ "train_tokens_per_second": 224.711
611
+ },
612
+ {
613
+ "epoch": 0.014437869822485207,
614
+ "grad_norm": 0.5483338236808777,
615
+ "learning_rate": 0.00014270270270270272,
616
+ "loss": 0.2725352644920349,
617
+ "num_input_tokens_seen": 58036,
618
+ "step": 61,
619
+ "train_runtime": 260.2844,
620
+ "train_tokens_per_second": 222.971
621
+ },
622
+ {
623
+ "epoch": 0.014674556213017751,
624
+ "grad_norm": 0.6806847453117371,
625
+ "learning_rate": 0.00014162162162162161,
626
+ "loss": 0.31959983706474304,
627
+ "num_input_tokens_seen": 59026,
628
+ "step": 62,
629
+ "train_runtime": 263.3802,
630
+ "train_tokens_per_second": 224.109
631
+ },
632
+ {
633
+ "epoch": 0.014911242603550296,
634
+ "grad_norm": 0.6916660070419312,
635
+ "learning_rate": 0.00014054054054054056,
636
+ "loss": 0.312305748462677,
637
+ "num_input_tokens_seen": 59982,
638
+ "step": 63,
639
+ "train_runtime": 266.3794,
640
+ "train_tokens_per_second": 225.175
641
+ },
642
+ {
643
+ "epoch": 0.01514792899408284,
644
+ "grad_norm": 0.5516266226768494,
645
+ "learning_rate": 0.00013945945945945947,
646
+ "loss": 0.2653953433036804,
647
+ "num_input_tokens_seen": 60926,
648
+ "step": 64,
649
+ "train_runtime": 269.3893,
650
+ "train_tokens_per_second": 226.163
651
+ },
652
+ {
653
+ "epoch": 0.015384615384615385,
654
+ "grad_norm": 0.5579270720481873,
655
+ "learning_rate": 0.0001383783783783784,
656
+ "loss": 0.20462818443775177,
657
+ "num_input_tokens_seen": 61844,
658
+ "step": 65,
659
+ "train_runtime": 272.3982,
660
+ "train_tokens_per_second": 227.035
661
+ },
662
+ {
663
+ "epoch": 0.015621301775147929,
664
+ "grad_norm": 0.7645660638809204,
665
+ "learning_rate": 0.0001372972972972973,
666
+ "loss": 0.2650893032550812,
667
+ "num_input_tokens_seen": 62766,
668
+ "step": 66,
669
+ "train_runtime": 275.404,
670
+ "train_tokens_per_second": 227.905
671
+ },
672
+ {
673
+ "epoch": 0.015857988165680473,
674
+ "grad_norm": 0.5974738001823425,
675
+ "learning_rate": 0.00013621621621621622,
676
+ "loss": 0.21247778832912445,
677
+ "num_input_tokens_seen": 63690,
678
+ "step": 67,
679
+ "train_runtime": 278.4021,
680
+ "train_tokens_per_second": 228.77
681
+ },
682
+ {
683
+ "epoch": 0.016094674556213016,
684
+ "grad_norm": 0.5831193327903748,
685
+ "learning_rate": 0.00013513513513513514,
686
+ "loss": 0.2024172842502594,
687
+ "num_input_tokens_seen": 64496,
688
+ "step": 68,
689
+ "train_runtime": 281.4313,
690
+ "train_tokens_per_second": 229.171
691
+ },
692
+ {
693
+ "epoch": 0.016331360946745564,
694
+ "grad_norm": 0.8678795695304871,
695
+ "learning_rate": 0.00013405405405405408,
696
+ "loss": 0.31490638852119446,
697
+ "num_input_tokens_seen": 65460,
698
+ "step": 69,
699
+ "train_runtime": 284.6137,
700
+ "train_tokens_per_second": 229.996
701
+ },
702
+ {
703
+ "epoch": 0.016568047337278107,
704
+ "grad_norm": 0.7611344456672668,
705
+ "learning_rate": 0.00013297297297297297,
706
+ "loss": 0.24661925435066223,
707
+ "num_input_tokens_seen": 66374,
708
+ "step": 70,
709
+ "train_runtime": 287.7742,
710
+ "train_tokens_per_second": 230.646
711
+ },
712
+ {
713
+ "epoch": 0.01680473372781065,
714
+ "grad_norm": 0.6685308814048767,
715
+ "learning_rate": 0.0001318918918918919,
716
+ "loss": 0.1843535602092743,
717
+ "num_input_tokens_seen": 67112,
718
+ "step": 71,
719
+ "train_runtime": 290.9321,
720
+ "train_tokens_per_second": 230.679
721
+ },
722
+ {
723
+ "epoch": 0.017041420118343194,
724
+ "grad_norm": 0.6150883436203003,
725
+ "learning_rate": 0.0001308108108108108,
726
+ "loss": 0.23372295498847961,
727
+ "num_input_tokens_seen": 68154,
728
+ "step": 72,
729
+ "train_runtime": 294.0809,
730
+ "train_tokens_per_second": 231.753
731
+ },
732
+ {
733
+ "epoch": 0.017278106508875738,
734
+ "grad_norm": 0.6938812732696533,
735
+ "learning_rate": 0.00012972972972972974,
736
+ "loss": 0.22619186341762543,
737
+ "num_input_tokens_seen": 69102,
738
+ "step": 73,
739
+ "train_runtime": 297.1843,
740
+ "train_tokens_per_second": 232.522
741
+ },
742
+ {
743
+ "epoch": 0.017514792899408285,
744
+ "grad_norm": 0.617787778377533,
745
+ "learning_rate": 0.00012864864864864866,
746
+ "loss": 0.2535328269004822,
747
+ "num_input_tokens_seen": 70010,
748
+ "step": 74,
749
+ "train_runtime": 300.2851,
750
+ "train_tokens_per_second": 233.145
751
+ },
752
+ {
753
+ "epoch": 0.01775147928994083,
754
+ "grad_norm": 0.6884900331497192,
755
+ "learning_rate": 0.00012756756756756758,
756
+ "loss": 0.21632739901542664,
757
+ "num_input_tokens_seen": 70940,
758
+ "step": 75,
759
+ "train_runtime": 303.3533,
760
+ "train_tokens_per_second": 233.853
761
+ },
762
+ {
763
+ "epoch": 0.017988165680473372,
764
+ "grad_norm": 0.5782189965248108,
765
+ "learning_rate": 0.0001264864864864865,
766
+ "loss": 0.20167675614356995,
767
+ "num_input_tokens_seen": 71862,
768
+ "step": 76,
769
+ "train_runtime": 306.4503,
770
+ "train_tokens_per_second": 234.498
771
+ },
772
+ {
773
+ "epoch": 0.018224852071005916,
774
+ "grad_norm": 0.6541483402252197,
775
+ "learning_rate": 0.0001254054054054054,
776
+ "loss": 0.23231221735477448,
777
+ "num_input_tokens_seen": 72812,
778
+ "step": 77,
779
+ "train_runtime": 309.5743,
780
+ "train_tokens_per_second": 235.2
781
+ },
782
+ {
783
+ "epoch": 0.018461538461538463,
784
+ "grad_norm": 0.7524886727333069,
785
+ "learning_rate": 0.00012432432432432433,
786
+ "loss": 0.27632662653923035,
787
+ "num_input_tokens_seen": 73774,
788
+ "step": 78,
789
+ "train_runtime": 312.6963,
790
+ "train_tokens_per_second": 235.929
791
+ },
792
+ {
793
+ "epoch": 0.018698224852071007,
794
+ "grad_norm": 0.716282308101654,
795
+ "learning_rate": 0.00012324324324324327,
796
+ "loss": 0.2734939754009247,
797
+ "num_input_tokens_seen": 74716,
798
+ "step": 79,
799
+ "train_runtime": 315.7612,
800
+ "train_tokens_per_second": 236.622
801
+ },
802
+ {
803
+ "epoch": 0.01893491124260355,
804
+ "grad_norm": 0.7010631561279297,
805
+ "learning_rate": 0.00012216216216216216,
806
+ "loss": 0.2774362564086914,
807
+ "num_input_tokens_seen": 75692,
808
+ "step": 80,
809
+ "train_runtime": 318.7983,
810
+ "train_tokens_per_second": 237.429
811
+ },
812
+ {
813
+ "epoch": 0.019171597633136094,
814
+ "grad_norm": 0.6561240553855896,
815
+ "learning_rate": 0.00012108108108108109,
816
+ "loss": 0.21640686690807343,
817
+ "num_input_tokens_seen": 76558,
818
+ "step": 81,
819
+ "train_runtime": 321.889,
820
+ "train_tokens_per_second": 237.84
821
+ },
822
+ {
823
+ "epoch": 0.019408284023668638,
824
+ "grad_norm": 0.5552831888198853,
825
+ "learning_rate": 0.00012,
826
+ "loss": 0.21232956647872925,
827
+ "num_input_tokens_seen": 77408,
828
+ "step": 82,
829
+ "train_runtime": 325.0469,
830
+ "train_tokens_per_second": 238.144
831
+ },
832
+ {
833
+ "epoch": 0.019644970414201185,
834
+ "grad_norm": 0.5905120372772217,
835
+ "learning_rate": 0.00011891891891891893,
836
+ "loss": 0.22443550825119019,
837
+ "num_input_tokens_seen": 78278,
838
+ "step": 83,
839
+ "train_runtime": 328.099,
840
+ "train_tokens_per_second": 238.58
841
+ },
842
+ {
843
+ "epoch": 0.01988165680473373,
844
+ "grad_norm": 0.6606366634368896,
845
+ "learning_rate": 0.00011783783783783784,
846
+ "loss": 0.2400747388601303,
847
+ "num_input_tokens_seen": 79262,
848
+ "step": 84,
849
+ "train_runtime": 331.1361,
850
+ "train_tokens_per_second": 239.364
851
+ },
852
+ {
853
+ "epoch": 0.020118343195266272,
854
+ "grad_norm": 0.6508055329322815,
855
+ "learning_rate": 0.00011675675675675676,
856
+ "loss": 0.19072499871253967,
857
+ "num_input_tokens_seen": 80144,
858
+ "step": 85,
859
+ "train_runtime": 334.2158,
860
+ "train_tokens_per_second": 239.797
861
+ },
862
+ {
863
+ "epoch": 0.020355029585798816,
864
+ "grad_norm": 0.606562614440918,
865
+ "learning_rate": 0.00011567567567567568,
866
+ "loss": 0.20951862633228302,
867
+ "num_input_tokens_seen": 81108,
868
+ "step": 86,
869
+ "train_runtime": 337.3105,
870
+ "train_tokens_per_second": 240.455
871
+ },
872
+ {
873
+ "epoch": 0.02059171597633136,
874
+ "grad_norm": 0.6757521629333496,
875
+ "learning_rate": 0.00011459459459459461,
876
+ "loss": 0.17614495754241943,
877
+ "num_input_tokens_seen": 82032,
878
+ "step": 87,
879
+ "train_runtime": 340.3438,
880
+ "train_tokens_per_second": 241.027
881
+ },
882
+ {
883
+ "epoch": 0.020828402366863907,
884
+ "grad_norm": 0.671481728553772,
885
+ "learning_rate": 0.00011351351351351351,
886
+ "loss": 0.1901710331439972,
887
+ "num_input_tokens_seen": 82954,
888
+ "step": 88,
889
+ "train_runtime": 343.4059,
890
+ "train_tokens_per_second": 241.563
891
+ },
892
+ {
893
+ "epoch": 0.02106508875739645,
894
+ "grad_norm": 0.5919027924537659,
895
+ "learning_rate": 0.00011243243243243244,
896
+ "loss": 0.1563737988471985,
897
+ "num_input_tokens_seen": 83880,
898
+ "step": 89,
899
+ "train_runtime": 346.5125,
900
+ "train_tokens_per_second": 242.069
901
+ },
902
+ {
903
+ "epoch": 0.021301775147928994,
904
+ "grad_norm": 0.7108309268951416,
905
+ "learning_rate": 0.00011135135135135135,
906
+ "loss": 0.17256894707679749,
907
+ "num_input_tokens_seen": 84852,
908
+ "step": 90,
909
+ "train_runtime": 349.6508,
910
+ "train_tokens_per_second": 242.676
911
+ },
912
+ {
913
+ "epoch": 0.021538461538461538,
914
+ "grad_norm": 0.8608420491218567,
915
+ "learning_rate": 0.00011027027027027029,
916
+ "loss": 0.1845530867576599,
917
+ "num_input_tokens_seen": 85820,
918
+ "step": 91,
919
+ "train_runtime": 356.267,
920
+ "train_tokens_per_second": 240.887
921
+ },
922
+ {
923
+ "epoch": 0.02177514792899408,
924
+ "grad_norm": 0.9677010774612427,
925
+ "learning_rate": 0.00010918918918918919,
926
+ "loss": 0.24358105659484863,
927
+ "num_input_tokens_seen": 86770,
928
+ "step": 92,
929
+ "train_runtime": 359.3722,
930
+ "train_tokens_per_second": 241.449
931
+ },
932
+ {
933
+ "epoch": 0.02201183431952663,
934
+ "grad_norm": 0.6302966475486755,
935
+ "learning_rate": 0.00010810810810810812,
936
+ "loss": 0.18155133724212646,
937
+ "num_input_tokens_seen": 87736,
938
+ "step": 93,
939
+ "train_runtime": 362.5874,
940
+ "train_tokens_per_second": 241.972
941
+ },
942
+ {
943
+ "epoch": 0.022248520710059172,
944
+ "grad_norm": 0.7171812653541565,
945
+ "learning_rate": 0.00010702702702702702,
946
+ "loss": 0.15349116921424866,
947
+ "num_input_tokens_seen": 88626,
948
+ "step": 94,
949
+ "train_runtime": 365.7055,
950
+ "train_tokens_per_second": 242.343
951
+ },
952
+ {
953
+ "epoch": 0.022485207100591716,
954
+ "grad_norm": 0.7326887845993042,
955
+ "learning_rate": 0.00010594594594594595,
956
+ "loss": 0.18243758380413055,
957
+ "num_input_tokens_seen": 89548,
958
+ "step": 95,
959
+ "train_runtime": 368.799,
960
+ "train_tokens_per_second": 242.81
961
+ },
962
+ {
963
+ "epoch": 0.02272189349112426,
964
+ "grad_norm": 0.6686019897460938,
965
+ "learning_rate": 0.00010486486486486487,
966
+ "loss": 0.15622465312480927,
967
+ "num_input_tokens_seen": 90504,
968
+ "step": 96,
969
+ "train_runtime": 371.8404,
970
+ "train_tokens_per_second": 243.395
971
+ },
972
+ {
973
+ "epoch": 0.022958579881656806,
974
+ "grad_norm": 0.6306503415107727,
975
+ "learning_rate": 0.0001037837837837838,
976
+ "loss": 0.1272117793560028,
977
+ "num_input_tokens_seen": 91454,
978
+ "step": 97,
979
+ "train_runtime": 375.0199,
980
+ "train_tokens_per_second": 243.864
981
+ },
982
+ {
983
+ "epoch": 0.02319526627218935,
984
+ "grad_norm": 0.6068493127822876,
985
+ "learning_rate": 0.0001027027027027027,
986
+ "loss": 0.1577155441045761,
987
+ "num_input_tokens_seen": 92466,
988
+ "step": 98,
989
+ "train_runtime": 378.0978,
990
+ "train_tokens_per_second": 244.556
991
+ },
992
+ {
993
+ "epoch": 0.023431952662721894,
994
+ "grad_norm": 0.7263635993003845,
995
+ "learning_rate": 0.00010162162162162163,
996
+ "loss": 0.1845340132713318,
997
+ "num_input_tokens_seen": 93398,
998
+ "step": 99,
999
+ "train_runtime": 381.1358,
1000
+ "train_tokens_per_second": 245.052
1001
+ },
1002
+ {
1003
+ "epoch": 0.023668639053254437,
1004
+ "grad_norm": 0.6466794013977051,
1005
+ "learning_rate": 0.00010054054054054053,
1006
+ "loss": 0.14443959295749664,
1007
+ "num_input_tokens_seen": 94230,
1008
+ "step": 100,
1009
+ "train_runtime": 384.2906,
1010
+ "train_tokens_per_second": 245.205
1011
+ },
1012
+ {
1013
+ "epoch": 0.02390532544378698,
1014
+ "grad_norm": 0.6924043297767639,
1015
+ "learning_rate": 9.945945945945948e-05,
1016
+ "loss": 0.18473166227340698,
1017
+ "num_input_tokens_seen": 95132,
1018
+ "step": 101,
1019
+ "train_runtime": 387.5005,
1020
+ "train_tokens_per_second": 245.502
1021
+ },
1022
+ {
1023
+ "epoch": 0.024142011834319528,
1024
+ "grad_norm": 0.732337236404419,
1025
+ "learning_rate": 9.837837837837839e-05,
1026
+ "loss": 0.16353662312030792,
1027
+ "num_input_tokens_seen": 96080,
1028
+ "step": 102,
1029
+ "train_runtime": 390.584,
1030
+ "train_tokens_per_second": 245.991
1031
+ },
1032
+ {
1033
+ "epoch": 0.02437869822485207,
1034
+ "grad_norm": 0.5956932902336121,
1035
+ "learning_rate": 9.729729729729731e-05,
1036
+ "loss": 0.16359847784042358,
1037
+ "num_input_tokens_seen": 97048,
1038
+ "step": 103,
1039
+ "train_runtime": 393.6474,
1040
+ "train_tokens_per_second": 246.535
1041
+ },
1042
+ {
1043
+ "epoch": 0.024615384615384615,
1044
+ "grad_norm": 0.6137388348579407,
1045
+ "learning_rate": 9.621621621621622e-05,
1046
+ "loss": 0.15779417753219604,
1047
+ "num_input_tokens_seen": 97986,
1048
+ "step": 104,
1049
+ "train_runtime": 396.7042,
1050
+ "train_tokens_per_second": 247.0
1051
+ },
1052
+ {
1053
+ "epoch": 0.02485207100591716,
1054
+ "grad_norm": 0.7936599254608154,
1055
+ "learning_rate": 9.513513513513514e-05,
1056
+ "loss": 0.21357379853725433,
1057
+ "num_input_tokens_seen": 98894,
1058
+ "step": 105,
1059
+ "train_runtime": 399.8228,
1060
+ "train_tokens_per_second": 247.345
1061
+ },
1062
+ {
1063
+ "epoch": 0.025088757396449703,
1064
+ "grad_norm": 0.6574507355690002,
1065
+ "learning_rate": 9.405405405405407e-05,
1066
+ "loss": 0.1685573309659958,
1067
+ "num_input_tokens_seen": 99856,
1068
+ "step": 106,
1069
+ "train_runtime": 402.8867,
1070
+ "train_tokens_per_second": 247.851
1071
+ },
1072
+ {
1073
+ "epoch": 0.02532544378698225,
1074
+ "grad_norm": 0.5891634821891785,
1075
+ "learning_rate": 9.297297297297299e-05,
1076
+ "loss": 0.12082140147686005,
1077
+ "num_input_tokens_seen": 100810,
1078
+ "step": 107,
1079
+ "train_runtime": 405.9752,
1080
+ "train_tokens_per_second": 248.316
1081
+ },
1082
+ {
1083
+ "epoch": 0.025562130177514793,
1084
+ "grad_norm": 0.6624342799186707,
1085
+ "learning_rate": 9.18918918918919e-05,
1086
+ "loss": 0.1473795771598816,
1087
+ "num_input_tokens_seen": 101744,
1088
+ "step": 108,
1089
+ "train_runtime": 409.0084,
1090
+ "train_tokens_per_second": 248.758
1091
+ },
1092
+ {
1093
+ "epoch": 0.025798816568047337,
1094
+ "grad_norm": 0.5968036651611328,
1095
+ "learning_rate": 9.081081081081082e-05,
1096
+ "loss": 0.12923751771450043,
1097
+ "num_input_tokens_seen": 102702,
1098
+ "step": 109,
1099
+ "train_runtime": 412.199,
1100
+ "train_tokens_per_second": 249.156
1101
+ },
1102
+ {
1103
+ "epoch": 0.02603550295857988,
1104
+ "grad_norm": 0.5954580307006836,
1105
+ "learning_rate": 8.972972972972973e-05,
1106
+ "loss": 0.1161399856209755,
1107
+ "num_input_tokens_seen": 103672,
1108
+ "step": 110,
1109
+ "train_runtime": 415.3736,
1110
+ "train_tokens_per_second": 249.587
1111
+ },
1112
+ {
1113
+ "epoch": 0.026272189349112424,
1114
+ "grad_norm": 0.6139379739761353,
1115
+ "learning_rate": 8.864864864864866e-05,
1116
+ "loss": 0.12352899461984634,
1117
+ "num_input_tokens_seen": 104598,
1118
+ "step": 111,
1119
+ "train_runtime": 418.3271,
1120
+ "train_tokens_per_second": 250.039
1121
+ },
1122
+ {
1123
+ "epoch": 0.02650887573964497,
1124
+ "grad_norm": 0.7933095097541809,
1125
+ "learning_rate": 8.756756756756758e-05,
1126
+ "loss": 0.1709994524717331,
1127
+ "num_input_tokens_seen": 105580,
1128
+ "step": 112,
1129
+ "train_runtime": 421.3565,
1130
+ "train_tokens_per_second": 250.572
1131
+ },
1132
+ {
1133
+ "epoch": 0.026745562130177515,
1134
+ "grad_norm": 0.7641370892524719,
1135
+ "learning_rate": 8.64864864864865e-05,
1136
+ "loss": 0.13086311519145966,
1137
+ "num_input_tokens_seen": 106444,
1138
+ "step": 113,
1139
+ "train_runtime": 424.4668,
1140
+ "train_tokens_per_second": 250.771
1141
+ },
1142
+ {
1143
+ "epoch": 0.02698224852071006,
1144
+ "grad_norm": 0.7611938118934631,
1145
+ "learning_rate": 8.540540540540541e-05,
1146
+ "loss": 0.16537640988826752,
1147
+ "num_input_tokens_seen": 107454,
1148
+ "step": 114,
1149
+ "train_runtime": 427.5,
1150
+ "train_tokens_per_second": 251.354
1151
+ },
1152
+ {
1153
+ "epoch": 0.027218934911242602,
1154
+ "grad_norm": 0.7495304346084595,
1155
+ "learning_rate": 8.432432432432433e-05,
1156
+ "loss": 0.14228317141532898,
1157
+ "num_input_tokens_seen": 108426,
1158
+ "step": 115,
1159
+ "train_runtime": 430.5035,
1160
+ "train_tokens_per_second": 251.859
1161
+ },
1162
+ {
1163
+ "epoch": 0.02745562130177515,
1164
+ "grad_norm": 1.2078282833099365,
1165
+ "learning_rate": 8.324324324324326e-05,
1166
+ "loss": 0.10530930012464523,
1167
+ "num_input_tokens_seen": 109446,
1168
+ "step": 116,
1169
+ "train_runtime": 433.4964,
1170
+ "train_tokens_per_second": 252.473
1171
+ },
1172
+ {
1173
+ "epoch": 0.027692307692307693,
1174
+ "grad_norm": 0.6765572428703308,
1175
+ "learning_rate": 8.216216216216217e-05,
1176
+ "loss": 0.10717278718948364,
1177
+ "num_input_tokens_seen": 110374,
1178
+ "step": 117,
1179
+ "train_runtime": 436.5901,
1180
+ "train_tokens_per_second": 252.809
1181
+ },
1182
+ {
1183
+ "epoch": 0.027928994082840237,
1184
+ "grad_norm": 0.8501551151275635,
1185
+ "learning_rate": 8.108108108108109e-05,
1186
+ "loss": 0.17784273624420166,
1187
+ "num_input_tokens_seen": 111340,
1188
+ "step": 118,
1189
+ "train_runtime": 439.6526,
1190
+ "train_tokens_per_second": 253.245
1191
+ },
1192
+ {
1193
+ "epoch": 0.02816568047337278,
1194
+ "grad_norm": 0.7583926320075989,
1195
+ "learning_rate": 8e-05,
1196
+ "loss": 0.16588062047958374,
1197
+ "num_input_tokens_seen": 112288,
1198
+ "step": 119,
1199
+ "train_runtime": 442.8046,
1200
+ "train_tokens_per_second": 253.584
1201
+ },
1202
+ {
1203
+ "epoch": 0.028402366863905324,
1204
+ "grad_norm": 0.6620173454284668,
1205
+ "learning_rate": 7.891891891891892e-05,
1206
+ "loss": 0.11676295101642609,
1207
+ "num_input_tokens_seen": 113206,
1208
+ "step": 120,
1209
+ "train_runtime": 445.9682,
1210
+ "train_tokens_per_second": 253.843
1211
+ },
1212
+ {
1213
+ "epoch": 0.02863905325443787,
1214
+ "grad_norm": 0.6844660639762878,
1215
+ "learning_rate": 7.783783783783785e-05,
1216
+ "loss": 0.09329869598150253,
1217
+ "num_input_tokens_seen": 114134,
1218
+ "step": 121,
1219
+ "train_runtime": 452.5692,
1220
+ "train_tokens_per_second": 252.191
1221
+ },
1222
+ {
1223
+ "epoch": 0.028875739644970415,
1224
+ "grad_norm": 0.7968376874923706,
1225
+ "learning_rate": 7.675675675675677e-05,
1226
+ "loss": 0.13320565223693848,
1227
+ "num_input_tokens_seen": 115088,
1228
+ "step": 122,
1229
+ "train_runtime": 455.6245,
1230
+ "train_tokens_per_second": 252.594
1231
+ },
1232
+ {
1233
+ "epoch": 0.02911242603550296,
1234
+ "grad_norm": 0.8079301118850708,
1235
+ "learning_rate": 7.567567567567568e-05,
1236
+ "loss": 0.1387016326189041,
1237
+ "num_input_tokens_seen": 116006,
1238
+ "step": 123,
1239
+ "train_runtime": 458.7601,
1240
+ "train_tokens_per_second": 252.869
1241
+ },
1242
+ {
1243
+ "epoch": 0.029349112426035502,
1244
+ "grad_norm": 1.1301748752593994,
1245
+ "learning_rate": 7.45945945945946e-05,
1246
+ "loss": 0.197790265083313,
1247
+ "num_input_tokens_seen": 116958,
1248
+ "step": 124,
1249
+ "train_runtime": 461.861,
1250
+ "train_tokens_per_second": 253.232
1251
+ },
1252
+ {
1253
+ "epoch": 0.029585798816568046,
1254
+ "grad_norm": 0.7272042632102966,
1255
+ "learning_rate": 7.351351351351352e-05,
1256
+ "loss": 0.12121891975402832,
1257
+ "num_input_tokens_seen": 117886,
1258
+ "step": 125,
1259
+ "train_runtime": 465.0328,
1260
+ "train_tokens_per_second": 253.5
1261
+ },
1262
+ {
1263
+ "epoch": 0.029822485207100593,
1264
+ "grad_norm": 0.6021063923835754,
1265
+ "learning_rate": 7.243243243243245e-05,
1266
+ "loss": 0.08234579861164093,
1267
+ "num_input_tokens_seen": 118806,
1268
+ "step": 126,
1269
+ "train_runtime": 468.1518,
1270
+ "train_tokens_per_second": 253.777
1271
+ },
1272
+ {
1273
+ "epoch": 0.030059171597633137,
1274
+ "grad_norm": 0.5659137964248657,
1275
+ "learning_rate": 7.135135135135136e-05,
1276
+ "loss": 0.07469362020492554,
1277
+ "num_input_tokens_seen": 119716,
1278
+ "step": 127,
1279
+ "train_runtime": 471.2559,
1280
+ "train_tokens_per_second": 254.036
1281
+ },
1282
+ {
1283
+ "epoch": 0.03029585798816568,
1284
+ "grad_norm": 0.7604466080665588,
1285
+ "learning_rate": 7.027027027027028e-05,
1286
+ "loss": 0.15167035162448883,
1287
+ "num_input_tokens_seen": 120710,
1288
+ "step": 128,
1289
+ "train_runtime": 474.3536,
1290
+ "train_tokens_per_second": 254.473
1291
+ },
1292
+ {
1293
+ "epoch": 0.030532544378698224,
1294
+ "grad_norm": 0.898281455039978,
1295
+ "learning_rate": 6.91891891891892e-05,
1296
+ "loss": 0.18358615040779114,
1297
+ "num_input_tokens_seen": 121668,
1298
+ "step": 129,
1299
+ "train_runtime": 477.4676,
1300
+ "train_tokens_per_second": 254.819
1301
+ },
1302
+ {
1303
+ "epoch": 0.03076923076923077,
1304
+ "grad_norm": 1.046539545059204,
1305
+ "learning_rate": 6.810810810810811e-05,
1306
+ "loss": 0.14173445105552673,
1307
+ "num_input_tokens_seen": 122654,
1308
+ "step": 130,
1309
+ "train_runtime": 480.6242,
1310
+ "train_tokens_per_second": 255.197
1311
+ },
1312
+ {
1313
+ "epoch": 0.031005917159763315,
1314
+ "grad_norm": 0.689860463142395,
1315
+ "learning_rate": 6.702702702702704e-05,
1316
+ "loss": 0.1232977956533432,
1317
+ "num_input_tokens_seen": 123610,
1318
+ "step": 131,
1319
+ "train_runtime": 483.7904,
1320
+ "train_tokens_per_second": 255.503
1321
+ },
1322
+ {
1323
+ "epoch": 0.031242603550295858,
1324
+ "grad_norm": 0.6417210698127747,
1325
+ "learning_rate": 6.594594594594596e-05,
1326
+ "loss": 0.08111131936311722,
1327
+ "num_input_tokens_seen": 124526,
1328
+ "step": 132,
1329
+ "train_runtime": 486.9241,
1330
+ "train_tokens_per_second": 255.74
1331
+ },
1332
+ {
1333
+ "epoch": 0.0314792899408284,
1334
+ "grad_norm": 0.8019670844078064,
1335
+ "learning_rate": 6.486486486486487e-05,
1336
+ "loss": 0.13235916197299957,
1337
+ "num_input_tokens_seen": 125438,
1338
+ "step": 133,
1339
+ "train_runtime": 490.0416,
1340
+ "train_tokens_per_second": 255.974
1341
+ },
1342
+ {
1343
+ "epoch": 0.031715976331360946,
1344
+ "grad_norm": 0.8618372082710266,
1345
+ "learning_rate": 6.378378378378379e-05,
1346
+ "loss": 0.11209192872047424,
1347
+ "num_input_tokens_seen": 126378,
1348
+ "step": 134,
1349
+ "train_runtime": 493.0816,
1350
+ "train_tokens_per_second": 256.302
1351
+ },
1352
+ {
1353
+ "epoch": 0.03195266272189349,
1354
+ "grad_norm": 0.6566123962402344,
1355
+ "learning_rate": 6.27027027027027e-05,
1356
+ "loss": 0.08107412606477737,
1357
+ "num_input_tokens_seen": 127280,
1358
+ "step": 135,
1359
+ "train_runtime": 496.1373,
1360
+ "train_tokens_per_second": 256.542
1361
+ },
1362
+ {
1363
+ "epoch": 0.03218934911242603,
1364
+ "grad_norm": 0.671886682510376,
1365
+ "learning_rate": 6.162162162162163e-05,
1366
+ "loss": 0.10395485162734985,
1367
+ "num_input_tokens_seen": 128246,
1368
+ "step": 136,
1369
+ "train_runtime": 499.2198,
1370
+ "train_tokens_per_second": 256.893
1371
+ },
1372
+ {
1373
+ "epoch": 0.032426035502958576,
1374
+ "grad_norm": 0.5853029489517212,
1375
+ "learning_rate": 6.0540540540540543e-05,
1376
+ "loss": 0.06830425560474396,
1377
+ "num_input_tokens_seen": 129142,
1378
+ "step": 137,
1379
+ "train_runtime": 502.2719,
1380
+ "train_tokens_per_second": 257.116
1381
+ },
1382
+ {
1383
+ "epoch": 0.03266272189349113,
1384
+ "grad_norm": 0.9871621131896973,
1385
+ "learning_rate": 5.9459459459459466e-05,
1386
+ "loss": 0.14138084650039673,
1387
+ "num_input_tokens_seen": 130060,
1388
+ "step": 138,
1389
+ "train_runtime": 505.3687,
1390
+ "train_tokens_per_second": 257.357
1391
+ },
1392
+ {
1393
+ "epoch": 0.03289940828402367,
1394
+ "grad_norm": 0.6716732978820801,
1395
+ "learning_rate": 5.837837837837838e-05,
1396
+ "loss": 0.07525799423456192,
1397
+ "num_input_tokens_seen": 130982,
1398
+ "step": 139,
1399
+ "train_runtime": 508.4248,
1400
+ "train_tokens_per_second": 257.623
1401
+ },
1402
+ {
1403
+ "epoch": 0.033136094674556214,
1404
+ "grad_norm": 0.7177789807319641,
1405
+ "learning_rate": 5.7297297297297305e-05,
1406
+ "loss": 0.09517940878868103,
1407
+ "num_input_tokens_seen": 131900,
1408
+ "step": 140,
1409
+ "train_runtime": 511.57,
1410
+ "train_tokens_per_second": 257.834
1411
+ },
1412
+ {
1413
+ "epoch": 0.03337278106508876,
1414
+ "grad_norm": 0.703551709651947,
1415
+ "learning_rate": 5.621621621621622e-05,
1416
+ "loss": 0.09361469000577927,
1417
+ "num_input_tokens_seen": 132890,
1418
+ "step": 141,
1419
+ "train_runtime": 514.7186,
1420
+ "train_tokens_per_second": 258.18
1421
+ },
1422
+ {
1423
+ "epoch": 0.0336094674556213,
1424
+ "grad_norm": 0.7156357765197754,
1425
+ "learning_rate": 5.5135135135135144e-05,
1426
+ "loss": 0.08832181245088577,
1427
+ "num_input_tokens_seen": 133810,
1428
+ "step": 142,
1429
+ "train_runtime": 517.7644,
1430
+ "train_tokens_per_second": 258.438
1431
+ },
1432
+ {
1433
+ "epoch": 0.033846153846153845,
1434
+ "grad_norm": 0.7757886648178101,
1435
+ "learning_rate": 5.405405405405406e-05,
1436
+ "loss": 0.12478049844503403,
1437
+ "num_input_tokens_seen": 134758,
1438
+ "step": 143,
1439
+ "train_runtime": 520.8215,
1440
+ "train_tokens_per_second": 258.741
1441
+ },
1442
+ {
1443
+ "epoch": 0.03408284023668639,
1444
+ "grad_norm": 0.6047410368919373,
1445
+ "learning_rate": 5.2972972972972976e-05,
1446
+ "loss": 0.07645335048437119,
1447
+ "num_input_tokens_seen": 135724,
1448
+ "step": 144,
1449
+ "train_runtime": 523.9787,
1450
+ "train_tokens_per_second": 259.026
1451
+ },
1452
+ {
1453
+ "epoch": 0.03431952662721893,
1454
+ "grad_norm": 0.7750188708305359,
1455
+ "learning_rate": 5.18918918918919e-05,
1456
+ "loss": 0.10904290527105331,
1457
+ "num_input_tokens_seen": 136662,
1458
+ "step": 145,
1459
+ "train_runtime": 527.0633,
1460
+ "train_tokens_per_second": 259.29
1461
+ },
1462
+ {
1463
+ "epoch": 0.034556213017751476,
1464
+ "grad_norm": 0.6400149464607239,
1465
+ "learning_rate": 5.0810810810810815e-05,
1466
+ "loss": 0.07525563985109329,
1467
+ "num_input_tokens_seen": 137628,
1468
+ "step": 146,
1469
+ "train_runtime": 530.1299,
1470
+ "train_tokens_per_second": 259.612
1471
+ },
1472
+ {
1473
+ "epoch": 0.03479289940828403,
1474
+ "grad_norm": 0.6681646108627319,
1475
+ "learning_rate": 4.972972972972974e-05,
1476
+ "loss": 0.08536849915981293,
1477
+ "num_input_tokens_seen": 138610,
1478
+ "step": 147,
1479
+ "train_runtime": 533.159,
1480
+ "train_tokens_per_second": 259.979
1481
+ },
1482
+ {
1483
+ "epoch": 0.03502958579881657,
1484
+ "grad_norm": 0.6158885359764099,
1485
+ "learning_rate": 4.8648648648648654e-05,
1486
+ "loss": 0.0753958523273468,
1487
+ "num_input_tokens_seen": 139486,
1488
+ "step": 148,
1489
+ "train_runtime": 536.2632,
1490
+ "train_tokens_per_second": 260.107
1491
+ },
1492
+ {
1493
+ "epoch": 0.035266272189349114,
1494
+ "grad_norm": 0.665565550327301,
1495
+ "learning_rate": 4.756756756756757e-05,
1496
+ "loss": 0.08198000490665436,
1497
+ "num_input_tokens_seen": 140440,
1498
+ "step": 149,
1499
+ "train_runtime": 539.3165,
1500
+ "train_tokens_per_second": 260.404
1501
+ },
1502
+ {
1503
+ "epoch": 0.03550295857988166,
1504
+ "grad_norm": 0.6818353533744812,
1505
+ "learning_rate": 4.648648648648649e-05,
1506
+ "loss": 0.0710655227303505,
1507
+ "num_input_tokens_seen": 141344,
1508
+ "step": 150,
1509
+ "train_runtime": 542.3824,
1510
+ "train_tokens_per_second": 260.598
1511
+ }
1512
+ ],
1513
+ "logging_steps": 1,
1514
+ "max_steps": 190,
1515
+ "num_input_tokens_seen": 141344,
1516
+ "num_train_epochs": 1,
1517
+ "save_steps": 30,
1518
+ "stateful_callbacks": {
1519
+ "TrainerControl": {
1520
+ "args": {
1521
+ "should_epoch_stop": false,
1522
+ "should_evaluate": false,
1523
+ "should_log": false,
1524
+ "should_save": true,
1525
+ "should_training_stop": false
1526
+ },
1527
+ "attributes": {}
1528
+ }
1529
+ },
1530
+ "total_flos": 2037689236101120.0,
1531
+ "train_batch_size": 2,
1532
+ "trial_name": null,
1533
+ "trial_params": null
1534
+ }
checkpoint-150/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:34c4ca6d50f04301fb38588516a91421e14a8ad1ffddf12afdfd9ba35cd4b64e
3
+ size 5777
checkpoint-180/README.md ADDED
@@ -0,0 +1,210 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: unsloth/gemma-4-e2b-it-unsloth-bnb-4bit
3
+ library_name: peft
4
+ pipeline_tag: text-generation
5
+ tags:
6
+ - base_model:adapter:unsloth/gemma-4-e2b-it-unsloth-bnb-4bit
7
+ - lora
8
+ - sft
9
+ - transformers
10
+ - trl
11
+ - unsloth
12
+ ---
13
+
14
+ # Model Card for Model ID
15
+
16
+ <!-- Provide a quick summary of what the model is/does. -->
17
+
18
+
19
+
20
+ ## Model Details
21
+
22
+ ### Model Description
23
+
24
+ <!-- Provide a longer summary of what this model is. -->
25
+
26
+
27
+
28
+ - **Developed by:** [More Information Needed]
29
+ - **Funded by [optional]:** [More Information Needed]
30
+ - **Shared by [optional]:** [More Information Needed]
31
+ - **Model type:** [More Information Needed]
32
+ - **Language(s) (NLP):** [More Information Needed]
33
+ - **License:** [More Information Needed]
34
+ - **Finetuned from model [optional]:** [More Information Needed]
35
+
36
+ ### Model Sources [optional]
37
+
38
+ <!-- Provide the basic links for the model. -->
39
+
40
+ - **Repository:** [More Information Needed]
41
+ - **Paper [optional]:** [More Information Needed]
42
+ - **Demo [optional]:** [More Information Needed]
43
+
44
+ ## Uses
45
+
46
+ <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
47
+
48
+ ### Direct Use
49
+
50
+ <!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
51
+
52
+ [More Information Needed]
53
+
54
+ ### Downstream Use [optional]
55
+
56
+ <!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
57
+
58
+ [More Information Needed]
59
+
60
+ ### Out-of-Scope Use
61
+
62
+ <!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
63
+
64
+ [More Information Needed]
65
+
66
+ ## Bias, Risks, and Limitations
67
+
68
+ <!-- This section is meant to convey both technical and sociotechnical limitations. -->
69
+
70
+ [More Information Needed]
71
+
72
+ ### Recommendations
73
+
74
+ <!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
75
+
76
+ Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
77
+
78
+ ## How to Get Started with the Model
79
+
80
+ Use the code below to get started with the model.
81
+
82
+ [More Information Needed]
83
+
84
+ ## Training Details
85
+
86
+ ### Training Data
87
+
88
+ <!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
89
+
90
+ [More Information Needed]
91
+
92
+ ### Training Procedure
93
+
94
+ <!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
95
+
96
+ #### Preprocessing [optional]
97
+
98
+ [More Information Needed]
99
+
100
+
101
+ #### Training Hyperparameters
102
+
103
+ - **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
104
+
105
+ #### Speeds, Sizes, Times [optional]
106
+
107
+ <!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
108
+
109
+ [More Information Needed]
110
+
111
+ ## Evaluation
112
+
113
+ <!-- This section describes the evaluation protocols and provides the results. -->
114
+
115
+ ### Testing Data, Factors & Metrics
116
+
117
+ #### Testing Data
118
+
119
+ <!-- This should link to a Dataset Card if possible. -->
120
+
121
+ [More Information Needed]
122
+
123
+ #### Factors
124
+
125
+ <!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
126
+
127
+ [More Information Needed]
128
+
129
+ #### Metrics
130
+
131
+ <!-- These are the evaluation metrics being used, ideally with a description of why. -->
132
+
133
+ [More Information Needed]
134
+
135
+ ### Results
136
+
137
+ [More Information Needed]
138
+
139
+ #### Summary
140
+
141
+
142
+
143
+ ## Model Examination [optional]
144
+
145
+ <!-- Relevant interpretability work for the model goes here -->
146
+
147
+ [More Information Needed]
148
+
149
+ ## Environmental Impact
150
+
151
+ <!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
152
+
153
+ Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
154
+
155
+ - **Hardware Type:** [More Information Needed]
156
+ - **Hours used:** [More Information Needed]
157
+ - **Cloud Provider:** [More Information Needed]
158
+ - **Compute Region:** [More Information Needed]
159
+ - **Carbon Emitted:** [More Information Needed]
160
+
161
+ ## Technical Specifications [optional]
162
+
163
+ ### Model Architecture and Objective
164
+
165
+ [More Information Needed]
166
+
167
+ ### Compute Infrastructure
168
+
169
+ [More Information Needed]
170
+
171
+ #### Hardware
172
+
173
+ [More Information Needed]
174
+
175
+ #### Software
176
+
177
+ [More Information Needed]
178
+
179
+ ## Citation [optional]
180
+
181
+ <!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
182
+
183
+ **BibTeX:**
184
+
185
+ [More Information Needed]
186
+
187
+ **APA:**
188
+
189
+ [More Information Needed]
190
+
191
+ ## Glossary [optional]
192
+
193
+ <!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
194
+
195
+ [More Information Needed]
196
+
197
+ ## More Information [optional]
198
+
199
+ [More Information Needed]
200
+
201
+ ## Model Card Authors [optional]
202
+
203
+ [More Information Needed]
204
+
205
+ ## Model Card Contact
206
+
207
+ [More Information Needed]
208
+ ### Framework versions
209
+
210
+ - PEFT 0.18.1
checkpoint-180/adapter_config.json ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alora_invocation_tokens": null,
3
+ "alpha_pattern": {},
4
+ "arrow_config": null,
5
+ "auto_mapping": {
6
+ "base_model_class": "Gemma4ForConditionalGeneration",
7
+ "parent_library": "transformers.models.gemma4.modeling_gemma4",
8
+ "unsloth_fixed": true
9
+ },
10
+ "base_model_name_or_path": "unsloth/gemma-4-e2b-it-unsloth-bnb-4bit",
11
+ "bias": "none",
12
+ "corda_config": null,
13
+ "ensure_weight_tying": false,
14
+ "eva_config": null,
15
+ "exclude_modules": null,
16
+ "fan_in_fan_out": false,
17
+ "inference_mode": true,
18
+ "init_lora_weights": true,
19
+ "layer_replication": null,
20
+ "layers_pattern": null,
21
+ "layers_to_transform": null,
22
+ "loftq_config": {},
23
+ "lora_alpha": 32,
24
+ "lora_bias": false,
25
+ "lora_dropout": 0.0,
26
+ "megatron_config": null,
27
+ "megatron_core": "megatron.core",
28
+ "modules_to_save": null,
29
+ "peft_type": "LORA",
30
+ "peft_version": "0.18.1",
31
+ "qalora_group_size": 16,
32
+ "r": 16,
33
+ "rank_pattern": {},
34
+ "revision": null,
35
+ "target_modules": [
36
+ "q_proj",
37
+ "gate_proj",
38
+ "v_proj",
39
+ "k_proj",
40
+ "down_proj",
41
+ "o_proj",
42
+ "up_proj"
43
+ ],
44
+ "target_parameters": null,
45
+ "task_type": "CAUSAL_LM",
46
+ "trainable_token_indices": null,
47
+ "use_dora": false,
48
+ "use_qalora": false,
49
+ "use_rslora": false
50
+ }
checkpoint-180/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4e2c1029454f1834f511c4c2d3a52d76531655516a9d3045514d762b5156feb6
3
+ size 124277728
checkpoint-180/chat_template.jinja ADDED
@@ -0,0 +1,351 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {%- macro format_parameters(properties, required, filter_keys=false) -%}
2
+ {%- set standard_keys = ['description', 'type', 'properties', 'required', 'nullable'] -%}
3
+ {%- set ns = namespace(found_first=false) -%}
4
+ {%- for key, value in properties | dictsort -%}
5
+ {%- set add_comma = false -%}
6
+ {%- if not filter_keys or key not in standard_keys -%}
7
+ {%- if ns.found_first %},{% endif -%}
8
+ {%- set ns.found_first = true -%}
9
+ {{ key }}:{
10
+ {%- if value['description'] -%}
11
+ description:<|"|>{{ value['description'] }}<|"|>
12
+ {%- set add_comma = true -%}
13
+ {%- endif -%}
14
+ {%- if value['type'] | upper == 'STRING' -%}
15
+ {%- if value['enum'] -%}
16
+ {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
17
+ enum:{{ format_argument(value['enum']) }}
18
+ {%- endif -%}
19
+ {%- elif value['type'] | upper == 'ARRAY' -%}
20
+ {%- if value['items'] is mapping and value['items'] -%}
21
+ {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
22
+ items:{
23
+ {%- set ns_items = namespace(found_first=false) -%}
24
+ {%- for item_key, item_value in value['items'] | dictsort -%}
25
+ {%- if item_value is not none -%}
26
+ {%- if ns_items.found_first %},{% endif -%}
27
+ {%- set ns_items.found_first = true -%}
28
+ {%- if item_key == 'properties' -%}
29
+ properties:{
30
+ {%- if item_value is mapping -%}
31
+ {{- format_parameters(item_value, value['items']['required'] | default([])) -}}
32
+ {%- endif -%}
33
+ }
34
+ {%- elif item_key == 'required' -%}
35
+ required:[
36
+ {%- for req_item in item_value -%}
37
+ <|"|>{{- req_item -}}<|"|>
38
+ {%- if not loop.last %},{% endif -%}
39
+ {%- endfor -%}
40
+ ]
41
+ {%- elif item_key == 'type' -%}
42
+ {%- if item_value is string -%}
43
+ type:{{ format_argument(item_value | upper) }}
44
+ {%- else -%}
45
+ type:{{ format_argument(item_value | map('upper') | list) }}
46
+ {%- endif -%}
47
+ {%- else -%}
48
+ {{ item_key }}:{{ format_argument(item_value) }}
49
+ {%- endif -%}
50
+ {%- endif -%}
51
+ {%- endfor -%}
52
+ }
53
+ {%- endif -%}
54
+ {%- endif -%}
55
+ {%- if value['nullable'] %}
56
+ {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
57
+ nullable:true
58
+ {%- endif -%}
59
+ {%- if value['type'] | upper == 'OBJECT' -%}
60
+ {%- if value['properties'] is defined and value['properties'] is mapping -%}
61
+ {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
62
+ properties:{
63
+ {{- format_parameters(value['properties'], value['required'] | default([])) -}}
64
+ }
65
+ {%- elif value is mapping -%}
66
+ {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
67
+ properties:{
68
+ {{- format_parameters(value, value['required'] | default([]), filter_keys=true) -}}
69
+ }
70
+ {%- endif -%}
71
+ {%- if value['required'] -%}
72
+ {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
73
+ required:[
74
+ {%- for item in value['required'] | default([]) -%}
75
+ <|"|>{{- item -}}<|"|>
76
+ {%- if not loop.last %},{% endif -%}
77
+ {%- endfor -%}
78
+ ]
79
+ {%- endif -%}
80
+ {%- endif -%}
81
+ {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
82
+ type:<|"|>{{ value['type'] | upper }}<|"|>}
83
+ {%- endif -%}
84
+ {%- endfor -%}
85
+ {%- endmacro -%}
86
+ {%- macro format_function_declaration(tool_data) -%}
87
+ declaration:{{- tool_data['function']['name'] -}}{description:<|"|>{{- tool_data['function']['description'] -}}<|"|>
88
+ {%- set params = tool_data['function']['parameters'] -%}
89
+ {%- if params -%}
90
+ ,parameters:{
91
+ {%- if params['properties'] -%}
92
+ properties:{ {{- format_parameters(params['properties'], params['required']) -}} },
93
+ {%- endif -%}
94
+ {%- if params['required'] -%}
95
+ required:[
96
+ {%- for item in params['required'] -%}
97
+ <|"|>{{- item -}}<|"|>
98
+ {{- ',' if not loop.last -}}
99
+ {%- endfor -%}
100
+ ],
101
+ {%- endif -%}
102
+ {%- if params['type'] -%}
103
+ type:<|"|>{{- params['type'] | upper -}}<|"|>}
104
+ {%- endif -%}
105
+ {%- endif -%}
106
+ {%- if 'response' in tool_data['function'] -%}
107
+ {%- set response_declaration = tool_data['function']['response'] -%}
108
+ ,response:{
109
+ {%- if response_declaration['description'] -%}
110
+ description:<|"|>{{- response_declaration['description'] -}}<|"|>,
111
+ {%- endif -%}
112
+ {%- if response_declaration['type'] | upper == 'OBJECT' -%}
113
+ type:<|"|>{{- response_declaration['type'] | upper -}}<|"|>}
114
+ {%- endif -%}
115
+ {%- endif -%}
116
+ }
117
+ {%- endmacro -%}
118
+ {%- macro format_argument(argument, escape_keys=True) -%}
119
+ {%- if argument is string -%}
120
+ {{- '<|"|>' + argument + '<|"|>' -}}
121
+ {%- elif argument is boolean -%}
122
+ {{- 'true' if argument else 'false' -}}
123
+ {%- elif argument is mapping -%}
124
+ {{- '{' -}}
125
+ {%- set ns = namespace(found_first=false) -%}
126
+ {%- for key, value in argument | dictsort -%}
127
+ {%- if ns.found_first %},{% endif -%}
128
+ {%- set ns.found_first = true -%}
129
+ {%- if escape_keys -%}
130
+ {{- '<|"|>' + key + '<|"|>' -}}
131
+ {%- else -%}
132
+ {{- key -}}
133
+ {%- endif -%}
134
+ :{{- format_argument(value, escape_keys=escape_keys) -}}
135
+ {%- endfor -%}
136
+ {{- '}' -}}
137
+ {%- elif argument is sequence -%}
138
+ {{- '[' -}}
139
+ {%- for item in argument -%}
140
+ {{- format_argument(item, escape_keys=escape_keys) -}}
141
+ {%- if not loop.last %},{% endif -%}
142
+ {%- endfor -%}
143
+ {{- ']' -}}
144
+ {%- else -%}
145
+ {{- argument -}}
146
+ {%- endif -%}
147
+ {%- endmacro -%}
148
+ {%- macro strip_thinking(text) -%}
149
+ {%- set ns = namespace(result='') -%}
150
+ {%- for part in text.split('<channel|>') -%}
151
+ {%- if '<|channel>' in part -%}
152
+ {%- set ns.result = ns.result + part.split('<|channel>')[0] -%}
153
+ {%- else -%}
154
+ {%- set ns.result = ns.result + part -%}
155
+ {%- endif -%}
156
+ {%- endfor -%}
157
+ {{- ns.result | trim -}}
158
+ {%- endmacro -%}
159
+
160
+ {%- macro format_tool_response_block(tool_name, response) -%}
161
+ {{- '<|tool_response>' -}}
162
+ {%- if response is mapping -%}
163
+ {{- 'response:' + tool_name + '{' -}}
164
+ {%- for key, value in response | dictsort -%}
165
+ {{- key -}}:{{- format_argument(value, escape_keys=False) -}}
166
+ {%- if not loop.last %},{% endif -%}
167
+ {%- endfor -%}
168
+ {{- '}' -}}
169
+ {%- else -%}
170
+ {{- 'response:' + tool_name + '{value:' + format_argument(response, escape_keys=False) + '}' -}}
171
+ {%- endif -%}
172
+ {{- '<tool_response|>' -}}
173
+ {%- endmacro -%}
174
+
175
+ {%- set ns = namespace(prev_message_type=None) -%}
176
+ {%- set loop_messages = messages -%}
177
+ {{- bos_token -}}
178
+ {#- Handle System/Tool Definitions Block -#}
179
+ {%- if (enable_thinking is defined and enable_thinking) or tools or messages[0]['role'] in ['system', 'developer'] -%}
180
+ {{- '<|turn>system\n' -}}
181
+ {#- Inject Thinking token at the very top of the FIRST system turn -#}
182
+ {%- if enable_thinking is defined and enable_thinking -%}
183
+ {{- '<|think|>\n' -}}
184
+ {%- set ns.prev_message_type = 'think' -%}
185
+ {%- endif -%}
186
+ {%- if messages[0]['role'] in ['system', 'developer'] -%}
187
+ {%- if messages[0]['content'] is string -%}
188
+ {{- messages[0]['content'] | trim -}}
189
+ {%- elif messages[0]['content'] is sequence -%}
190
+ {%- for item in messages[0]['content'] -%}
191
+ {{- item['text'] | trim + ' '-}}
192
+ {%- endfor -%}
193
+ {%- endif -%}
194
+ {%- set loop_messages = messages[1:] -%}
195
+ {%- endif -%}
196
+ {%- if tools -%}
197
+ {%- for tool in tools %}
198
+ {{- '<|tool>' -}}
199
+ {{- format_function_declaration(tool) | trim -}}
200
+ {{- '<tool|>' -}}
201
+ {%- endfor %}
202
+ {%- set ns.prev_message_type = 'tool' -%}
203
+ {%- endif -%}
204
+ {{- '<turn|>\n' -}}
205
+ {%- endif %}
206
+
207
+ {#- Pre-scan: find last user message index for reasoning guard -#}
208
+ {%- set ns_turn = namespace(last_user_idx=-1) -%}
209
+ {%- for i in range(loop_messages | length) -%}
210
+ {%- if loop_messages[i]['role'] == 'user' -%}
211
+ {%- set ns_turn.last_user_idx = i -%}
212
+ {%- endif -%}
213
+ {%- endfor -%}
214
+
215
+ {#- Loop through messages -#}
216
+ {%- for message in loop_messages -%}
217
+ {%- if message['role'] != 'tool' -%}
218
+ {%- set ns.prev_message_type = None -%}
219
+ {%- set role = 'model' if message['role'] == 'assistant' else message['role'] -%}
220
+ {#- Detect continuation: suppress duplicate <|turn>model when previous non-tool message was also assistant -#}
221
+ {%- set prev_nt = namespace(role=None, found=false) -%}
222
+ {%- if loop.index0 > 0 -%}
223
+ {%- for j in range(loop.index0 - 1, -1, -1) -%}
224
+ {%- if not prev_nt.found -%}
225
+ {%- if loop_messages[j]['role'] != 'tool' -%}
226
+ {%- set prev_nt.role = loop_messages[j]['role'] -%}
227
+ {%- set prev_nt.found = true -%}
228
+ {%- endif -%}
229
+ {%- endif -%}
230
+ {%- endfor -%}
231
+ {%- endif -%}
232
+ {%- set continue_same_model_turn = (role == 'model' and prev_nt.role == 'assistant') -%}
233
+ {%- if not continue_same_model_turn -%}
234
+ {{- '<|turn>' + role + '\n' }}
235
+ {%- endif -%}
236
+
237
+ {#- Render reasoning/reasoning_content as thinking channel -#}
238
+ {%- set thinking_text = message.get('reasoning') or message.get('reasoning_content') -%}
239
+ {%- if thinking_text and loop.index0 > ns_turn.last_user_idx and message.get('tool_calls') -%}
240
+ {{- '<|channel>thought\n' + thinking_text + '\n<channel|>' -}}
241
+ {%- endif -%}
242
+
243
+ {%- if message['tool_calls'] -%}
244
+ {%- for tool_call in message['tool_calls'] -%}
245
+ {%- set function = tool_call['function'] -%}
246
+ {{- '<|tool_call>call:' + function['name'] + '{' -}}
247
+ {%- if function['arguments'] is mapping -%}
248
+ {%- set ns_args = namespace(found_first=false) -%}
249
+ {%- for key, value in function['arguments'] | dictsort -%}
250
+ {%- if ns_args.found_first %},{% endif -%}
251
+ {%- set ns_args.found_first = true -%}
252
+ {{- key -}}:{{- format_argument(value, escape_keys=False) -}}
253
+ {%- endfor -%}
254
+ {%- elif function['arguments'] is string -%}
255
+ {{- function['arguments'] -}}
256
+ {%- endif -%}
257
+ {{- '}<tool_call|>' -}}
258
+ {%- endfor -%}
259
+ {%- set ns.prev_message_type = 'tool_call' -%}
260
+ {%- endif -%}
261
+
262
+ {%- set ns_tr_out = namespace(flag=false) -%}
263
+ {%- if message.get('tool_responses') -%}
264
+ {#- Legacy: tool_responses embedded on the assistant message (Google/Gemma native) -#}
265
+ {%- for tool_response in message['tool_responses'] -%}
266
+ {{- format_tool_response_block(tool_response['name'] | default('unknown'), tool_response['response']) -}}
267
+ {%- set ns_tr_out.flag = true -%}
268
+ {%- set ns.prev_message_type = 'tool_response' -%}
269
+ {%- endfor -%}
270
+ {%- elif message.get('tool_calls') -%}
271
+ {#- OpenAI Chat Completions: forward-scan consecutive role:tool messages -#}
272
+ {%- set ns_tool_scan = namespace(stopped=false) -%}
273
+ {%- for k in range(loop.index0 + 1, loop_messages | length) -%}
274
+ {%- if ns_tool_scan.stopped -%}
275
+ {%- elif loop_messages[k]['role'] != 'tool' -%}
276
+ {%- set ns_tool_scan.stopped = true -%}
277
+ {%- else -%}
278
+ {%- set follow = loop_messages[k] -%}
279
+ {#- Resolve tool_call_id to function name -#}
280
+ {%- set ns_tname = namespace(name=follow.get('name') | default('unknown')) -%}
281
+ {%- for tc in message['tool_calls'] -%}
282
+ {%- if tc.get('id') == follow.get('tool_call_id') -%}
283
+ {%- set ns_tname.name = tc['function']['name'] -%}
284
+ {%- endif -%}
285
+ {%- endfor -%}
286
+ {#- Handle content as string or content-parts array -#}
287
+ {%- set tool_body = follow.get('content') -%}
288
+ {%- if tool_body is string -%}
289
+ {{- format_tool_response_block(ns_tname.name, tool_body) -}}
290
+ {%- elif tool_body is sequence and tool_body is not string -%}
291
+ {%- set ns_txt = namespace(s='') -%}
292
+ {%- for part in tool_body -%}
293
+ {%- if part.get('type') == 'text' -%}
294
+ {%- set ns_txt.s = ns_txt.s + (part.get('text') | default('')) -%}
295
+ {%- endif -%}
296
+ {%- endfor -%}
297
+ {{- format_tool_response_block(ns_tname.name, ns_txt.s) -}}
298
+ {%- else -%}
299
+ {{- format_tool_response_block(ns_tname.name, tool_body) -}}
300
+ {%- endif -%}
301
+ {%- set ns_tr_out.flag = true -%}
302
+ {%- set ns.prev_message_type = 'tool_response' -%}
303
+ {%- endif -%}
304
+ {%- endfor -%}
305
+ {%- endif -%}
306
+
307
+ {%- set captured_content -%}
308
+ {%- if message['content'] is string -%}
309
+ {%- if role == 'model' -%}
310
+ {{- strip_thinking(message['content']) -}}
311
+ {%- else -%}
312
+ {{- message['content'] | trim -}}
313
+ {%- endif -%}
314
+ {%- elif message['content'] is sequence -%}
315
+ {%- for item in message['content'] -%}
316
+ {%- if item['type'] == 'text' -%}
317
+ {%- if role == 'model' -%}
318
+ {{- strip_thinking(item['text']) -}}
319
+ {%- else -%}
320
+ {{- item['text'] | trim -}}
321
+ {%- endif -%}
322
+ {%- elif item['type'] == 'image' -%}
323
+ {{- '<|image|>' -}}
324
+ {%- set ns.prev_message_type = 'image' -%}
325
+ {%- elif item['type'] == 'audio' -%}
326
+ {{- '<|audio|>' -}}
327
+ {%- set ns.prev_message_type = 'audio' -%}
328
+ {%- elif item['type'] == 'video' -%}
329
+ {{- '<|video|>' -}}
330
+ {%- set ns.prev_message_type = 'video' -%}
331
+ {%- endif -%}
332
+ {%- endfor -%}
333
+ {%- endif -%}
334
+ {%- endset -%}
335
+
336
+ {{- captured_content -}}
337
+ {%- set has_content = captured_content | trim | length > 0 -%}
338
+
339
+ {%- if ns.prev_message_type == 'tool_call' and not ns_tr_out.flag -%}
340
+ {{- '<|tool_response>' -}}
341
+ {%- elif not (ns_tr_out.flag and not has_content) -%}
342
+ {{- '<turn|>\n' -}}
343
+ {%- endif -%}
344
+ {%- endif -%}
345
+ {%- endfor -%}
346
+
347
+ {%- if add_generation_prompt -%}
348
+ {%- if ns.prev_message_type != 'tool_response' and ns.prev_message_type != 'tool_call' -%}
349
+ {{- '<|turn>model\n' -}}
350
+ {%- endif -%}
351
+ {%- endif -%}
checkpoint-180/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d28a0dc846feefdc15b5a30c4ff928dcdd8416bb2c13beef4fa5f4ca242bd821
3
+ size 52047117
checkpoint-180/processor_config.json ADDED
@@ -0,0 +1,75 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "audio_ms_per_token": 40,
3
+ "audio_seq_length": 750,
4
+ "feature_extractor": {
5
+ "dither": 0.0,
6
+ "feature_extractor_type": "Gemma4AudioFeatureExtractor",
7
+ "feature_size": 128,
8
+ "fft_length": 512,
9
+ "fft_overdrive": false,
10
+ "frame_length": 320,
11
+ "hop_length": 160,
12
+ "input_scale_factor": 1.0,
13
+ "max_frequency": 8000.0,
14
+ "mel_floor": 0.001,
15
+ "min_frequency": 0.0,
16
+ "padding_side": "left",
17
+ "padding_value": 0.0,
18
+ "per_bin_mean": null,
19
+ "per_bin_stddev": null,
20
+ "preemphasis": 0.0,
21
+ "preemphasis_htk_flavor": true,
22
+ "return_attention_mask": true,
23
+ "sampling_rate": 16000
24
+ },
25
+ "image_processor": {
26
+ "do_convert_rgb": true,
27
+ "do_normalize": false,
28
+ "do_rescale": true,
29
+ "do_resize": true,
30
+ "image_mean": [
31
+ 0.0,
32
+ 0.0,
33
+ 0.0
34
+ ],
35
+ "image_processor_type": "Gemma4ImageProcessor",
36
+ "image_seq_length": 280,
37
+ "image_std": [
38
+ 1.0,
39
+ 1.0,
40
+ 1.0
41
+ ],
42
+ "max_soft_tokens": 280,
43
+ "patch_size": 16,
44
+ "pooling_kernel_size": 3,
45
+ "resample": 3,
46
+ "rescale_factor": 0.00392156862745098
47
+ },
48
+ "image_seq_length": 280,
49
+ "processor_class": "Gemma4Processor",
50
+ "video_processor": {
51
+ "do_convert_rgb": true,
52
+ "do_normalize": true,
53
+ "do_rescale": true,
54
+ "do_resize": true,
55
+ "do_sample_frames": true,
56
+ "image_mean": [
57
+ 0.0,
58
+ 0.0,
59
+ 0.0
60
+ ],
61
+ "image_std": [
62
+ 1.0,
63
+ 1.0,
64
+ 1.0
65
+ ],
66
+ "max_soft_tokens": 70,
67
+ "num_frames": 32,
68
+ "patch_size": 16,
69
+ "pooling_kernel_size": 3,
70
+ "resample": 3,
71
+ "rescale_factor": 0.00392156862745098,
72
+ "return_metadata": false,
73
+ "video_processor_type": "Gemma4VideoProcessor"
74
+ }
75
+ }
checkpoint-180/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f1d565802a8e26c4e8a31328752b7a7fdc186d9401aa008e65697d0ad8c22e33
3
+ size 14645
checkpoint-180/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:157cc77e7dd6e9c2bce809a6b120c6be65dd2e62a42b62dfda1d462877a40ec3
3
+ size 1383
checkpoint-180/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ec2bd51ad4fa32551317ffe221c08d896ed511f96c24b01b67eb1975b12513fc
3
+ size 1465
checkpoint-180/tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cc8d3a0ce36466ccc1278bf987df5f71db1719b9ca6b4118264f45cb627bfe0f
3
+ size 32169626
checkpoint-180/tokenizer_config.json ADDED
@@ -0,0 +1,289 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
+ "<|video|>"
18
+ ],
19
+ "image_token": "<|image|>",
20
+ "is_local": false,
21
+ "mask_token": "<mask>",
22
+ "model_max_length": 131072,
23
+ "model_specific_special_tokens": {
24
+ "audio_token": "<|audio|>",
25
+ "boa_token": "<|audio>",
26
+ "boi_token": "<|image>",
27
+ "eoa_token": "<audio|>",
28
+ "eoc_token": "<channel|>",
29
+ "eoi_token": "<image|>",
30
+ "eot_token": "<turn|>",
31
+ "escape_token": "<|\"|>",
32
+ "etc_token": "<tool_call|>",
33
+ "etd_token": "<tool|>",
34
+ "etr_token": "<tool_response|>",
35
+ "image_token": "<|image|>",
36
+ "soc_token": "<|channel>",
37
+ "sot_token": "<|turn>",
38
+ "stc_token": "<|tool_call>",
39
+ "std_token": "<|tool>",
40
+ "str_token": "<|tool_response>",
41
+ "think_token": "<|think|>"
42
+ },
43
+ "pad_token": "<pad>",
44
+ "padding_side": "right",
45
+ "processor_class": "Gemma4Processor",
46
+ "response_schema": {
47
+ "properties": {
48
+ "content": {
49
+ "type": "string"
50
+ },
51
+ "role": {
52
+ "const": "assistant"
53
+ },
54
+ "thinking": {
55
+ "type": "string"
56
+ },
57
+ "tool_calls": {
58
+ "items": {
59
+ "properties": {
60
+ "function": {
61
+ "properties": {
62
+ "arguments": {
63
+ "additionalProperties": {},
64
+ "type": "object",
65
+ "x-parser": "gemma4-tool-call"
66
+ },
67
+ "name": {
68
+ "type": "string"
69
+ }
70
+ },
71
+ "type": "object",
72
+ "x-regex": "call\\:(?P<name>\\w+)(?P<arguments>\\{.*\\})"
73
+ },
74
+ "type": {
75
+ "const": "function"
76
+ }
77
+ },
78
+ "type": "object"
79
+ },
80
+ "type": "array",
81
+ "x-regex-iterator": "<\\|tool_call>(.*?)<tool_call\\|>"
82
+ }
83
+ },
84
+ "type": "object",
85
+ "x-regex": "(\\<\\|channel\\>thought\\n(?P<thinking>.*?)\\<channel\\|\\>)?(?P<tool_calls>\\<\\|tool_call\\>.*\\<tool_call\\|\\>)?(?P<content>(?:(?!\\<turn\\|\\>)(?!\\<\\|tool_response\\>).)+)?(?:\\<turn\\|\\>|\\<\\|tool_response\\>)?"
86
+ },
87
+ "soc_token": "<|channel>",
88
+ "sot_token": "<|turn>",
89
+ "stc_token": "<|tool_call>",
90
+ "std_token": "<|tool>",
91
+ "str_token": "<|tool_response>",
92
+ "think_token": "<|think|>",
93
+ "tokenizer_class": "GemmaTokenizer",
94
+ "unk_token": "<unk>",
95
+ "added_tokens_decoder": {
96
+ "0": {
97
+ "content": "<pad>",
98
+ "single_word": false,
99
+ "lstrip": false,
100
+ "rstrip": false,
101
+ "normalized": false,
102
+ "special": true
103
+ },
104
+ "1": {
105
+ "content": "<eos>",
106
+ "single_word": false,
107
+ "lstrip": false,
108
+ "rstrip": false,
109
+ "normalized": false,
110
+ "special": true
111
+ },
112
+ "2": {
113
+ "content": "<bos>",
114
+ "single_word": false,
115
+ "lstrip": false,
116
+ "rstrip": false,
117
+ "normalized": false,
118
+ "special": true
119
+ },
120
+ "3": {
121
+ "content": "<unk>",
122
+ "single_word": false,
123
+ "lstrip": false,
124
+ "rstrip": false,
125
+ "normalized": false,
126
+ "special": true
127
+ },
128
+ "4": {
129
+ "content": "<mask>",
130
+ "single_word": false,
131
+ "lstrip": false,
132
+ "rstrip": false,
133
+ "normalized": false,
134
+ "special": true
135
+ },
136
+ "46": {
137
+ "content": "<|tool>",
138
+ "single_word": false,
139
+ "lstrip": false,
140
+ "rstrip": false,
141
+ "normalized": false,
142
+ "special": true
143
+ },
144
+ "47": {
145
+ "content": "<tool|>",
146
+ "single_word": false,
147
+ "lstrip": false,
148
+ "rstrip": false,
149
+ "normalized": false,
150
+ "special": true
151
+ },
152
+ "48": {
153
+ "content": "<|tool_call>",
154
+ "single_word": false,
155
+ "lstrip": false,
156
+ "rstrip": false,
157
+ "normalized": false,
158
+ "special": true
159
+ },
160
+ "49": {
161
+ "content": "<tool_call|>",
162
+ "single_word": false,
163
+ "lstrip": false,
164
+ "rstrip": false,
165
+ "normalized": false,
166
+ "special": true
167
+ },
168
+ "50": {
169
+ "content": "<|tool_response>",
170
+ "single_word": false,
171
+ "lstrip": false,
172
+ "rstrip": false,
173
+ "normalized": false,
174
+ "special": true
175
+ },
176
+ "51": {
177
+ "content": "<tool_response|>",
178
+ "single_word": false,
179
+ "lstrip": false,
180
+ "rstrip": false,
181
+ "normalized": false,
182
+ "special": true
183
+ },
184
+ "52": {
185
+ "content": "<|\"|>",
186
+ "single_word": false,
187
+ "lstrip": false,
188
+ "rstrip": false,
189
+ "normalized": false,
190
+ "special": true
191
+ },
192
+ "98": {
193
+ "content": "<|think|>",
194
+ "single_word": false,
195
+ "lstrip": false,
196
+ "rstrip": false,
197
+ "normalized": false,
198
+ "special": true
199
+ },
200
+ "100": {
201
+ "content": "<|channel>",
202
+ "single_word": false,
203
+ "lstrip": false,
204
+ "rstrip": false,
205
+ "normalized": false,
206
+ "special": true
207
+ },
208
+ "101": {
209
+ "content": "<channel|>",
210
+ "single_word": false,
211
+ "lstrip": false,
212
+ "rstrip": false,
213
+ "normalized": false,
214
+ "special": true
215
+ },
216
+ "105": {
217
+ "content": "<|turn>",
218
+ "single_word": false,
219
+ "lstrip": false,
220
+ "rstrip": false,
221
+ "normalized": false,
222
+ "special": true
223
+ },
224
+ "106": {
225
+ "content": "<turn|>",
226
+ "single_word": false,
227
+ "lstrip": false,
228
+ "rstrip": false,
229
+ "normalized": false,
230
+ "special": true
231
+ },
232
+ "255999": {
233
+ "content": "<|image>",
234
+ "single_word": false,
235
+ "lstrip": false,
236
+ "rstrip": false,
237
+ "normalized": false,
238
+ "special": true
239
+ },
240
+ "256000": {
241
+ "content": "<|audio>",
242
+ "single_word": false,
243
+ "lstrip": false,
244
+ "rstrip": false,
245
+ "normalized": false,
246
+ "special": true
247
+ },
248
+ "258880": {
249
+ "content": "<|image|>",
250
+ "single_word": false,
251
+ "lstrip": false,
252
+ "rstrip": false,
253
+ "normalized": false,
254
+ "special": true
255
+ },
256
+ "258881": {
257
+ "content": "<|audio|>",
258
+ "single_word": false,
259
+ "lstrip": false,
260
+ "rstrip": false,
261
+ "normalized": false,
262
+ "special": true
263
+ },
264
+ "258882": {
265
+ "content": "<image|>",
266
+ "single_word": false,
267
+ "lstrip": false,
268
+ "rstrip": false,
269
+ "normalized": false,
270
+ "special": true
271
+ },
272
+ "258883": {
273
+ "content": "<audio|>",
274
+ "single_word": false,
275
+ "lstrip": false,
276
+ "rstrip": false,
277
+ "normalized": false,
278
+ "special": true
279
+ },
280
+ "258884": {
281
+ "content": "<|video|>",
282
+ "single_word": false,
283
+ "lstrip": false,
284
+ "rstrip": false,
285
+ "normalized": false,
286
+ "special": true
287
+ }
288
+ }
289
+ }
checkpoint-180/trainer_state.json ADDED
@@ -0,0 +1,1834 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 0.04260355029585799,
6
+ "eval_steps": 500,
7
+ "global_step": 180,
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.00023668639053254438,
14
+ "grad_norm": NaN,
15
+ "learning_rate": 0.0,
16
+ "loss": 4.347962379455566,
17
+ "num_input_tokens_seen": 842,
18
+ "step": 1,
19
+ "train_runtime": 63.0289,
20
+ "train_tokens_per_second": 13.359
21
+ },
22
+ {
23
+ "epoch": 0.00047337278106508875,
24
+ "grad_norm": NaN,
25
+ "learning_rate": 0.0,
26
+ "loss": 4.634688854217529,
27
+ "num_input_tokens_seen": 1820,
28
+ "step": 2,
29
+ "train_runtime": 67.7033,
30
+ "train_tokens_per_second": 26.882
31
+ },
32
+ {
33
+ "epoch": 0.0007100591715976331,
34
+ "grad_norm": 13.944212913513184,
35
+ "learning_rate": 0.0,
36
+ "loss": 4.42241096496582,
37
+ "num_input_tokens_seen": 2754,
38
+ "step": 3,
39
+ "train_runtime": 72.324,
40
+ "train_tokens_per_second": 38.079
41
+ },
42
+ {
43
+ "epoch": 0.0009467455621301775,
44
+ "grad_norm": 13.445413589477539,
45
+ "learning_rate": 4e-05,
46
+ "loss": 4.682804107666016,
47
+ "num_input_tokens_seen": 3700,
48
+ "step": 4,
49
+ "train_runtime": 75.3925,
50
+ "train_tokens_per_second": 49.077
51
+ },
52
+ {
53
+ "epoch": 0.001183431952662722,
54
+ "grad_norm": 10.576727867126465,
55
+ "learning_rate": 8e-05,
56
+ "loss": 4.166159629821777,
57
+ "num_input_tokens_seen": 4714,
58
+ "step": 5,
59
+ "train_runtime": 78.474,
60
+ "train_tokens_per_second": 60.071
61
+ },
62
+ {
63
+ "epoch": 0.0014201183431952662,
64
+ "grad_norm": 8.077850341796875,
65
+ "learning_rate": 0.00012,
66
+ "loss": 3.2006261348724365,
67
+ "num_input_tokens_seen": 5674,
68
+ "step": 6,
69
+ "train_runtime": 81.6139,
70
+ "train_tokens_per_second": 69.522
71
+ },
72
+ {
73
+ "epoch": 0.0016568047337278107,
74
+ "grad_norm": 6.60789680480957,
75
+ "learning_rate": 0.00016,
76
+ "loss": 2.7067415714263916,
77
+ "num_input_tokens_seen": 6646,
78
+ "step": 7,
79
+ "train_runtime": 84.786,
80
+ "train_tokens_per_second": 78.386
81
+ },
82
+ {
83
+ "epoch": 0.001893491124260355,
84
+ "grad_norm": 9.175209045410156,
85
+ "learning_rate": 0.0002,
86
+ "loss": 2.367570638656616,
87
+ "num_input_tokens_seen": 7518,
88
+ "step": 8,
89
+ "train_runtime": 87.8178,
90
+ "train_tokens_per_second": 85.609
91
+ },
92
+ {
93
+ "epoch": 0.0021301775147928993,
94
+ "grad_norm": 3.438948631286621,
95
+ "learning_rate": 0.00019891891891891895,
96
+ "loss": 2.1013119220733643,
97
+ "num_input_tokens_seen": 8470,
98
+ "step": 9,
99
+ "train_runtime": 90.9458,
100
+ "train_tokens_per_second": 93.132
101
+ },
102
+ {
103
+ "epoch": 0.002366863905325444,
104
+ "grad_norm": 3.459472894668579,
105
+ "learning_rate": 0.00019783783783783784,
106
+ "loss": 1.8991940021514893,
107
+ "num_input_tokens_seen": 9458,
108
+ "step": 10,
109
+ "train_runtime": 94.0577,
110
+ "train_tokens_per_second": 100.555
111
+ },
112
+ {
113
+ "epoch": 0.0026035502958579883,
114
+ "grad_norm": 2.169750928878784,
115
+ "learning_rate": 0.00019675675675675678,
116
+ "loss": 1.7395055294036865,
117
+ "num_input_tokens_seen": 10456,
118
+ "step": 11,
119
+ "train_runtime": 97.2068,
120
+ "train_tokens_per_second": 107.565
121
+ },
122
+ {
123
+ "epoch": 0.0028402366863905324,
124
+ "grad_norm": 1.8638927936553955,
125
+ "learning_rate": 0.00019567567567567567,
126
+ "loss": 1.5622950792312622,
127
+ "num_input_tokens_seen": 11410,
128
+ "step": 12,
129
+ "train_runtime": 100.2911,
130
+ "train_tokens_per_second": 113.769
131
+ },
132
+ {
133
+ "epoch": 0.003076923076923077,
134
+ "grad_norm": 1.4741116762161255,
135
+ "learning_rate": 0.00019459459459459462,
136
+ "loss": 1.392699956893921,
137
+ "num_input_tokens_seen": 12316,
138
+ "step": 13,
139
+ "train_runtime": 103.3664,
140
+ "train_tokens_per_second": 119.149
141
+ },
142
+ {
143
+ "epoch": 0.0033136094674556214,
144
+ "grad_norm": 1.3238699436187744,
145
+ "learning_rate": 0.00019351351351351353,
146
+ "loss": 1.2535074949264526,
147
+ "num_input_tokens_seen": 13256,
148
+ "step": 14,
149
+ "train_runtime": 106.4031,
150
+ "train_tokens_per_second": 124.583
151
+ },
152
+ {
153
+ "epoch": 0.0035502958579881655,
154
+ "grad_norm": 1.276484489440918,
155
+ "learning_rate": 0.00019243243243243245,
156
+ "loss": 1.135048508644104,
157
+ "num_input_tokens_seen": 14228,
158
+ "step": 15,
159
+ "train_runtime": 109.4909,
160
+ "train_tokens_per_second": 129.947
161
+ },
162
+ {
163
+ "epoch": 0.00378698224852071,
164
+ "grad_norm": 1.1560865640640259,
165
+ "learning_rate": 0.00019135135135135137,
166
+ "loss": 1.0317097902297974,
167
+ "num_input_tokens_seen": 15192,
168
+ "step": 16,
169
+ "train_runtime": 112.5741,
170
+ "train_tokens_per_second": 134.951
171
+ },
172
+ {
173
+ "epoch": 0.004023668639053254,
174
+ "grad_norm": 1.1331353187561035,
175
+ "learning_rate": 0.00019027027027027028,
176
+ "loss": 0.9967219233512878,
177
+ "num_input_tokens_seen": 16130,
178
+ "step": 17,
179
+ "train_runtime": 115.6495,
180
+ "train_tokens_per_second": 139.473
181
+ },
182
+ {
183
+ "epoch": 0.004260355029585799,
184
+ "grad_norm": 1.1567702293395996,
185
+ "learning_rate": 0.0001891891891891892,
186
+ "loss": 0.905368447303772,
187
+ "num_input_tokens_seen": 17078,
188
+ "step": 18,
189
+ "train_runtime": 118.6796,
190
+ "train_tokens_per_second": 143.9
191
+ },
192
+ {
193
+ "epoch": 0.004497041420118343,
194
+ "grad_norm": 0.9157600998878479,
195
+ "learning_rate": 0.00018810810810810814,
196
+ "loss": 0.8075690865516663,
197
+ "num_input_tokens_seen": 18062,
198
+ "step": 19,
199
+ "train_runtime": 121.8602,
200
+ "train_tokens_per_second": 148.219
201
+ },
202
+ {
203
+ "epoch": 0.004733727810650888,
204
+ "grad_norm": 0.8306605815887451,
205
+ "learning_rate": 0.00018702702702702703,
206
+ "loss": 0.8007175922393799,
207
+ "num_input_tokens_seen": 19078,
208
+ "step": 20,
209
+ "train_runtime": 125.0019,
210
+ "train_tokens_per_second": 152.622
211
+ },
212
+ {
213
+ "epoch": 0.004970414201183432,
214
+ "grad_norm": 0.690018355846405,
215
+ "learning_rate": 0.00018594594594594597,
216
+ "loss": 0.7810791730880737,
217
+ "num_input_tokens_seen": 20068,
218
+ "step": 21,
219
+ "train_runtime": 128.0932,
220
+ "train_tokens_per_second": 156.667
221
+ },
222
+ {
223
+ "epoch": 0.005207100591715977,
224
+ "grad_norm": 1.4768867492675781,
225
+ "learning_rate": 0.00018486486486486486,
226
+ "loss": 0.7172120809555054,
227
+ "num_input_tokens_seen": 20972,
228
+ "step": 22,
229
+ "train_runtime": 131.0901,
230
+ "train_tokens_per_second": 159.982
231
+ },
232
+ {
233
+ "epoch": 0.00544378698224852,
234
+ "grad_norm": 0.7999809980392456,
235
+ "learning_rate": 0.0001837837837837838,
236
+ "loss": 0.6333726048469543,
237
+ "num_input_tokens_seen": 21900,
238
+ "step": 23,
239
+ "train_runtime": 134.212,
240
+ "train_tokens_per_second": 163.175
241
+ },
242
+ {
243
+ "epoch": 0.005680473372781065,
244
+ "grad_norm": 0.9285996556282043,
245
+ "learning_rate": 0.00018270270270270272,
246
+ "loss": 0.6750729084014893,
247
+ "num_input_tokens_seen": 22770,
248
+ "step": 24,
249
+ "train_runtime": 137.3193,
250
+ "train_tokens_per_second": 165.818
251
+ },
252
+ {
253
+ "epoch": 0.005917159763313609,
254
+ "grad_norm": 0.7733820080757141,
255
+ "learning_rate": 0.00018162162162162164,
256
+ "loss": 0.672032356262207,
257
+ "num_input_tokens_seen": 23746,
258
+ "step": 25,
259
+ "train_runtime": 140.3175,
260
+ "train_tokens_per_second": 169.23
261
+ },
262
+ {
263
+ "epoch": 0.006153846153846154,
264
+ "grad_norm": 0.73944091796875,
265
+ "learning_rate": 0.00018054054054054055,
266
+ "loss": 0.6716101169586182,
267
+ "num_input_tokens_seen": 24716,
268
+ "step": 26,
269
+ "train_runtime": 143.3683,
270
+ "train_tokens_per_second": 172.395
271
+ },
272
+ {
273
+ "epoch": 0.006390532544378698,
274
+ "grad_norm": 0.7659372091293335,
275
+ "learning_rate": 0.00017945945945945947,
276
+ "loss": 0.6067175269126892,
277
+ "num_input_tokens_seen": 25724,
278
+ "step": 27,
279
+ "train_runtime": 146.5377,
280
+ "train_tokens_per_second": 175.545
281
+ },
282
+ {
283
+ "epoch": 0.006627218934911243,
284
+ "grad_norm": 0.5428617000579834,
285
+ "learning_rate": 0.00017837837837837839,
286
+ "loss": 0.6083395481109619,
287
+ "num_input_tokens_seen": 26712,
288
+ "step": 28,
289
+ "train_runtime": 149.6429,
290
+ "train_tokens_per_second": 178.505
291
+ },
292
+ {
293
+ "epoch": 0.006863905325443787,
294
+ "grad_norm": 0.5915114283561707,
295
+ "learning_rate": 0.00017729729729729733,
296
+ "loss": 0.6463742256164551,
297
+ "num_input_tokens_seen": 27704,
298
+ "step": 29,
299
+ "train_runtime": 152.7501,
300
+ "train_tokens_per_second": 181.368
301
+ },
302
+ {
303
+ "epoch": 0.007100591715976331,
304
+ "grad_norm": 1.3350757360458374,
305
+ "learning_rate": 0.00017621621621621622,
306
+ "loss": 0.5705698132514954,
307
+ "num_input_tokens_seen": 28648,
308
+ "step": 30,
309
+ "train_runtime": 155.8918,
310
+ "train_tokens_per_second": 183.768
311
+ },
312
+ {
313
+ "epoch": 0.0073372781065088755,
314
+ "grad_norm": 0.68421471118927,
315
+ "learning_rate": 0.00017513513513513516,
316
+ "loss": 0.5660746097564697,
317
+ "num_input_tokens_seen": 29622,
318
+ "step": 31,
319
+ "train_runtime": 163.3209,
320
+ "train_tokens_per_second": 181.373
321
+ },
322
+ {
323
+ "epoch": 0.00757396449704142,
324
+ "grad_norm": 0.5762604475021362,
325
+ "learning_rate": 0.00017405405405405405,
326
+ "loss": 0.5483309030532837,
327
+ "num_input_tokens_seen": 30534,
328
+ "step": 32,
329
+ "train_runtime": 166.5185,
330
+ "train_tokens_per_second": 183.367
331
+ },
332
+ {
333
+ "epoch": 0.0078106508875739646,
334
+ "grad_norm": 0.5751784443855286,
335
+ "learning_rate": 0.000172972972972973,
336
+ "loss": 0.5207005739212036,
337
+ "num_input_tokens_seen": 31458,
338
+ "step": 33,
339
+ "train_runtime": 169.7055,
340
+ "train_tokens_per_second": 185.368
341
+ },
342
+ {
343
+ "epoch": 0.008047337278106508,
344
+ "grad_norm": 0.5752039551734924,
345
+ "learning_rate": 0.0001718918918918919,
346
+ "loss": 0.5534584522247314,
347
+ "num_input_tokens_seen": 32424,
348
+ "step": 34,
349
+ "train_runtime": 172.9184,
350
+ "train_tokens_per_second": 187.51
351
+ },
352
+ {
353
+ "epoch": 0.008284023668639054,
354
+ "grad_norm": 0.5918363332748413,
355
+ "learning_rate": 0.00017081081081081083,
356
+ "loss": 0.482683002948761,
357
+ "num_input_tokens_seen": 33398,
358
+ "step": 35,
359
+ "train_runtime": 176.2158,
360
+ "train_tokens_per_second": 189.529
361
+ },
362
+ {
363
+ "epoch": 0.008520710059171597,
364
+ "grad_norm": 0.6921846270561218,
365
+ "learning_rate": 0.00016972972972972974,
366
+ "loss": 0.46112507581710815,
367
+ "num_input_tokens_seen": 34324,
368
+ "step": 36,
369
+ "train_runtime": 179.3173,
370
+ "train_tokens_per_second": 191.415
371
+ },
372
+ {
373
+ "epoch": 0.008757396449704143,
374
+ "grad_norm": 0.6429721117019653,
375
+ "learning_rate": 0.00016864864864864866,
376
+ "loss": 0.45629197359085083,
377
+ "num_input_tokens_seen": 35270,
378
+ "step": 37,
379
+ "train_runtime": 182.3558,
380
+ "train_tokens_per_second": 193.413
381
+ },
382
+ {
383
+ "epoch": 0.008994082840236686,
384
+ "grad_norm": 0.5477020740509033,
385
+ "learning_rate": 0.00016756756756756757,
386
+ "loss": 0.4547877907752991,
387
+ "num_input_tokens_seen": 36206,
388
+ "step": 38,
389
+ "train_runtime": 185.4483,
390
+ "train_tokens_per_second": 195.235
391
+ },
392
+ {
393
+ "epoch": 0.009230769230769232,
394
+ "grad_norm": 0.6158634424209595,
395
+ "learning_rate": 0.00016648648648648652,
396
+ "loss": 0.367482990026474,
397
+ "num_input_tokens_seen": 37080,
398
+ "step": 39,
399
+ "train_runtime": 188.5114,
400
+ "train_tokens_per_second": 196.699
401
+ },
402
+ {
403
+ "epoch": 0.009467455621301775,
404
+ "grad_norm": 0.5405091643333435,
405
+ "learning_rate": 0.0001654054054054054,
406
+ "loss": 0.42049410939216614,
407
+ "num_input_tokens_seen": 38030,
408
+ "step": 40,
409
+ "train_runtime": 191.5986,
410
+ "train_tokens_per_second": 198.488
411
+ },
412
+ {
413
+ "epoch": 0.009704142011834319,
414
+ "grad_norm": 0.6389183402061462,
415
+ "learning_rate": 0.00016432432432432435,
416
+ "loss": 0.3935014307498932,
417
+ "num_input_tokens_seen": 38942,
418
+ "step": 41,
419
+ "train_runtime": 194.8426,
420
+ "train_tokens_per_second": 199.864
421
+ },
422
+ {
423
+ "epoch": 0.009940828402366864,
424
+ "grad_norm": 0.6742956042289734,
425
+ "learning_rate": 0.00016324324324324324,
426
+ "loss": 0.4373692274093628,
427
+ "num_input_tokens_seen": 39902,
428
+ "step": 42,
429
+ "train_runtime": 197.8938,
430
+ "train_tokens_per_second": 201.633
431
+ },
432
+ {
433
+ "epoch": 0.010177514792899408,
434
+ "grad_norm": 0.6083794236183167,
435
+ "learning_rate": 0.00016216216216216218,
436
+ "loss": 0.3776765465736389,
437
+ "num_input_tokens_seen": 40858,
438
+ "step": 43,
439
+ "train_runtime": 200.9708,
440
+ "train_tokens_per_second": 203.303
441
+ },
442
+ {
443
+ "epoch": 0.010414201183431953,
444
+ "grad_norm": 0.5469992160797119,
445
+ "learning_rate": 0.0001610810810810811,
446
+ "loss": 0.31133151054382324,
447
+ "num_input_tokens_seen": 41760,
448
+ "step": 44,
449
+ "train_runtime": 204.1304,
450
+ "train_tokens_per_second": 204.575
451
+ },
452
+ {
453
+ "epoch": 0.010650887573964497,
454
+ "grad_norm": 0.6452498435974121,
455
+ "learning_rate": 0.00016,
456
+ "loss": 0.42721521854400635,
457
+ "num_input_tokens_seen": 42740,
458
+ "step": 45,
459
+ "train_runtime": 207.1845,
460
+ "train_tokens_per_second": 206.29
461
+ },
462
+ {
463
+ "epoch": 0.01088757396449704,
464
+ "grad_norm": 0.5988945960998535,
465
+ "learning_rate": 0.00015891891891891893,
466
+ "loss": 0.41375258564949036,
467
+ "num_input_tokens_seen": 43750,
468
+ "step": 46,
469
+ "train_runtime": 210.2912,
470
+ "train_tokens_per_second": 208.045
471
+ },
472
+ {
473
+ "epoch": 0.011124260355029586,
474
+ "grad_norm": 0.6452038288116455,
475
+ "learning_rate": 0.00015783783783783785,
476
+ "loss": 0.3485276699066162,
477
+ "num_input_tokens_seen": 44690,
478
+ "step": 47,
479
+ "train_runtime": 213.3684,
480
+ "train_tokens_per_second": 209.45
481
+ },
482
+ {
483
+ "epoch": 0.01136094674556213,
484
+ "grad_norm": 0.7041454911231995,
485
+ "learning_rate": 0.00015675675675675676,
486
+ "loss": 0.3296637535095215,
487
+ "num_input_tokens_seen": 45666,
488
+ "step": 48,
489
+ "train_runtime": 216.4459,
490
+ "train_tokens_per_second": 210.981
491
+ },
492
+ {
493
+ "epoch": 0.011597633136094675,
494
+ "grad_norm": 0.6726324558258057,
495
+ "learning_rate": 0.0001556756756756757,
496
+ "loss": 0.3179457187652588,
497
+ "num_input_tokens_seen": 46560,
498
+ "step": 49,
499
+ "train_runtime": 219.4697,
500
+ "train_tokens_per_second": 212.148
501
+ },
502
+ {
503
+ "epoch": 0.011834319526627219,
504
+ "grad_norm": 0.7513703107833862,
505
+ "learning_rate": 0.0001545945945945946,
506
+ "loss": 0.3722391426563263,
507
+ "num_input_tokens_seen": 47528,
508
+ "step": 50,
509
+ "train_runtime": 222.596,
510
+ "train_tokens_per_second": 213.517
511
+ },
512
+ {
513
+ "epoch": 0.012071005917159764,
514
+ "grad_norm": 0.6454160213470459,
515
+ "learning_rate": 0.00015351351351351354,
516
+ "loss": 0.283376544713974,
517
+ "num_input_tokens_seen": 48468,
518
+ "step": 51,
519
+ "train_runtime": 225.7262,
520
+ "train_tokens_per_second": 214.72
521
+ },
522
+ {
523
+ "epoch": 0.012307692307692308,
524
+ "grad_norm": 0.9116244316101074,
525
+ "learning_rate": 0.00015243243243243243,
526
+ "loss": 0.4286752939224243,
527
+ "num_input_tokens_seen": 49444,
528
+ "step": 52,
529
+ "train_runtime": 228.803,
530
+ "train_tokens_per_second": 216.099
531
+ },
532
+ {
533
+ "epoch": 0.012544378698224851,
534
+ "grad_norm": 0.6157099604606628,
535
+ "learning_rate": 0.00015135135135135137,
536
+ "loss": 0.27355721592903137,
537
+ "num_input_tokens_seen": 50424,
538
+ "step": 53,
539
+ "train_runtime": 231.8413,
540
+ "train_tokens_per_second": 217.494
541
+ },
542
+ {
543
+ "epoch": 0.012781065088757397,
544
+ "grad_norm": 0.6335808038711548,
545
+ "learning_rate": 0.00015027027027027028,
546
+ "loss": 0.37197017669677734,
547
+ "num_input_tokens_seen": 51328,
548
+ "step": 54,
549
+ "train_runtime": 234.9265,
550
+ "train_tokens_per_second": 218.485
551
+ },
552
+ {
553
+ "epoch": 0.01301775147928994,
554
+ "grad_norm": 0.6838424205780029,
555
+ "learning_rate": 0.0001491891891891892,
556
+ "loss": 0.3504602909088135,
557
+ "num_input_tokens_seen": 52274,
558
+ "step": 55,
559
+ "train_runtime": 238.0114,
560
+ "train_tokens_per_second": 219.628
561
+ },
562
+ {
563
+ "epoch": 0.013254437869822486,
564
+ "grad_norm": 0.6566418409347534,
565
+ "learning_rate": 0.00014810810810810812,
566
+ "loss": 0.3189927935600281,
567
+ "num_input_tokens_seen": 53214,
568
+ "step": 56,
569
+ "train_runtime": 241.0892,
570
+ "train_tokens_per_second": 220.723
571
+ },
572
+ {
573
+ "epoch": 0.01349112426035503,
574
+ "grad_norm": 0.6112971901893616,
575
+ "learning_rate": 0.00014702702702702703,
576
+ "loss": 0.3080369830131531,
577
+ "num_input_tokens_seen": 54162,
578
+ "step": 57,
579
+ "train_runtime": 244.2461,
580
+ "train_tokens_per_second": 221.752
581
+ },
582
+ {
583
+ "epoch": 0.013727810650887575,
584
+ "grad_norm": 0.6469757556915283,
585
+ "learning_rate": 0.00014594594594594595,
586
+ "loss": 0.24279898405075073,
587
+ "num_input_tokens_seen": 55158,
588
+ "step": 58,
589
+ "train_runtime": 247.4064,
590
+ "train_tokens_per_second": 222.945
591
+ },
592
+ {
593
+ "epoch": 0.013964497041420118,
594
+ "grad_norm": 0.7099384665489197,
595
+ "learning_rate": 0.0001448648648648649,
596
+ "loss": 0.31188705563545227,
597
+ "num_input_tokens_seen": 56096,
598
+ "step": 59,
599
+ "train_runtime": 250.4708,
600
+ "train_tokens_per_second": 223.962
601
+ },
602
+ {
603
+ "epoch": 0.014201183431952662,
604
+ "grad_norm": 0.5927040576934814,
605
+ "learning_rate": 0.00014378378378378378,
606
+ "loss": 0.25696688890457153,
607
+ "num_input_tokens_seen": 57012,
608
+ "step": 60,
609
+ "train_runtime": 253.7131,
610
+ "train_tokens_per_second": 224.711
611
+ },
612
+ {
613
+ "epoch": 0.014437869822485207,
614
+ "grad_norm": 0.5483338236808777,
615
+ "learning_rate": 0.00014270270270270272,
616
+ "loss": 0.2725352644920349,
617
+ "num_input_tokens_seen": 58036,
618
+ "step": 61,
619
+ "train_runtime": 260.2844,
620
+ "train_tokens_per_second": 222.971
621
+ },
622
+ {
623
+ "epoch": 0.014674556213017751,
624
+ "grad_norm": 0.6806847453117371,
625
+ "learning_rate": 0.00014162162162162161,
626
+ "loss": 0.31959983706474304,
627
+ "num_input_tokens_seen": 59026,
628
+ "step": 62,
629
+ "train_runtime": 263.3802,
630
+ "train_tokens_per_second": 224.109
631
+ },
632
+ {
633
+ "epoch": 0.014911242603550296,
634
+ "grad_norm": 0.6916660070419312,
635
+ "learning_rate": 0.00014054054054054056,
636
+ "loss": 0.312305748462677,
637
+ "num_input_tokens_seen": 59982,
638
+ "step": 63,
639
+ "train_runtime": 266.3794,
640
+ "train_tokens_per_second": 225.175
641
+ },
642
+ {
643
+ "epoch": 0.01514792899408284,
644
+ "grad_norm": 0.5516266226768494,
645
+ "learning_rate": 0.00013945945945945947,
646
+ "loss": 0.2653953433036804,
647
+ "num_input_tokens_seen": 60926,
648
+ "step": 64,
649
+ "train_runtime": 269.3893,
650
+ "train_tokens_per_second": 226.163
651
+ },
652
+ {
653
+ "epoch": 0.015384615384615385,
654
+ "grad_norm": 0.5579270720481873,
655
+ "learning_rate": 0.0001383783783783784,
656
+ "loss": 0.20462818443775177,
657
+ "num_input_tokens_seen": 61844,
658
+ "step": 65,
659
+ "train_runtime": 272.3982,
660
+ "train_tokens_per_second": 227.035
661
+ },
662
+ {
663
+ "epoch": 0.015621301775147929,
664
+ "grad_norm": 0.7645660638809204,
665
+ "learning_rate": 0.0001372972972972973,
666
+ "loss": 0.2650893032550812,
667
+ "num_input_tokens_seen": 62766,
668
+ "step": 66,
669
+ "train_runtime": 275.404,
670
+ "train_tokens_per_second": 227.905
671
+ },
672
+ {
673
+ "epoch": 0.015857988165680473,
674
+ "grad_norm": 0.5974738001823425,
675
+ "learning_rate": 0.00013621621621621622,
676
+ "loss": 0.21247778832912445,
677
+ "num_input_tokens_seen": 63690,
678
+ "step": 67,
679
+ "train_runtime": 278.4021,
680
+ "train_tokens_per_second": 228.77
681
+ },
682
+ {
683
+ "epoch": 0.016094674556213016,
684
+ "grad_norm": 0.5831193327903748,
685
+ "learning_rate": 0.00013513513513513514,
686
+ "loss": 0.2024172842502594,
687
+ "num_input_tokens_seen": 64496,
688
+ "step": 68,
689
+ "train_runtime": 281.4313,
690
+ "train_tokens_per_second": 229.171
691
+ },
692
+ {
693
+ "epoch": 0.016331360946745564,
694
+ "grad_norm": 0.8678795695304871,
695
+ "learning_rate": 0.00013405405405405408,
696
+ "loss": 0.31490638852119446,
697
+ "num_input_tokens_seen": 65460,
698
+ "step": 69,
699
+ "train_runtime": 284.6137,
700
+ "train_tokens_per_second": 229.996
701
+ },
702
+ {
703
+ "epoch": 0.016568047337278107,
704
+ "grad_norm": 0.7611344456672668,
705
+ "learning_rate": 0.00013297297297297297,
706
+ "loss": 0.24661925435066223,
707
+ "num_input_tokens_seen": 66374,
708
+ "step": 70,
709
+ "train_runtime": 287.7742,
710
+ "train_tokens_per_second": 230.646
711
+ },
712
+ {
713
+ "epoch": 0.01680473372781065,
714
+ "grad_norm": 0.6685308814048767,
715
+ "learning_rate": 0.0001318918918918919,
716
+ "loss": 0.1843535602092743,
717
+ "num_input_tokens_seen": 67112,
718
+ "step": 71,
719
+ "train_runtime": 290.9321,
720
+ "train_tokens_per_second": 230.679
721
+ },
722
+ {
723
+ "epoch": 0.017041420118343194,
724
+ "grad_norm": 0.6150883436203003,
725
+ "learning_rate": 0.0001308108108108108,
726
+ "loss": 0.23372295498847961,
727
+ "num_input_tokens_seen": 68154,
728
+ "step": 72,
729
+ "train_runtime": 294.0809,
730
+ "train_tokens_per_second": 231.753
731
+ },
732
+ {
733
+ "epoch": 0.017278106508875738,
734
+ "grad_norm": 0.6938812732696533,
735
+ "learning_rate": 0.00012972972972972974,
736
+ "loss": 0.22619186341762543,
737
+ "num_input_tokens_seen": 69102,
738
+ "step": 73,
739
+ "train_runtime": 297.1843,
740
+ "train_tokens_per_second": 232.522
741
+ },
742
+ {
743
+ "epoch": 0.017514792899408285,
744
+ "grad_norm": 0.617787778377533,
745
+ "learning_rate": 0.00012864864864864866,
746
+ "loss": 0.2535328269004822,
747
+ "num_input_tokens_seen": 70010,
748
+ "step": 74,
749
+ "train_runtime": 300.2851,
750
+ "train_tokens_per_second": 233.145
751
+ },
752
+ {
753
+ "epoch": 0.01775147928994083,
754
+ "grad_norm": 0.6884900331497192,
755
+ "learning_rate": 0.00012756756756756758,
756
+ "loss": 0.21632739901542664,
757
+ "num_input_tokens_seen": 70940,
758
+ "step": 75,
759
+ "train_runtime": 303.3533,
760
+ "train_tokens_per_second": 233.853
761
+ },
762
+ {
763
+ "epoch": 0.017988165680473372,
764
+ "grad_norm": 0.5782189965248108,
765
+ "learning_rate": 0.0001264864864864865,
766
+ "loss": 0.20167675614356995,
767
+ "num_input_tokens_seen": 71862,
768
+ "step": 76,
769
+ "train_runtime": 306.4503,
770
+ "train_tokens_per_second": 234.498
771
+ },
772
+ {
773
+ "epoch": 0.018224852071005916,
774
+ "grad_norm": 0.6541483402252197,
775
+ "learning_rate": 0.0001254054054054054,
776
+ "loss": 0.23231221735477448,
777
+ "num_input_tokens_seen": 72812,
778
+ "step": 77,
779
+ "train_runtime": 309.5743,
780
+ "train_tokens_per_second": 235.2
781
+ },
782
+ {
783
+ "epoch": 0.018461538461538463,
784
+ "grad_norm": 0.7524886727333069,
785
+ "learning_rate": 0.00012432432432432433,
786
+ "loss": 0.27632662653923035,
787
+ "num_input_tokens_seen": 73774,
788
+ "step": 78,
789
+ "train_runtime": 312.6963,
790
+ "train_tokens_per_second": 235.929
791
+ },
792
+ {
793
+ "epoch": 0.018698224852071007,
794
+ "grad_norm": 0.716282308101654,
795
+ "learning_rate": 0.00012324324324324327,
796
+ "loss": 0.2734939754009247,
797
+ "num_input_tokens_seen": 74716,
798
+ "step": 79,
799
+ "train_runtime": 315.7612,
800
+ "train_tokens_per_second": 236.622
801
+ },
802
+ {
803
+ "epoch": 0.01893491124260355,
804
+ "grad_norm": 0.7010631561279297,
805
+ "learning_rate": 0.00012216216216216216,
806
+ "loss": 0.2774362564086914,
807
+ "num_input_tokens_seen": 75692,
808
+ "step": 80,
809
+ "train_runtime": 318.7983,
810
+ "train_tokens_per_second": 237.429
811
+ },
812
+ {
813
+ "epoch": 0.019171597633136094,
814
+ "grad_norm": 0.6561240553855896,
815
+ "learning_rate": 0.00012108108108108109,
816
+ "loss": 0.21640686690807343,
817
+ "num_input_tokens_seen": 76558,
818
+ "step": 81,
819
+ "train_runtime": 321.889,
820
+ "train_tokens_per_second": 237.84
821
+ },
822
+ {
823
+ "epoch": 0.019408284023668638,
824
+ "grad_norm": 0.5552831888198853,
825
+ "learning_rate": 0.00012,
826
+ "loss": 0.21232956647872925,
827
+ "num_input_tokens_seen": 77408,
828
+ "step": 82,
829
+ "train_runtime": 325.0469,
830
+ "train_tokens_per_second": 238.144
831
+ },
832
+ {
833
+ "epoch": 0.019644970414201185,
834
+ "grad_norm": 0.5905120372772217,
835
+ "learning_rate": 0.00011891891891891893,
836
+ "loss": 0.22443550825119019,
837
+ "num_input_tokens_seen": 78278,
838
+ "step": 83,
839
+ "train_runtime": 328.099,
840
+ "train_tokens_per_second": 238.58
841
+ },
842
+ {
843
+ "epoch": 0.01988165680473373,
844
+ "grad_norm": 0.6606366634368896,
845
+ "learning_rate": 0.00011783783783783784,
846
+ "loss": 0.2400747388601303,
847
+ "num_input_tokens_seen": 79262,
848
+ "step": 84,
849
+ "train_runtime": 331.1361,
850
+ "train_tokens_per_second": 239.364
851
+ },
852
+ {
853
+ "epoch": 0.020118343195266272,
854
+ "grad_norm": 0.6508055329322815,
855
+ "learning_rate": 0.00011675675675675676,
856
+ "loss": 0.19072499871253967,
857
+ "num_input_tokens_seen": 80144,
858
+ "step": 85,
859
+ "train_runtime": 334.2158,
860
+ "train_tokens_per_second": 239.797
861
+ },
862
+ {
863
+ "epoch": 0.020355029585798816,
864
+ "grad_norm": 0.606562614440918,
865
+ "learning_rate": 0.00011567567567567568,
866
+ "loss": 0.20951862633228302,
867
+ "num_input_tokens_seen": 81108,
868
+ "step": 86,
869
+ "train_runtime": 337.3105,
870
+ "train_tokens_per_second": 240.455
871
+ },
872
+ {
873
+ "epoch": 0.02059171597633136,
874
+ "grad_norm": 0.6757521629333496,
875
+ "learning_rate": 0.00011459459459459461,
876
+ "loss": 0.17614495754241943,
877
+ "num_input_tokens_seen": 82032,
878
+ "step": 87,
879
+ "train_runtime": 340.3438,
880
+ "train_tokens_per_second": 241.027
881
+ },
882
+ {
883
+ "epoch": 0.020828402366863907,
884
+ "grad_norm": 0.671481728553772,
885
+ "learning_rate": 0.00011351351351351351,
886
+ "loss": 0.1901710331439972,
887
+ "num_input_tokens_seen": 82954,
888
+ "step": 88,
889
+ "train_runtime": 343.4059,
890
+ "train_tokens_per_second": 241.563
891
+ },
892
+ {
893
+ "epoch": 0.02106508875739645,
894
+ "grad_norm": 0.5919027924537659,
895
+ "learning_rate": 0.00011243243243243244,
896
+ "loss": 0.1563737988471985,
897
+ "num_input_tokens_seen": 83880,
898
+ "step": 89,
899
+ "train_runtime": 346.5125,
900
+ "train_tokens_per_second": 242.069
901
+ },
902
+ {
903
+ "epoch": 0.021301775147928994,
904
+ "grad_norm": 0.7108309268951416,
905
+ "learning_rate": 0.00011135135135135135,
906
+ "loss": 0.17256894707679749,
907
+ "num_input_tokens_seen": 84852,
908
+ "step": 90,
909
+ "train_runtime": 349.6508,
910
+ "train_tokens_per_second": 242.676
911
+ },
912
+ {
913
+ "epoch": 0.021538461538461538,
914
+ "grad_norm": 0.8608420491218567,
915
+ "learning_rate": 0.00011027027027027029,
916
+ "loss": 0.1845530867576599,
917
+ "num_input_tokens_seen": 85820,
918
+ "step": 91,
919
+ "train_runtime": 356.267,
920
+ "train_tokens_per_second": 240.887
921
+ },
922
+ {
923
+ "epoch": 0.02177514792899408,
924
+ "grad_norm": 0.9677010774612427,
925
+ "learning_rate": 0.00010918918918918919,
926
+ "loss": 0.24358105659484863,
927
+ "num_input_tokens_seen": 86770,
928
+ "step": 92,
929
+ "train_runtime": 359.3722,
930
+ "train_tokens_per_second": 241.449
931
+ },
932
+ {
933
+ "epoch": 0.02201183431952663,
934
+ "grad_norm": 0.6302966475486755,
935
+ "learning_rate": 0.00010810810810810812,
936
+ "loss": 0.18155133724212646,
937
+ "num_input_tokens_seen": 87736,
938
+ "step": 93,
939
+ "train_runtime": 362.5874,
940
+ "train_tokens_per_second": 241.972
941
+ },
942
+ {
943
+ "epoch": 0.022248520710059172,
944
+ "grad_norm": 0.7171812653541565,
945
+ "learning_rate": 0.00010702702702702702,
946
+ "loss": 0.15349116921424866,
947
+ "num_input_tokens_seen": 88626,
948
+ "step": 94,
949
+ "train_runtime": 365.7055,
950
+ "train_tokens_per_second": 242.343
951
+ },
952
+ {
953
+ "epoch": 0.022485207100591716,
954
+ "grad_norm": 0.7326887845993042,
955
+ "learning_rate": 0.00010594594594594595,
956
+ "loss": 0.18243758380413055,
957
+ "num_input_tokens_seen": 89548,
958
+ "step": 95,
959
+ "train_runtime": 368.799,
960
+ "train_tokens_per_second": 242.81
961
+ },
962
+ {
963
+ "epoch": 0.02272189349112426,
964
+ "grad_norm": 0.6686019897460938,
965
+ "learning_rate": 0.00010486486486486487,
966
+ "loss": 0.15622465312480927,
967
+ "num_input_tokens_seen": 90504,
968
+ "step": 96,
969
+ "train_runtime": 371.8404,
970
+ "train_tokens_per_second": 243.395
971
+ },
972
+ {
973
+ "epoch": 0.022958579881656806,
974
+ "grad_norm": 0.6306503415107727,
975
+ "learning_rate": 0.0001037837837837838,
976
+ "loss": 0.1272117793560028,
977
+ "num_input_tokens_seen": 91454,
978
+ "step": 97,
979
+ "train_runtime": 375.0199,
980
+ "train_tokens_per_second": 243.864
981
+ },
982
+ {
983
+ "epoch": 0.02319526627218935,
984
+ "grad_norm": 0.6068493127822876,
985
+ "learning_rate": 0.0001027027027027027,
986
+ "loss": 0.1577155441045761,
987
+ "num_input_tokens_seen": 92466,
988
+ "step": 98,
989
+ "train_runtime": 378.0978,
990
+ "train_tokens_per_second": 244.556
991
+ },
992
+ {
993
+ "epoch": 0.023431952662721894,
994
+ "grad_norm": 0.7263635993003845,
995
+ "learning_rate": 0.00010162162162162163,
996
+ "loss": 0.1845340132713318,
997
+ "num_input_tokens_seen": 93398,
998
+ "step": 99,
999
+ "train_runtime": 381.1358,
1000
+ "train_tokens_per_second": 245.052
1001
+ },
1002
+ {
1003
+ "epoch": 0.023668639053254437,
1004
+ "grad_norm": 0.6466794013977051,
1005
+ "learning_rate": 0.00010054054054054053,
1006
+ "loss": 0.14443959295749664,
1007
+ "num_input_tokens_seen": 94230,
1008
+ "step": 100,
1009
+ "train_runtime": 384.2906,
1010
+ "train_tokens_per_second": 245.205
1011
+ },
1012
+ {
1013
+ "epoch": 0.02390532544378698,
1014
+ "grad_norm": 0.6924043297767639,
1015
+ "learning_rate": 9.945945945945948e-05,
1016
+ "loss": 0.18473166227340698,
1017
+ "num_input_tokens_seen": 95132,
1018
+ "step": 101,
1019
+ "train_runtime": 387.5005,
1020
+ "train_tokens_per_second": 245.502
1021
+ },
1022
+ {
1023
+ "epoch": 0.024142011834319528,
1024
+ "grad_norm": 0.732337236404419,
1025
+ "learning_rate": 9.837837837837839e-05,
1026
+ "loss": 0.16353662312030792,
1027
+ "num_input_tokens_seen": 96080,
1028
+ "step": 102,
1029
+ "train_runtime": 390.584,
1030
+ "train_tokens_per_second": 245.991
1031
+ },
1032
+ {
1033
+ "epoch": 0.02437869822485207,
1034
+ "grad_norm": 0.5956932902336121,
1035
+ "learning_rate": 9.729729729729731e-05,
1036
+ "loss": 0.16359847784042358,
1037
+ "num_input_tokens_seen": 97048,
1038
+ "step": 103,
1039
+ "train_runtime": 393.6474,
1040
+ "train_tokens_per_second": 246.535
1041
+ },
1042
+ {
1043
+ "epoch": 0.024615384615384615,
1044
+ "grad_norm": 0.6137388348579407,
1045
+ "learning_rate": 9.621621621621622e-05,
1046
+ "loss": 0.15779417753219604,
1047
+ "num_input_tokens_seen": 97986,
1048
+ "step": 104,
1049
+ "train_runtime": 396.7042,
1050
+ "train_tokens_per_second": 247.0
1051
+ },
1052
+ {
1053
+ "epoch": 0.02485207100591716,
1054
+ "grad_norm": 0.7936599254608154,
1055
+ "learning_rate": 9.513513513513514e-05,
1056
+ "loss": 0.21357379853725433,
1057
+ "num_input_tokens_seen": 98894,
1058
+ "step": 105,
1059
+ "train_runtime": 399.8228,
1060
+ "train_tokens_per_second": 247.345
1061
+ },
1062
+ {
1063
+ "epoch": 0.025088757396449703,
1064
+ "grad_norm": 0.6574507355690002,
1065
+ "learning_rate": 9.405405405405407e-05,
1066
+ "loss": 0.1685573309659958,
1067
+ "num_input_tokens_seen": 99856,
1068
+ "step": 106,
1069
+ "train_runtime": 402.8867,
1070
+ "train_tokens_per_second": 247.851
1071
+ },
1072
+ {
1073
+ "epoch": 0.02532544378698225,
1074
+ "grad_norm": 0.5891634821891785,
1075
+ "learning_rate": 9.297297297297299e-05,
1076
+ "loss": 0.12082140147686005,
1077
+ "num_input_tokens_seen": 100810,
1078
+ "step": 107,
1079
+ "train_runtime": 405.9752,
1080
+ "train_tokens_per_second": 248.316
1081
+ },
1082
+ {
1083
+ "epoch": 0.025562130177514793,
1084
+ "grad_norm": 0.6624342799186707,
1085
+ "learning_rate": 9.18918918918919e-05,
1086
+ "loss": 0.1473795771598816,
1087
+ "num_input_tokens_seen": 101744,
1088
+ "step": 108,
1089
+ "train_runtime": 409.0084,
1090
+ "train_tokens_per_second": 248.758
1091
+ },
1092
+ {
1093
+ "epoch": 0.025798816568047337,
1094
+ "grad_norm": 0.5968036651611328,
1095
+ "learning_rate": 9.081081081081082e-05,
1096
+ "loss": 0.12923751771450043,
1097
+ "num_input_tokens_seen": 102702,
1098
+ "step": 109,
1099
+ "train_runtime": 412.199,
1100
+ "train_tokens_per_second": 249.156
1101
+ },
1102
+ {
1103
+ "epoch": 0.02603550295857988,
1104
+ "grad_norm": 0.5954580307006836,
1105
+ "learning_rate": 8.972972972972973e-05,
1106
+ "loss": 0.1161399856209755,
1107
+ "num_input_tokens_seen": 103672,
1108
+ "step": 110,
1109
+ "train_runtime": 415.3736,
1110
+ "train_tokens_per_second": 249.587
1111
+ },
1112
+ {
1113
+ "epoch": 0.026272189349112424,
1114
+ "grad_norm": 0.6139379739761353,
1115
+ "learning_rate": 8.864864864864866e-05,
1116
+ "loss": 0.12352899461984634,
1117
+ "num_input_tokens_seen": 104598,
1118
+ "step": 111,
1119
+ "train_runtime": 418.3271,
1120
+ "train_tokens_per_second": 250.039
1121
+ },
1122
+ {
1123
+ "epoch": 0.02650887573964497,
1124
+ "grad_norm": 0.7933095097541809,
1125
+ "learning_rate": 8.756756756756758e-05,
1126
+ "loss": 0.1709994524717331,
1127
+ "num_input_tokens_seen": 105580,
1128
+ "step": 112,
1129
+ "train_runtime": 421.3565,
1130
+ "train_tokens_per_second": 250.572
1131
+ },
1132
+ {
1133
+ "epoch": 0.026745562130177515,
1134
+ "grad_norm": 0.7641370892524719,
1135
+ "learning_rate": 8.64864864864865e-05,
1136
+ "loss": 0.13086311519145966,
1137
+ "num_input_tokens_seen": 106444,
1138
+ "step": 113,
1139
+ "train_runtime": 424.4668,
1140
+ "train_tokens_per_second": 250.771
1141
+ },
1142
+ {
1143
+ "epoch": 0.02698224852071006,
1144
+ "grad_norm": 0.7611938118934631,
1145
+ "learning_rate": 8.540540540540541e-05,
1146
+ "loss": 0.16537640988826752,
1147
+ "num_input_tokens_seen": 107454,
1148
+ "step": 114,
1149
+ "train_runtime": 427.5,
1150
+ "train_tokens_per_second": 251.354
1151
+ },
1152
+ {
1153
+ "epoch": 0.027218934911242602,
1154
+ "grad_norm": 0.7495304346084595,
1155
+ "learning_rate": 8.432432432432433e-05,
1156
+ "loss": 0.14228317141532898,
1157
+ "num_input_tokens_seen": 108426,
1158
+ "step": 115,
1159
+ "train_runtime": 430.5035,
1160
+ "train_tokens_per_second": 251.859
1161
+ },
1162
+ {
1163
+ "epoch": 0.02745562130177515,
1164
+ "grad_norm": 1.2078282833099365,
1165
+ "learning_rate": 8.324324324324326e-05,
1166
+ "loss": 0.10530930012464523,
1167
+ "num_input_tokens_seen": 109446,
1168
+ "step": 116,
1169
+ "train_runtime": 433.4964,
1170
+ "train_tokens_per_second": 252.473
1171
+ },
1172
+ {
1173
+ "epoch": 0.027692307692307693,
1174
+ "grad_norm": 0.6765572428703308,
1175
+ "learning_rate": 8.216216216216217e-05,
1176
+ "loss": 0.10717278718948364,
1177
+ "num_input_tokens_seen": 110374,
1178
+ "step": 117,
1179
+ "train_runtime": 436.5901,
1180
+ "train_tokens_per_second": 252.809
1181
+ },
1182
+ {
1183
+ "epoch": 0.027928994082840237,
1184
+ "grad_norm": 0.8501551151275635,
1185
+ "learning_rate": 8.108108108108109e-05,
1186
+ "loss": 0.17784273624420166,
1187
+ "num_input_tokens_seen": 111340,
1188
+ "step": 118,
1189
+ "train_runtime": 439.6526,
1190
+ "train_tokens_per_second": 253.245
1191
+ },
1192
+ {
1193
+ "epoch": 0.02816568047337278,
1194
+ "grad_norm": 0.7583926320075989,
1195
+ "learning_rate": 8e-05,
1196
+ "loss": 0.16588062047958374,
1197
+ "num_input_tokens_seen": 112288,
1198
+ "step": 119,
1199
+ "train_runtime": 442.8046,
1200
+ "train_tokens_per_second": 253.584
1201
+ },
1202
+ {
1203
+ "epoch": 0.028402366863905324,
1204
+ "grad_norm": 0.6620173454284668,
1205
+ "learning_rate": 7.891891891891892e-05,
1206
+ "loss": 0.11676295101642609,
1207
+ "num_input_tokens_seen": 113206,
1208
+ "step": 120,
1209
+ "train_runtime": 445.9682,
1210
+ "train_tokens_per_second": 253.843
1211
+ },
1212
+ {
1213
+ "epoch": 0.02863905325443787,
1214
+ "grad_norm": 0.6844660639762878,
1215
+ "learning_rate": 7.783783783783785e-05,
1216
+ "loss": 0.09329869598150253,
1217
+ "num_input_tokens_seen": 114134,
1218
+ "step": 121,
1219
+ "train_runtime": 452.5692,
1220
+ "train_tokens_per_second": 252.191
1221
+ },
1222
+ {
1223
+ "epoch": 0.028875739644970415,
1224
+ "grad_norm": 0.7968376874923706,
1225
+ "learning_rate": 7.675675675675677e-05,
1226
+ "loss": 0.13320565223693848,
1227
+ "num_input_tokens_seen": 115088,
1228
+ "step": 122,
1229
+ "train_runtime": 455.6245,
1230
+ "train_tokens_per_second": 252.594
1231
+ },
1232
+ {
1233
+ "epoch": 0.02911242603550296,
1234
+ "grad_norm": 0.8079301118850708,
1235
+ "learning_rate": 7.567567567567568e-05,
1236
+ "loss": 0.1387016326189041,
1237
+ "num_input_tokens_seen": 116006,
1238
+ "step": 123,
1239
+ "train_runtime": 458.7601,
1240
+ "train_tokens_per_second": 252.869
1241
+ },
1242
+ {
1243
+ "epoch": 0.029349112426035502,
1244
+ "grad_norm": 1.1301748752593994,
1245
+ "learning_rate": 7.45945945945946e-05,
1246
+ "loss": 0.197790265083313,
1247
+ "num_input_tokens_seen": 116958,
1248
+ "step": 124,
1249
+ "train_runtime": 461.861,
1250
+ "train_tokens_per_second": 253.232
1251
+ },
1252
+ {
1253
+ "epoch": 0.029585798816568046,
1254
+ "grad_norm": 0.7272042632102966,
1255
+ "learning_rate": 7.351351351351352e-05,
1256
+ "loss": 0.12121891975402832,
1257
+ "num_input_tokens_seen": 117886,
1258
+ "step": 125,
1259
+ "train_runtime": 465.0328,
1260
+ "train_tokens_per_second": 253.5
1261
+ },
1262
+ {
1263
+ "epoch": 0.029822485207100593,
1264
+ "grad_norm": 0.6021063923835754,
1265
+ "learning_rate": 7.243243243243245e-05,
1266
+ "loss": 0.08234579861164093,
1267
+ "num_input_tokens_seen": 118806,
1268
+ "step": 126,
1269
+ "train_runtime": 468.1518,
1270
+ "train_tokens_per_second": 253.777
1271
+ },
1272
+ {
1273
+ "epoch": 0.030059171597633137,
1274
+ "grad_norm": 0.5659137964248657,
1275
+ "learning_rate": 7.135135135135136e-05,
1276
+ "loss": 0.07469362020492554,
1277
+ "num_input_tokens_seen": 119716,
1278
+ "step": 127,
1279
+ "train_runtime": 471.2559,
1280
+ "train_tokens_per_second": 254.036
1281
+ },
1282
+ {
1283
+ "epoch": 0.03029585798816568,
1284
+ "grad_norm": 0.7604466080665588,
1285
+ "learning_rate": 7.027027027027028e-05,
1286
+ "loss": 0.15167035162448883,
1287
+ "num_input_tokens_seen": 120710,
1288
+ "step": 128,
1289
+ "train_runtime": 474.3536,
1290
+ "train_tokens_per_second": 254.473
1291
+ },
1292
+ {
1293
+ "epoch": 0.030532544378698224,
1294
+ "grad_norm": 0.898281455039978,
1295
+ "learning_rate": 6.91891891891892e-05,
1296
+ "loss": 0.18358615040779114,
1297
+ "num_input_tokens_seen": 121668,
1298
+ "step": 129,
1299
+ "train_runtime": 477.4676,
1300
+ "train_tokens_per_second": 254.819
1301
+ },
1302
+ {
1303
+ "epoch": 0.03076923076923077,
1304
+ "grad_norm": 1.046539545059204,
1305
+ "learning_rate": 6.810810810810811e-05,
1306
+ "loss": 0.14173445105552673,
1307
+ "num_input_tokens_seen": 122654,
1308
+ "step": 130,
1309
+ "train_runtime": 480.6242,
1310
+ "train_tokens_per_second": 255.197
1311
+ },
1312
+ {
1313
+ "epoch": 0.031005917159763315,
1314
+ "grad_norm": 0.689860463142395,
1315
+ "learning_rate": 6.702702702702704e-05,
1316
+ "loss": 0.1232977956533432,
1317
+ "num_input_tokens_seen": 123610,
1318
+ "step": 131,
1319
+ "train_runtime": 483.7904,
1320
+ "train_tokens_per_second": 255.503
1321
+ },
1322
+ {
1323
+ "epoch": 0.031242603550295858,
1324
+ "grad_norm": 0.6417210698127747,
1325
+ "learning_rate": 6.594594594594596e-05,
1326
+ "loss": 0.08111131936311722,
1327
+ "num_input_tokens_seen": 124526,
1328
+ "step": 132,
1329
+ "train_runtime": 486.9241,
1330
+ "train_tokens_per_second": 255.74
1331
+ },
1332
+ {
1333
+ "epoch": 0.0314792899408284,
1334
+ "grad_norm": 0.8019670844078064,
1335
+ "learning_rate": 6.486486486486487e-05,
1336
+ "loss": 0.13235916197299957,
1337
+ "num_input_tokens_seen": 125438,
1338
+ "step": 133,
1339
+ "train_runtime": 490.0416,
1340
+ "train_tokens_per_second": 255.974
1341
+ },
1342
+ {
1343
+ "epoch": 0.031715976331360946,
1344
+ "grad_norm": 0.8618372082710266,
1345
+ "learning_rate": 6.378378378378379e-05,
1346
+ "loss": 0.11209192872047424,
1347
+ "num_input_tokens_seen": 126378,
1348
+ "step": 134,
1349
+ "train_runtime": 493.0816,
1350
+ "train_tokens_per_second": 256.302
1351
+ },
1352
+ {
1353
+ "epoch": 0.03195266272189349,
1354
+ "grad_norm": 0.6566123962402344,
1355
+ "learning_rate": 6.27027027027027e-05,
1356
+ "loss": 0.08107412606477737,
1357
+ "num_input_tokens_seen": 127280,
1358
+ "step": 135,
1359
+ "train_runtime": 496.1373,
1360
+ "train_tokens_per_second": 256.542
1361
+ },
1362
+ {
1363
+ "epoch": 0.03218934911242603,
1364
+ "grad_norm": 0.671886682510376,
1365
+ "learning_rate": 6.162162162162163e-05,
1366
+ "loss": 0.10395485162734985,
1367
+ "num_input_tokens_seen": 128246,
1368
+ "step": 136,
1369
+ "train_runtime": 499.2198,
1370
+ "train_tokens_per_second": 256.893
1371
+ },
1372
+ {
1373
+ "epoch": 0.032426035502958576,
1374
+ "grad_norm": 0.5853029489517212,
1375
+ "learning_rate": 6.0540540540540543e-05,
1376
+ "loss": 0.06830425560474396,
1377
+ "num_input_tokens_seen": 129142,
1378
+ "step": 137,
1379
+ "train_runtime": 502.2719,
1380
+ "train_tokens_per_second": 257.116
1381
+ },
1382
+ {
1383
+ "epoch": 0.03266272189349113,
1384
+ "grad_norm": 0.9871621131896973,
1385
+ "learning_rate": 5.9459459459459466e-05,
1386
+ "loss": 0.14138084650039673,
1387
+ "num_input_tokens_seen": 130060,
1388
+ "step": 138,
1389
+ "train_runtime": 505.3687,
1390
+ "train_tokens_per_second": 257.357
1391
+ },
1392
+ {
1393
+ "epoch": 0.03289940828402367,
1394
+ "grad_norm": 0.6716732978820801,
1395
+ "learning_rate": 5.837837837837838e-05,
1396
+ "loss": 0.07525799423456192,
1397
+ "num_input_tokens_seen": 130982,
1398
+ "step": 139,
1399
+ "train_runtime": 508.4248,
1400
+ "train_tokens_per_second": 257.623
1401
+ },
1402
+ {
1403
+ "epoch": 0.033136094674556214,
1404
+ "grad_norm": 0.7177789807319641,
1405
+ "learning_rate": 5.7297297297297305e-05,
1406
+ "loss": 0.09517940878868103,
1407
+ "num_input_tokens_seen": 131900,
1408
+ "step": 140,
1409
+ "train_runtime": 511.57,
1410
+ "train_tokens_per_second": 257.834
1411
+ },
1412
+ {
1413
+ "epoch": 0.03337278106508876,
1414
+ "grad_norm": 0.703551709651947,
1415
+ "learning_rate": 5.621621621621622e-05,
1416
+ "loss": 0.09361469000577927,
1417
+ "num_input_tokens_seen": 132890,
1418
+ "step": 141,
1419
+ "train_runtime": 514.7186,
1420
+ "train_tokens_per_second": 258.18
1421
+ },
1422
+ {
1423
+ "epoch": 0.0336094674556213,
1424
+ "grad_norm": 0.7156357765197754,
1425
+ "learning_rate": 5.5135135135135144e-05,
1426
+ "loss": 0.08832181245088577,
1427
+ "num_input_tokens_seen": 133810,
1428
+ "step": 142,
1429
+ "train_runtime": 517.7644,
1430
+ "train_tokens_per_second": 258.438
1431
+ },
1432
+ {
1433
+ "epoch": 0.033846153846153845,
1434
+ "grad_norm": 0.7757886648178101,
1435
+ "learning_rate": 5.405405405405406e-05,
1436
+ "loss": 0.12478049844503403,
1437
+ "num_input_tokens_seen": 134758,
1438
+ "step": 143,
1439
+ "train_runtime": 520.8215,
1440
+ "train_tokens_per_second": 258.741
1441
+ },
1442
+ {
1443
+ "epoch": 0.03408284023668639,
1444
+ "grad_norm": 0.6047410368919373,
1445
+ "learning_rate": 5.2972972972972976e-05,
1446
+ "loss": 0.07645335048437119,
1447
+ "num_input_tokens_seen": 135724,
1448
+ "step": 144,
1449
+ "train_runtime": 523.9787,
1450
+ "train_tokens_per_second": 259.026
1451
+ },
1452
+ {
1453
+ "epoch": 0.03431952662721893,
1454
+ "grad_norm": 0.7750188708305359,
1455
+ "learning_rate": 5.18918918918919e-05,
1456
+ "loss": 0.10904290527105331,
1457
+ "num_input_tokens_seen": 136662,
1458
+ "step": 145,
1459
+ "train_runtime": 527.0633,
1460
+ "train_tokens_per_second": 259.29
1461
+ },
1462
+ {
1463
+ "epoch": 0.034556213017751476,
1464
+ "grad_norm": 0.6400149464607239,
1465
+ "learning_rate": 5.0810810810810815e-05,
1466
+ "loss": 0.07525563985109329,
1467
+ "num_input_tokens_seen": 137628,
1468
+ "step": 146,
1469
+ "train_runtime": 530.1299,
1470
+ "train_tokens_per_second": 259.612
1471
+ },
1472
+ {
1473
+ "epoch": 0.03479289940828403,
1474
+ "grad_norm": 0.6681646108627319,
1475
+ "learning_rate": 4.972972972972974e-05,
1476
+ "loss": 0.08536849915981293,
1477
+ "num_input_tokens_seen": 138610,
1478
+ "step": 147,
1479
+ "train_runtime": 533.159,
1480
+ "train_tokens_per_second": 259.979
1481
+ },
1482
+ {
1483
+ "epoch": 0.03502958579881657,
1484
+ "grad_norm": 0.6158885359764099,
1485
+ "learning_rate": 4.8648648648648654e-05,
1486
+ "loss": 0.0753958523273468,
1487
+ "num_input_tokens_seen": 139486,
1488
+ "step": 148,
1489
+ "train_runtime": 536.2632,
1490
+ "train_tokens_per_second": 260.107
1491
+ },
1492
+ {
1493
+ "epoch": 0.035266272189349114,
1494
+ "grad_norm": 0.665565550327301,
1495
+ "learning_rate": 4.756756756756757e-05,
1496
+ "loss": 0.08198000490665436,
1497
+ "num_input_tokens_seen": 140440,
1498
+ "step": 149,
1499
+ "train_runtime": 539.3165,
1500
+ "train_tokens_per_second": 260.404
1501
+ },
1502
+ {
1503
+ "epoch": 0.03550295857988166,
1504
+ "grad_norm": 0.6818353533744812,
1505
+ "learning_rate": 4.648648648648649e-05,
1506
+ "loss": 0.0710655227303505,
1507
+ "num_input_tokens_seen": 141344,
1508
+ "step": 150,
1509
+ "train_runtime": 542.3824,
1510
+ "train_tokens_per_second": 260.598
1511
+ },
1512
+ {
1513
+ "epoch": 0.0357396449704142,
1514
+ "grad_norm": 0.5292275547981262,
1515
+ "learning_rate": 4.540540540540541e-05,
1516
+ "loss": 0.04743894189596176,
1517
+ "num_input_tokens_seen": 142242,
1518
+ "step": 151,
1519
+ "train_runtime": 549.1326,
1520
+ "train_tokens_per_second": 259.03
1521
+ },
1522
+ {
1523
+ "epoch": 0.035976331360946745,
1524
+ "grad_norm": 0.9436084032058716,
1525
+ "learning_rate": 4.432432432432433e-05,
1526
+ "loss": 0.10009428858757019,
1527
+ "num_input_tokens_seen": 143104,
1528
+ "step": 152,
1529
+ "train_runtime": 552.245,
1530
+ "train_tokens_per_second": 259.131
1531
+ },
1532
+ {
1533
+ "epoch": 0.03621301775147929,
1534
+ "grad_norm": 0.8146379590034485,
1535
+ "learning_rate": 4.324324324324325e-05,
1536
+ "loss": 0.08397847414016724,
1537
+ "num_input_tokens_seen": 144036,
1538
+ "step": 153,
1539
+ "train_runtime": 555.3194,
1540
+ "train_tokens_per_second": 259.375
1541
+ },
1542
+ {
1543
+ "epoch": 0.03644970414201183,
1544
+ "grad_norm": 0.6370963454246521,
1545
+ "learning_rate": 4.2162162162162164e-05,
1546
+ "loss": 0.0647936463356018,
1547
+ "num_input_tokens_seen": 144978,
1548
+ "step": 154,
1549
+ "train_runtime": 558.5158,
1550
+ "train_tokens_per_second": 259.577
1551
+ },
1552
+ {
1553
+ "epoch": 0.036686390532544376,
1554
+ "grad_norm": 0.7876921892166138,
1555
+ "learning_rate": 4.108108108108109e-05,
1556
+ "loss": 0.10044769942760468,
1557
+ "num_input_tokens_seen": 145932,
1558
+ "step": 155,
1559
+ "train_runtime": 561.6626,
1560
+ "train_tokens_per_second": 259.821
1561
+ },
1562
+ {
1563
+ "epoch": 0.036923076923076927,
1564
+ "grad_norm": 0.5711075067520142,
1565
+ "learning_rate": 4e-05,
1566
+ "loss": 0.0628824308514595,
1567
+ "num_input_tokens_seen": 146822,
1568
+ "step": 156,
1569
+ "train_runtime": 564.7686,
1570
+ "train_tokens_per_second": 259.968
1571
+ },
1572
+ {
1573
+ "epoch": 0.03715976331360947,
1574
+ "grad_norm": 0.6890460848808289,
1575
+ "learning_rate": 3.8918918918918926e-05,
1576
+ "loss": 0.0714673101902008,
1577
+ "num_input_tokens_seen": 147746,
1578
+ "step": 157,
1579
+ "train_runtime": 567.7959,
1580
+ "train_tokens_per_second": 260.21
1581
+ },
1582
+ {
1583
+ "epoch": 0.037396449704142014,
1584
+ "grad_norm": 0.6936295032501221,
1585
+ "learning_rate": 3.783783783783784e-05,
1586
+ "loss": 0.05670904368162155,
1587
+ "num_input_tokens_seen": 148648,
1588
+ "step": 158,
1589
+ "train_runtime": 570.8255,
1590
+ "train_tokens_per_second": 260.409
1591
+ },
1592
+ {
1593
+ "epoch": 0.03763313609467456,
1594
+ "grad_norm": 0.6268077492713928,
1595
+ "learning_rate": 3.675675675675676e-05,
1596
+ "loss": 0.04924366623163223,
1597
+ "num_input_tokens_seen": 149622,
1598
+ "step": 159,
1599
+ "train_runtime": 573.9724,
1600
+ "train_tokens_per_second": 260.678
1601
+ },
1602
+ {
1603
+ "epoch": 0.0378698224852071,
1604
+ "grad_norm": 0.7037569284439087,
1605
+ "learning_rate": 3.567567567567568e-05,
1606
+ "loss": 0.06151504069566727,
1607
+ "num_input_tokens_seen": 150586,
1608
+ "step": 160,
1609
+ "train_runtime": 577.103,
1610
+ "train_tokens_per_second": 260.934
1611
+ },
1612
+ {
1613
+ "epoch": 0.038106508875739645,
1614
+ "grad_norm": 0.6471953392028809,
1615
+ "learning_rate": 3.45945945945946e-05,
1616
+ "loss": 0.040921665728092194,
1617
+ "num_input_tokens_seen": 151442,
1618
+ "step": 161,
1619
+ "train_runtime": 580.1219,
1620
+ "train_tokens_per_second": 261.052
1621
+ },
1622
+ {
1623
+ "epoch": 0.03834319526627219,
1624
+ "grad_norm": 0.7046090960502625,
1625
+ "learning_rate": 3.351351351351352e-05,
1626
+ "loss": 0.05204992741346359,
1627
+ "num_input_tokens_seen": 152350,
1628
+ "step": 162,
1629
+ "train_runtime": 583.1537,
1630
+ "train_tokens_per_second": 261.252
1631
+ },
1632
+ {
1633
+ "epoch": 0.03857988165680473,
1634
+ "grad_norm": 1.0495421886444092,
1635
+ "learning_rate": 3.2432432432432436e-05,
1636
+ "loss": 0.1057022288441658,
1637
+ "num_input_tokens_seen": 153278,
1638
+ "step": 163,
1639
+ "train_runtime": 586.3134,
1640
+ "train_tokens_per_second": 261.427
1641
+ },
1642
+ {
1643
+ "epoch": 0.038816568047337276,
1644
+ "grad_norm": 0.6432052850723267,
1645
+ "learning_rate": 3.135135135135135e-05,
1646
+ "loss": 0.05509451404213905,
1647
+ "num_input_tokens_seen": 154178,
1648
+ "step": 164,
1649
+ "train_runtime": 589.3544,
1650
+ "train_tokens_per_second": 261.605
1651
+ },
1652
+ {
1653
+ "epoch": 0.03905325443786982,
1654
+ "grad_norm": 0.6547747254371643,
1655
+ "learning_rate": 3.0270270270270272e-05,
1656
+ "loss": 0.04602127522230148,
1657
+ "num_input_tokens_seen": 155080,
1658
+ "step": 165,
1659
+ "train_runtime": 592.3642,
1660
+ "train_tokens_per_second": 261.798
1661
+ },
1662
+ {
1663
+ "epoch": 0.03928994082840237,
1664
+ "grad_norm": 0.6700029969215393,
1665
+ "learning_rate": 2.918918918918919e-05,
1666
+ "loss": 0.05721529945731163,
1667
+ "num_input_tokens_seen": 156140,
1668
+ "step": 166,
1669
+ "train_runtime": 595.4897,
1670
+ "train_tokens_per_second": 262.204
1671
+ },
1672
+ {
1673
+ "epoch": 0.039526627218934914,
1674
+ "grad_norm": 0.9145491719245911,
1675
+ "learning_rate": 2.810810810810811e-05,
1676
+ "loss": 0.06747015565633774,
1677
+ "num_input_tokens_seen": 157106,
1678
+ "step": 167,
1679
+ "train_runtime": 598.5891,
1680
+ "train_tokens_per_second": 262.461
1681
+ },
1682
+ {
1683
+ "epoch": 0.03976331360946746,
1684
+ "grad_norm": 0.8798087239265442,
1685
+ "learning_rate": 2.702702702702703e-05,
1686
+ "loss": 0.08232270181179047,
1687
+ "num_input_tokens_seen": 158056,
1688
+ "step": 168,
1689
+ "train_runtime": 601.7315,
1690
+ "train_tokens_per_second": 262.669
1691
+ },
1692
+ {
1693
+ "epoch": 0.04,
1694
+ "grad_norm": 1.0618623495101929,
1695
+ "learning_rate": 2.594594594594595e-05,
1696
+ "loss": 0.09569647908210754,
1697
+ "num_input_tokens_seen": 159060,
1698
+ "step": 169,
1699
+ "train_runtime": 604.8838,
1700
+ "train_tokens_per_second": 262.96
1701
+ },
1702
+ {
1703
+ "epoch": 0.040236686390532544,
1704
+ "grad_norm": 0.6945156455039978,
1705
+ "learning_rate": 2.486486486486487e-05,
1706
+ "loss": 0.0597086027264595,
1707
+ "num_input_tokens_seen": 160054,
1708
+ "step": 170,
1709
+ "train_runtime": 608.004,
1710
+ "train_tokens_per_second": 263.245
1711
+ },
1712
+ {
1713
+ "epoch": 0.04047337278106509,
1714
+ "grad_norm": 0.6867018342018127,
1715
+ "learning_rate": 2.3783783783783785e-05,
1716
+ "loss": 0.053483735769987106,
1717
+ "num_input_tokens_seen": 160992,
1718
+ "step": 171,
1719
+ "train_runtime": 611.0992,
1720
+ "train_tokens_per_second": 263.447
1721
+ },
1722
+ {
1723
+ "epoch": 0.04071005917159763,
1724
+ "grad_norm": 0.9504451751708984,
1725
+ "learning_rate": 2.2702702702702705e-05,
1726
+ "loss": 0.07532191276550293,
1727
+ "num_input_tokens_seen": 161910,
1728
+ "step": 172,
1729
+ "train_runtime": 614.2523,
1730
+ "train_tokens_per_second": 263.589
1731
+ },
1732
+ {
1733
+ "epoch": 0.040946745562130175,
1734
+ "grad_norm": 0.6639146208763123,
1735
+ "learning_rate": 2.1621621621621624e-05,
1736
+ "loss": 0.04415765404701233,
1737
+ "num_input_tokens_seen": 162810,
1738
+ "step": 173,
1739
+ "train_runtime": 617.2712,
1740
+ "train_tokens_per_second": 263.758
1741
+ },
1742
+ {
1743
+ "epoch": 0.04118343195266272,
1744
+ "grad_norm": 0.5948973894119263,
1745
+ "learning_rate": 2.0540540540540544e-05,
1746
+ "loss": 0.04589955136179924,
1747
+ "num_input_tokens_seen": 163792,
1748
+ "step": 174,
1749
+ "train_runtime": 620.3367,
1750
+ "train_tokens_per_second": 264.037
1751
+ },
1752
+ {
1753
+ "epoch": 0.04142011834319527,
1754
+ "grad_norm": 0.8319454193115234,
1755
+ "learning_rate": 1.9459459459459463e-05,
1756
+ "loss": 0.07309269905090332,
1757
+ "num_input_tokens_seen": 164770,
1758
+ "step": 175,
1759
+ "train_runtime": 623.4097,
1760
+ "train_tokens_per_second": 264.305
1761
+ },
1762
+ {
1763
+ "epoch": 0.04165680473372781,
1764
+ "grad_norm": 0.591090202331543,
1765
+ "learning_rate": 1.837837837837838e-05,
1766
+ "loss": 0.04063474386930466,
1767
+ "num_input_tokens_seen": 165708,
1768
+ "step": 176,
1769
+ "train_runtime": 626.513,
1770
+ "train_tokens_per_second": 264.493
1771
+ },
1772
+ {
1773
+ "epoch": 0.04189349112426036,
1774
+ "grad_norm": 0.5455971360206604,
1775
+ "learning_rate": 1.72972972972973e-05,
1776
+ "loss": 0.0511426106095314,
1777
+ "num_input_tokens_seen": 166654,
1778
+ "step": 177,
1779
+ "train_runtime": 629.5606,
1780
+ "train_tokens_per_second": 264.715
1781
+ },
1782
+ {
1783
+ "epoch": 0.0421301775147929,
1784
+ "grad_norm": 0.7040805816650391,
1785
+ "learning_rate": 1.6216216216216218e-05,
1786
+ "loss": 0.06684890389442444,
1787
+ "num_input_tokens_seen": 167586,
1788
+ "step": 178,
1789
+ "train_runtime": 632.5911,
1790
+ "train_tokens_per_second": 264.92
1791
+ },
1792
+ {
1793
+ "epoch": 0.042366863905325444,
1794
+ "grad_norm": 0.9840870499610901,
1795
+ "learning_rate": 1.5135135135135136e-05,
1796
+ "loss": 0.08544427156448364,
1797
+ "num_input_tokens_seen": 168558,
1798
+ "step": 179,
1799
+ "train_runtime": 635.6343,
1800
+ "train_tokens_per_second": 265.181
1801
+ },
1802
+ {
1803
+ "epoch": 0.04260355029585799,
1804
+ "grad_norm": 0.763282835483551,
1805
+ "learning_rate": 1.4054054054054055e-05,
1806
+ "loss": 0.05402141064405441,
1807
+ "num_input_tokens_seen": 169594,
1808
+ "step": 180,
1809
+ "train_runtime": 638.8133,
1810
+ "train_tokens_per_second": 265.483
1811
+ }
1812
+ ],
1813
+ "logging_steps": 1,
1814
+ "max_steps": 190,
1815
+ "num_input_tokens_seen": 169594,
1816
+ "num_train_epochs": 1,
1817
+ "save_steps": 30,
1818
+ "stateful_callbacks": {
1819
+ "TrainerControl": {
1820
+ "args": {
1821
+ "should_epoch_stop": false,
1822
+ "should_evaluate": false,
1823
+ "should_log": false,
1824
+ "should_save": true,
1825
+ "should_training_stop": false
1826
+ },
1827
+ "attributes": {}
1828
+ }
1829
+ },
1830
+ "total_flos": 2444956052661120.0,
1831
+ "train_batch_size": 2,
1832
+ "trial_name": null,
1833
+ "trial_params": null
1834
+ }
checkpoint-180/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:34c4ca6d50f04301fb38588516a91421e14a8ad1ffddf12afdfd9ba35cd4b64e
3
+ size 5777
checkpoint-190/README.md ADDED
@@ -0,0 +1,210 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: unsloth/gemma-4-e2b-it-unsloth-bnb-4bit
3
+ library_name: peft
4
+ pipeline_tag: text-generation
5
+ tags:
6
+ - base_model:adapter:unsloth/gemma-4-e2b-it-unsloth-bnb-4bit
7
+ - lora
8
+ - sft
9
+ - transformers
10
+ - trl
11
+ - unsloth
12
+ ---
13
+
14
+ # Model Card for Model ID
15
+
16
+ <!-- Provide a quick summary of what the model is/does. -->
17
+
18
+
19
+
20
+ ## Model Details
21
+
22
+ ### Model Description
23
+
24
+ <!-- Provide a longer summary of what this model is. -->
25
+
26
+
27
+
28
+ - **Developed by:** [More Information Needed]
29
+ - **Funded by [optional]:** [More Information Needed]
30
+ - **Shared by [optional]:** [More Information Needed]
31
+ - **Model type:** [More Information Needed]
32
+ - **Language(s) (NLP):** [More Information Needed]
33
+ - **License:** [More Information Needed]
34
+ - **Finetuned from model [optional]:** [More Information Needed]
35
+
36
+ ### Model Sources [optional]
37
+
38
+ <!-- Provide the basic links for the model. -->
39
+
40
+ - **Repository:** [More Information Needed]
41
+ - **Paper [optional]:** [More Information Needed]
42
+ - **Demo [optional]:** [More Information Needed]
43
+
44
+ ## Uses
45
+
46
+ <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
47
+
48
+ ### Direct Use
49
+
50
+ <!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
51
+
52
+ [More Information Needed]
53
+
54
+ ### Downstream Use [optional]
55
+
56
+ <!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
57
+
58
+ [More Information Needed]
59
+
60
+ ### Out-of-Scope Use
61
+
62
+ <!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
63
+
64
+ [More Information Needed]
65
+
66
+ ## Bias, Risks, and Limitations
67
+
68
+ <!-- This section is meant to convey both technical and sociotechnical limitations. -->
69
+
70
+ [More Information Needed]
71
+
72
+ ### Recommendations
73
+
74
+ <!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
75
+
76
+ Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
77
+
78
+ ## How to Get Started with the Model
79
+
80
+ Use the code below to get started with the model.
81
+
82
+ [More Information Needed]
83
+
84
+ ## Training Details
85
+
86
+ ### Training Data
87
+
88
+ <!-- This should link to a Dataset Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
89
+
90
+ [More Information Needed]
91
+
92
+ ### Training Procedure
93
+
94
+ <!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
95
+
96
+ #### Preprocessing [optional]
97
+
98
+ [More Information Needed]
99
+
100
+
101
+ #### Training Hyperparameters
102
+
103
+ - **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
104
+
105
+ #### Speeds, Sizes, Times [optional]
106
+
107
+ <!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
108
+
109
+ [More Information Needed]
110
+
111
+ ## Evaluation
112
+
113
+ <!-- This section describes the evaluation protocols and provides the results. -->
114
+
115
+ ### Testing Data, Factors & Metrics
116
+
117
+ #### Testing Data
118
+
119
+ <!-- This should link to a Dataset Card if possible. -->
120
+
121
+ [More Information Needed]
122
+
123
+ #### Factors
124
+
125
+ <!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
126
+
127
+ [More Information Needed]
128
+
129
+ #### Metrics
130
+
131
+ <!-- These are the evaluation metrics being used, ideally with a description of why. -->
132
+
133
+ [More Information Needed]
134
+
135
+ ### Results
136
+
137
+ [More Information Needed]
138
+
139
+ #### Summary
140
+
141
+
142
+
143
+ ## Model Examination [optional]
144
+
145
+ <!-- Relevant interpretability work for the model goes here -->
146
+
147
+ [More Information Needed]
148
+
149
+ ## Environmental Impact
150
+
151
+ <!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
152
+
153
+ Carbon emissions can be estimated using the [Machine Learning Impact calculator](https://mlco2.github.io/impact#compute) presented in [Lacoste et al. (2019)](https://arxiv.org/abs/1910.09700).
154
+
155
+ - **Hardware Type:** [More Information Needed]
156
+ - **Hours used:** [More Information Needed]
157
+ - **Cloud Provider:** [More Information Needed]
158
+ - **Compute Region:** [More Information Needed]
159
+ - **Carbon Emitted:** [More Information Needed]
160
+
161
+ ## Technical Specifications [optional]
162
+
163
+ ### Model Architecture and Objective
164
+
165
+ [More Information Needed]
166
+
167
+ ### Compute Infrastructure
168
+
169
+ [More Information Needed]
170
+
171
+ #### Hardware
172
+
173
+ [More Information Needed]
174
+
175
+ #### Software
176
+
177
+ [More Information Needed]
178
+
179
+ ## Citation [optional]
180
+
181
+ <!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
182
+
183
+ **BibTeX:**
184
+
185
+ [More Information Needed]
186
+
187
+ **APA:**
188
+
189
+ [More Information Needed]
190
+
191
+ ## Glossary [optional]
192
+
193
+ <!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
194
+
195
+ [More Information Needed]
196
+
197
+ ## More Information [optional]
198
+
199
+ [More Information Needed]
200
+
201
+ ## Model Card Authors [optional]
202
+
203
+ [More Information Needed]
204
+
205
+ ## Model Card Contact
206
+
207
+ [More Information Needed]
208
+ ### Framework versions
209
+
210
+ - PEFT 0.18.1
checkpoint-190/adapter_config.json ADDED
@@ -0,0 +1,50 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alora_invocation_tokens": null,
3
+ "alpha_pattern": {},
4
+ "arrow_config": null,
5
+ "auto_mapping": {
6
+ "base_model_class": "Gemma4ForConditionalGeneration",
7
+ "parent_library": "transformers.models.gemma4.modeling_gemma4",
8
+ "unsloth_fixed": true
9
+ },
10
+ "base_model_name_or_path": "unsloth/gemma-4-e2b-it-unsloth-bnb-4bit",
11
+ "bias": "none",
12
+ "corda_config": null,
13
+ "ensure_weight_tying": false,
14
+ "eva_config": null,
15
+ "exclude_modules": null,
16
+ "fan_in_fan_out": false,
17
+ "inference_mode": true,
18
+ "init_lora_weights": true,
19
+ "layer_replication": null,
20
+ "layers_pattern": null,
21
+ "layers_to_transform": null,
22
+ "loftq_config": {},
23
+ "lora_alpha": 32,
24
+ "lora_bias": false,
25
+ "lora_dropout": 0.0,
26
+ "megatron_config": null,
27
+ "megatron_core": "megatron.core",
28
+ "modules_to_save": null,
29
+ "peft_type": "LORA",
30
+ "peft_version": "0.18.1",
31
+ "qalora_group_size": 16,
32
+ "r": 16,
33
+ "rank_pattern": {},
34
+ "revision": null,
35
+ "target_modules": [
36
+ "q_proj",
37
+ "gate_proj",
38
+ "v_proj",
39
+ "k_proj",
40
+ "down_proj",
41
+ "o_proj",
42
+ "up_proj"
43
+ ],
44
+ "target_parameters": null,
45
+ "task_type": "CAUSAL_LM",
46
+ "trainable_token_indices": null,
47
+ "use_dora": false,
48
+ "use_qalora": false,
49
+ "use_rslora": false
50
+ }
checkpoint-190/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:dae7c101be06c818341a1b3354a5715c93ad40fc4c4700a81f0c71c0e699b82b
3
+ size 124277728
checkpoint-190/chat_template.jinja ADDED
@@ -0,0 +1,351 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {%- macro format_parameters(properties, required, filter_keys=false) -%}
2
+ {%- set standard_keys = ['description', 'type', 'properties', 'required', 'nullable'] -%}
3
+ {%- set ns = namespace(found_first=false) -%}
4
+ {%- for key, value in properties | dictsort -%}
5
+ {%- set add_comma = false -%}
6
+ {%- if not filter_keys or key not in standard_keys -%}
7
+ {%- if ns.found_first %},{% endif -%}
8
+ {%- set ns.found_first = true -%}
9
+ {{ key }}:{
10
+ {%- if value['description'] -%}
11
+ description:<|"|>{{ value['description'] }}<|"|>
12
+ {%- set add_comma = true -%}
13
+ {%- endif -%}
14
+ {%- if value['type'] | upper == 'STRING' -%}
15
+ {%- if value['enum'] -%}
16
+ {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
17
+ enum:{{ format_argument(value['enum']) }}
18
+ {%- endif -%}
19
+ {%- elif value['type'] | upper == 'ARRAY' -%}
20
+ {%- if value['items'] is mapping and value['items'] -%}
21
+ {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
22
+ items:{
23
+ {%- set ns_items = namespace(found_first=false) -%}
24
+ {%- for item_key, item_value in value['items'] | dictsort -%}
25
+ {%- if item_value is not none -%}
26
+ {%- if ns_items.found_first %},{% endif -%}
27
+ {%- set ns_items.found_first = true -%}
28
+ {%- if item_key == 'properties' -%}
29
+ properties:{
30
+ {%- if item_value is mapping -%}
31
+ {{- format_parameters(item_value, value['items']['required'] | default([])) -}}
32
+ {%- endif -%}
33
+ }
34
+ {%- elif item_key == 'required' -%}
35
+ required:[
36
+ {%- for req_item in item_value -%}
37
+ <|"|>{{- req_item -}}<|"|>
38
+ {%- if not loop.last %},{% endif -%}
39
+ {%- endfor -%}
40
+ ]
41
+ {%- elif item_key == 'type' -%}
42
+ {%- if item_value is string -%}
43
+ type:{{ format_argument(item_value | upper) }}
44
+ {%- else -%}
45
+ type:{{ format_argument(item_value | map('upper') | list) }}
46
+ {%- endif -%}
47
+ {%- else -%}
48
+ {{ item_key }}:{{ format_argument(item_value) }}
49
+ {%- endif -%}
50
+ {%- endif -%}
51
+ {%- endfor -%}
52
+ }
53
+ {%- endif -%}
54
+ {%- endif -%}
55
+ {%- if value['nullable'] %}
56
+ {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
57
+ nullable:true
58
+ {%- endif -%}
59
+ {%- if value['type'] | upper == 'OBJECT' -%}
60
+ {%- if value['properties'] is defined and value['properties'] is mapping -%}
61
+ {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
62
+ properties:{
63
+ {{- format_parameters(value['properties'], value['required'] | default([])) -}}
64
+ }
65
+ {%- elif value is mapping -%}
66
+ {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
67
+ properties:{
68
+ {{- format_parameters(value, value['required'] | default([]), filter_keys=true) -}}
69
+ }
70
+ {%- endif -%}
71
+ {%- if value['required'] -%}
72
+ {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
73
+ required:[
74
+ {%- for item in value['required'] | default([]) -%}
75
+ <|"|>{{- item -}}<|"|>
76
+ {%- if not loop.last %},{% endif -%}
77
+ {%- endfor -%}
78
+ ]
79
+ {%- endif -%}
80
+ {%- endif -%}
81
+ {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
82
+ type:<|"|>{{ value['type'] | upper }}<|"|>}
83
+ {%- endif -%}
84
+ {%- endfor -%}
85
+ {%- endmacro -%}
86
+ {%- macro format_function_declaration(tool_data) -%}
87
+ declaration:{{- tool_data['function']['name'] -}}{description:<|"|>{{- tool_data['function']['description'] -}}<|"|>
88
+ {%- set params = tool_data['function']['parameters'] -%}
89
+ {%- if params -%}
90
+ ,parameters:{
91
+ {%- if params['properties'] -%}
92
+ properties:{ {{- format_parameters(params['properties'], params['required']) -}} },
93
+ {%- endif -%}
94
+ {%- if params['required'] -%}
95
+ required:[
96
+ {%- for item in params['required'] -%}
97
+ <|"|>{{- item -}}<|"|>
98
+ {{- ',' if not loop.last -}}
99
+ {%- endfor -%}
100
+ ],
101
+ {%- endif -%}
102
+ {%- if params['type'] -%}
103
+ type:<|"|>{{- params['type'] | upper -}}<|"|>}
104
+ {%- endif -%}
105
+ {%- endif -%}
106
+ {%- if 'response' in tool_data['function'] -%}
107
+ {%- set response_declaration = tool_data['function']['response'] -%}
108
+ ,response:{
109
+ {%- if response_declaration['description'] -%}
110
+ description:<|"|>{{- response_declaration['description'] -}}<|"|>,
111
+ {%- endif -%}
112
+ {%- if response_declaration['type'] | upper == 'OBJECT' -%}
113
+ type:<|"|>{{- response_declaration['type'] | upper -}}<|"|>}
114
+ {%- endif -%}
115
+ {%- endif -%}
116
+ }
117
+ {%- endmacro -%}
118
+ {%- macro format_argument(argument, escape_keys=True) -%}
119
+ {%- if argument is string -%}
120
+ {{- '<|"|>' + argument + '<|"|>' -}}
121
+ {%- elif argument is boolean -%}
122
+ {{- 'true' if argument else 'false' -}}
123
+ {%- elif argument is mapping -%}
124
+ {{- '{' -}}
125
+ {%- set ns = namespace(found_first=false) -%}
126
+ {%- for key, value in argument | dictsort -%}
127
+ {%- if ns.found_first %},{% endif -%}
128
+ {%- set ns.found_first = true -%}
129
+ {%- if escape_keys -%}
130
+ {{- '<|"|>' + key + '<|"|>' -}}
131
+ {%- else -%}
132
+ {{- key -}}
133
+ {%- endif -%}
134
+ :{{- format_argument(value, escape_keys=escape_keys) -}}
135
+ {%- endfor -%}
136
+ {{- '}' -}}
137
+ {%- elif argument is sequence -%}
138
+ {{- '[' -}}
139
+ {%- for item in argument -%}
140
+ {{- format_argument(item, escape_keys=escape_keys) -}}
141
+ {%- if not loop.last %},{% endif -%}
142
+ {%- endfor -%}
143
+ {{- ']' -}}
144
+ {%- else -%}
145
+ {{- argument -}}
146
+ {%- endif -%}
147
+ {%- endmacro -%}
148
+ {%- macro strip_thinking(text) -%}
149
+ {%- set ns = namespace(result='') -%}
150
+ {%- for part in text.split('<channel|>') -%}
151
+ {%- if '<|channel>' in part -%}
152
+ {%- set ns.result = ns.result + part.split('<|channel>')[0] -%}
153
+ {%- else -%}
154
+ {%- set ns.result = ns.result + part -%}
155
+ {%- endif -%}
156
+ {%- endfor -%}
157
+ {{- ns.result | trim -}}
158
+ {%- endmacro -%}
159
+
160
+ {%- macro format_tool_response_block(tool_name, response) -%}
161
+ {{- '<|tool_response>' -}}
162
+ {%- if response is mapping -%}
163
+ {{- 'response:' + tool_name + '{' -}}
164
+ {%- for key, value in response | dictsort -%}
165
+ {{- key -}}:{{- format_argument(value, escape_keys=False) -}}
166
+ {%- if not loop.last %},{% endif -%}
167
+ {%- endfor -%}
168
+ {{- '}' -}}
169
+ {%- else -%}
170
+ {{- 'response:' + tool_name + '{value:' + format_argument(response, escape_keys=False) + '}' -}}
171
+ {%- endif -%}
172
+ {{- '<tool_response|>' -}}
173
+ {%- endmacro -%}
174
+
175
+ {%- set ns = namespace(prev_message_type=None) -%}
176
+ {%- set loop_messages = messages -%}
177
+ {{- bos_token -}}
178
+ {#- Handle System/Tool Definitions Block -#}
179
+ {%- if (enable_thinking is defined and enable_thinking) or tools or messages[0]['role'] in ['system', 'developer'] -%}
180
+ {{- '<|turn>system\n' -}}
181
+ {#- Inject Thinking token at the very top of the FIRST system turn -#}
182
+ {%- if enable_thinking is defined and enable_thinking -%}
183
+ {{- '<|think|>\n' -}}
184
+ {%- set ns.prev_message_type = 'think' -%}
185
+ {%- endif -%}
186
+ {%- if messages[0]['role'] in ['system', 'developer'] -%}
187
+ {%- if messages[0]['content'] is string -%}
188
+ {{- messages[0]['content'] | trim -}}
189
+ {%- elif messages[0]['content'] is sequence -%}
190
+ {%- for item in messages[0]['content'] -%}
191
+ {{- item['text'] | trim + ' '-}}
192
+ {%- endfor -%}
193
+ {%- endif -%}
194
+ {%- set loop_messages = messages[1:] -%}
195
+ {%- endif -%}
196
+ {%- if tools -%}
197
+ {%- for tool in tools %}
198
+ {{- '<|tool>' -}}
199
+ {{- format_function_declaration(tool) | trim -}}
200
+ {{- '<tool|>' -}}
201
+ {%- endfor %}
202
+ {%- set ns.prev_message_type = 'tool' -%}
203
+ {%- endif -%}
204
+ {{- '<turn|>\n' -}}
205
+ {%- endif %}
206
+
207
+ {#- Pre-scan: find last user message index for reasoning guard -#}
208
+ {%- set ns_turn = namespace(last_user_idx=-1) -%}
209
+ {%- for i in range(loop_messages | length) -%}
210
+ {%- if loop_messages[i]['role'] == 'user' -%}
211
+ {%- set ns_turn.last_user_idx = i -%}
212
+ {%- endif -%}
213
+ {%- endfor -%}
214
+
215
+ {#- Loop through messages -#}
216
+ {%- for message in loop_messages -%}
217
+ {%- if message['role'] != 'tool' -%}
218
+ {%- set ns.prev_message_type = None -%}
219
+ {%- set role = 'model' if message['role'] == 'assistant' else message['role'] -%}
220
+ {#- Detect continuation: suppress duplicate <|turn>model when previous non-tool message was also assistant -#}
221
+ {%- set prev_nt = namespace(role=None, found=false) -%}
222
+ {%- if loop.index0 > 0 -%}
223
+ {%- for j in range(loop.index0 - 1, -1, -1) -%}
224
+ {%- if not prev_nt.found -%}
225
+ {%- if loop_messages[j]['role'] != 'tool' -%}
226
+ {%- set prev_nt.role = loop_messages[j]['role'] -%}
227
+ {%- set prev_nt.found = true -%}
228
+ {%- endif -%}
229
+ {%- endif -%}
230
+ {%- endfor -%}
231
+ {%- endif -%}
232
+ {%- set continue_same_model_turn = (role == 'model' and prev_nt.role == 'assistant') -%}
233
+ {%- if not continue_same_model_turn -%}
234
+ {{- '<|turn>' + role + '\n' }}
235
+ {%- endif -%}
236
+
237
+ {#- Render reasoning/reasoning_content as thinking channel -#}
238
+ {%- set thinking_text = message.get('reasoning') or message.get('reasoning_content') -%}
239
+ {%- if thinking_text and loop.index0 > ns_turn.last_user_idx and message.get('tool_calls') -%}
240
+ {{- '<|channel>thought\n' + thinking_text + '\n<channel|>' -}}
241
+ {%- endif -%}
242
+
243
+ {%- if message['tool_calls'] -%}
244
+ {%- for tool_call in message['tool_calls'] -%}
245
+ {%- set function = tool_call['function'] -%}
246
+ {{- '<|tool_call>call:' + function['name'] + '{' -}}
247
+ {%- if function['arguments'] is mapping -%}
248
+ {%- set ns_args = namespace(found_first=false) -%}
249
+ {%- for key, value in function['arguments'] | dictsort -%}
250
+ {%- if ns_args.found_first %},{% endif -%}
251
+ {%- set ns_args.found_first = true -%}
252
+ {{- key -}}:{{- format_argument(value, escape_keys=False) -}}
253
+ {%- endfor -%}
254
+ {%- elif function['arguments'] is string -%}
255
+ {{- function['arguments'] -}}
256
+ {%- endif -%}
257
+ {{- '}<tool_call|>' -}}
258
+ {%- endfor -%}
259
+ {%- set ns.prev_message_type = 'tool_call' -%}
260
+ {%- endif -%}
261
+
262
+ {%- set ns_tr_out = namespace(flag=false) -%}
263
+ {%- if message.get('tool_responses') -%}
264
+ {#- Legacy: tool_responses embedded on the assistant message (Google/Gemma native) -#}
265
+ {%- for tool_response in message['tool_responses'] -%}
266
+ {{- format_tool_response_block(tool_response['name'] | default('unknown'), tool_response['response']) -}}
267
+ {%- set ns_tr_out.flag = true -%}
268
+ {%- set ns.prev_message_type = 'tool_response' -%}
269
+ {%- endfor -%}
270
+ {%- elif message.get('tool_calls') -%}
271
+ {#- OpenAI Chat Completions: forward-scan consecutive role:tool messages -#}
272
+ {%- set ns_tool_scan = namespace(stopped=false) -%}
273
+ {%- for k in range(loop.index0 + 1, loop_messages | length) -%}
274
+ {%- if ns_tool_scan.stopped -%}
275
+ {%- elif loop_messages[k]['role'] != 'tool' -%}
276
+ {%- set ns_tool_scan.stopped = true -%}
277
+ {%- else -%}
278
+ {%- set follow = loop_messages[k] -%}
279
+ {#- Resolve tool_call_id to function name -#}
280
+ {%- set ns_tname = namespace(name=follow.get('name') | default('unknown')) -%}
281
+ {%- for tc in message['tool_calls'] -%}
282
+ {%- if tc.get('id') == follow.get('tool_call_id') -%}
283
+ {%- set ns_tname.name = tc['function']['name'] -%}
284
+ {%- endif -%}
285
+ {%- endfor -%}
286
+ {#- Handle content as string or content-parts array -#}
287
+ {%- set tool_body = follow.get('content') -%}
288
+ {%- if tool_body is string -%}
289
+ {{- format_tool_response_block(ns_tname.name, tool_body) -}}
290
+ {%- elif tool_body is sequence and tool_body is not string -%}
291
+ {%- set ns_txt = namespace(s='') -%}
292
+ {%- for part in tool_body -%}
293
+ {%- if part.get('type') == 'text' -%}
294
+ {%- set ns_txt.s = ns_txt.s + (part.get('text') | default('')) -%}
295
+ {%- endif -%}
296
+ {%- endfor -%}
297
+ {{- format_tool_response_block(ns_tname.name, ns_txt.s) -}}
298
+ {%- else -%}
299
+ {{- format_tool_response_block(ns_tname.name, tool_body) -}}
300
+ {%- endif -%}
301
+ {%- set ns_tr_out.flag = true -%}
302
+ {%- set ns.prev_message_type = 'tool_response' -%}
303
+ {%- endif -%}
304
+ {%- endfor -%}
305
+ {%- endif -%}
306
+
307
+ {%- set captured_content -%}
308
+ {%- if message['content'] is string -%}
309
+ {%- if role == 'model' -%}
310
+ {{- strip_thinking(message['content']) -}}
311
+ {%- else -%}
312
+ {{- message['content'] | trim -}}
313
+ {%- endif -%}
314
+ {%- elif message['content'] is sequence -%}
315
+ {%- for item in message['content'] -%}
316
+ {%- if item['type'] == 'text' -%}
317
+ {%- if role == 'model' -%}
318
+ {{- strip_thinking(item['text']) -}}
319
+ {%- else -%}
320
+ {{- item['text'] | trim -}}
321
+ {%- endif -%}
322
+ {%- elif item['type'] == 'image' -%}
323
+ {{- '<|image|>' -}}
324
+ {%- set ns.prev_message_type = 'image' -%}
325
+ {%- elif item['type'] == 'audio' -%}
326
+ {{- '<|audio|>' -}}
327
+ {%- set ns.prev_message_type = 'audio' -%}
328
+ {%- elif item['type'] == 'video' -%}
329
+ {{- '<|video|>' -}}
330
+ {%- set ns.prev_message_type = 'video' -%}
331
+ {%- endif -%}
332
+ {%- endfor -%}
333
+ {%- endif -%}
334
+ {%- endset -%}
335
+
336
+ {{- captured_content -}}
337
+ {%- set has_content = captured_content | trim | length > 0 -%}
338
+
339
+ {%- if ns.prev_message_type == 'tool_call' and not ns_tr_out.flag -%}
340
+ {{- '<|tool_response>' -}}
341
+ {%- elif not (ns_tr_out.flag and not has_content) -%}
342
+ {{- '<turn|>\n' -}}
343
+ {%- endif -%}
344
+ {%- endif -%}
345
+ {%- endfor -%}
346
+
347
+ {%- if add_generation_prompt -%}
348
+ {%- if ns.prev_message_type != 'tool_response' and ns.prev_message_type != 'tool_call' -%}
349
+ {{- '<|turn>model\n' -}}
350
+ {%- endif -%}
351
+ {%- endif -%}
checkpoint-190/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:b26ffa09ce09033bd748c71a5bd466483b607675e03fe146c587da6439440b9d
3
+ size 52047117
checkpoint-190/processor_config.json ADDED
@@ -0,0 +1,75 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "audio_ms_per_token": 40,
3
+ "audio_seq_length": 750,
4
+ "feature_extractor": {
5
+ "dither": 0.0,
6
+ "feature_extractor_type": "Gemma4AudioFeatureExtractor",
7
+ "feature_size": 128,
8
+ "fft_length": 512,
9
+ "fft_overdrive": false,
10
+ "frame_length": 320,
11
+ "hop_length": 160,
12
+ "input_scale_factor": 1.0,
13
+ "max_frequency": 8000.0,
14
+ "mel_floor": 0.001,
15
+ "min_frequency": 0.0,
16
+ "padding_side": "left",
17
+ "padding_value": 0.0,
18
+ "per_bin_mean": null,
19
+ "per_bin_stddev": null,
20
+ "preemphasis": 0.0,
21
+ "preemphasis_htk_flavor": true,
22
+ "return_attention_mask": true,
23
+ "sampling_rate": 16000
24
+ },
25
+ "image_processor": {
26
+ "do_convert_rgb": true,
27
+ "do_normalize": false,
28
+ "do_rescale": true,
29
+ "do_resize": true,
30
+ "image_mean": [
31
+ 0.0,
32
+ 0.0,
33
+ 0.0
34
+ ],
35
+ "image_processor_type": "Gemma4ImageProcessor",
36
+ "image_seq_length": 280,
37
+ "image_std": [
38
+ 1.0,
39
+ 1.0,
40
+ 1.0
41
+ ],
42
+ "max_soft_tokens": 280,
43
+ "patch_size": 16,
44
+ "pooling_kernel_size": 3,
45
+ "resample": 3,
46
+ "rescale_factor": 0.00392156862745098
47
+ },
48
+ "image_seq_length": 280,
49
+ "processor_class": "Gemma4Processor",
50
+ "video_processor": {
51
+ "do_convert_rgb": true,
52
+ "do_normalize": true,
53
+ "do_rescale": true,
54
+ "do_resize": true,
55
+ "do_sample_frames": true,
56
+ "image_mean": [
57
+ 0.0,
58
+ 0.0,
59
+ 0.0
60
+ ],
61
+ "image_std": [
62
+ 1.0,
63
+ 1.0,
64
+ 1.0
65
+ ],
66
+ "max_soft_tokens": 70,
67
+ "num_frames": 32,
68
+ "patch_size": 16,
69
+ "pooling_kernel_size": 3,
70
+ "resample": 3,
71
+ "rescale_factor": 0.00392156862745098,
72
+ "return_metadata": false,
73
+ "video_processor_type": "Gemma4VideoProcessor"
74
+ }
75
+ }
checkpoint-190/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f1d565802a8e26c4e8a31328752b7a7fdc186d9401aa008e65697d0ad8c22e33
3
+ size 14645
checkpoint-190/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e582bf3c00963bd3fcaa4b162e161d1ea274c715ce05768df1a86c80c59a424e
3
+ size 1383