pnkjkhani commited on
Commit
4c95f2f
·
verified ·
1 Parent(s): 3543a08

Upload folder using huggingface_hub

Browse files
.gitattributes CHANGED
@@ -33,3 +33,6 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
 
 
 
 
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
  *tfevents* filter=lfs diff=lfs merge=lfs -text
36
+ checkpoint-150/tokenizer.json filter=lfs diff=lfs merge=lfs -text
37
+ checkpoint-300/tokenizer.json filter=lfs diff=lfs merge=lfs -text
38
+ tokenizer.json filter=lfs diff=lfs merge=lfs -text
README.md ADDED
@@ -0,0 +1,116 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ library_name: peft
3
+ base_model: tiny-random/gpt-oss
4
+ tags:
5
+ - axolotl
6
+ - base_model:adapter:tiny-random/gpt-oss
7
+ - lora
8
+ - transformers
9
+ datasets:
10
+ - HuggingFaceH4/no_robots
11
+ pipeline_tag: text-generation
12
+ model-index:
13
+ - name: tmp/fine-tuning-results
14
+ results: []
15
+ ---
16
+
17
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
18
+ should probably proofread and complete it, then remove this comment. -->
19
+
20
+ [<img src="https://raw.githubusercontent.com/axolotl-ai-cloud/axolotl/main/image/axolotl-badge-web.png" alt="Built with Axolotl" width="200" height="32"/>](https://github.com/axolotl-ai-cloud/axolotl)
21
+ <details><summary>See axolotl config</summary>
22
+
23
+ axolotl version: `0.12.2`
24
+ ```yaml
25
+ base_model: tiny-random/gpt-oss
26
+ datasets:
27
+ - path: HuggingFaceH4/no_robots
28
+ type: chat_template
29
+ split: train
30
+ dataset_prepared_path: /tmp/dataset_prepared
31
+ val_set_size: 0.1
32
+ output_dir: /tmp/fine-tuning-results
33
+ sequence_len: 256
34
+ sample_packing: false
35
+ gradient_accumulation_steps: 8
36
+ micro_batch_size: 4
37
+ num_epochs: 1
38
+ max_steps: 300
39
+ optimizer: adamw_torch_8bit
40
+ lr_scheduler: constant_with_warmup
41
+ learning_rate: 0.0001
42
+ fp16: true
43
+ gradient_checkpointing: false
44
+ activation_offloading: false
45
+ adapter: lora
46
+ lora_r: 16
47
+ lora_alpha: 32
48
+ lora_dropout: 0.1
49
+ lora_target_modules:
50
+ - q_proj
51
+ - v_proj
52
+ - k_proj
53
+ - o_proj
54
+ - gate_proj
55
+ - up_proj
56
+ - down_proj
57
+ logging_steps: 10
58
+ save_steps: 150
59
+ warmup_ratio: 0.1
60
+ save_total_limit: 2
61
+ use_wandb: true
62
+ wandb_project: project-testing-100
63
+ wandb_watch: all
64
+ wandb_log_model: end
65
+ wandb_run_name: run-100
66
+ report_to: wandb
67
+
68
+ ```
69
+
70
+ </details><br>
71
+
72
+ [<img src="https://raw.githubusercontent.com/wandb/assets/main/wandb-github-badge-28.svg" alt="Visualize in Weights & Biases" width="200" height="32"/>](https://wandb.ai/manish_prakash-permitportalapp/project-testing-100/runs/18rba5vt)
73
+ # tmp/fine-tuning-results
74
+
75
+ This model is a fine-tuned version of [tiny-random/gpt-oss](https://huggingface.co/tiny-random/gpt-oss) on the HuggingFaceH4/no_robots dataset.
76
+
77
+ ## Model description
78
+
79
+ More information needed
80
+
81
+ ## Intended uses & limitations
82
+
83
+ More information needed
84
+
85
+ ## Training and evaluation data
86
+
87
+ More information needed
88
+
89
+ ## Training procedure
90
+
91
+ ### Training hyperparameters
92
+
93
+ The following hyperparameters were used during training:
94
+ - learning_rate: 0.0001
95
+ - train_batch_size: 4
96
+ - eval_batch_size: 4
97
+ - seed: 42
98
+ - gradient_accumulation_steps: 8
99
+ - total_train_batch_size: 32
100
+ - optimizer: Use adamw_torch_8bit with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
101
+ - lr_scheduler_type: constant_with_warmup
102
+ - lr_scheduler_warmup_steps: 11
103
+ - training_steps: 300
104
+ - mixed_precision_training: Native AMP
105
+
106
+ ### Training results
107
+
108
+
109
+
110
+ ### Framework versions
111
+
112
+ - PEFT 0.17.0
113
+ - Transformers 4.55.2
114
+ - Pytorch 2.6.0+cu124
115
+ - Datasets 4.0.0
116
+ - Tokenizers 0.21.4
adapter_config.json ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alpha_pattern": {},
3
+ "auto_mapping": null,
4
+ "base_model_name_or_path": "tiny-random/gpt-oss",
5
+ "bias": "none",
6
+ "corda_config": null,
7
+ "eva_config": null,
8
+ "exclude_modules": null,
9
+ "fan_in_fan_out": null,
10
+ "inference_mode": true,
11
+ "init_lora_weights": true,
12
+ "layer_replication": null,
13
+ "layers_pattern": null,
14
+ "layers_to_transform": null,
15
+ "loftq_config": {},
16
+ "lora_alpha": 32,
17
+ "lora_bias": false,
18
+ "lora_dropout": 0.1,
19
+ "megatron_config": null,
20
+ "megatron_core": "megatron.core",
21
+ "modules_to_save": null,
22
+ "peft_type": "LORA",
23
+ "qalora_group_size": 16,
24
+ "r": 16,
25
+ "rank_pattern": {},
26
+ "revision": null,
27
+ "target_modules": [
28
+ "v_proj",
29
+ "k_proj",
30
+ "o_proj",
31
+ "gate_proj",
32
+ "q_proj",
33
+ "up_proj",
34
+ "down_proj"
35
+ ],
36
+ "target_parameters": [],
37
+ "task_type": "CAUSAL_LM",
38
+ "trainable_token_indices": null,
39
+ "use_dora": false,
40
+ "use_qalora": false,
41
+ "use_rslora": false
42
+ }
adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6fe39862f113460291406bcad1362d191b55143cfb6b365ca9dcdc304ea5a47e
3
+ size 43008
chat_template.jinja ADDED
@@ -0,0 +1,397 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {#-
2
+ In addition to the normal inputs of `messages` and `tools`, this template also accepts the
3
+ following kwargs:
4
+ - "builtin_tools": A list, can contain "browser" and/or "python".
5
+ - "model_identity": A string that optionally describes the model identity.
6
+ - "reasoning_effort": A string that describes the reasoning effort, defaults to "medium".
7
+ #}
8
+
9
+ {#- Tool Definition Rendering ============================================== #}
10
+ {%- macro render_typescript_type(param_spec, required_params, is_nullable=false) -%}
11
+ {%- if param_spec.type == "array" -%}
12
+ {%- if param_spec['items'] -%}
13
+ {%- if param_spec['items']['type'] == "string" -%}
14
+ {{- "string[]" }}
15
+ {%- elif param_spec['items']['type'] == "number" -%}
16
+ {{- "number[]" }}
17
+ {%- elif param_spec['items']['type'] == "integer" -%}
18
+ {{- "number[]" }}
19
+ {%- elif param_spec['items']['type'] == "boolean" -%}
20
+ {{- "boolean[]" }}
21
+ {%- else -%}
22
+ {%- set inner_type = render_typescript_type(param_spec['items'], required_params) -%}
23
+ {%- if inner_type == "object | object" or inner_type|length > 50 -%}
24
+ {{- "any[]" }}
25
+ {%- else -%}
26
+ {{- inner_type + "[]" }}
27
+ {%- endif -%}
28
+ {%- endif -%}
29
+ {%- if param_spec.nullable -%}
30
+ {{- " | null" }}
31
+ {%- endif -%}
32
+ {%- else -%}
33
+ {{- "any[]" }}
34
+ {%- if param_spec.nullable -%}
35
+ {{- " | null" }}
36
+ {%- endif -%}
37
+ {%- endif -%}
38
+ {%- elif param_spec.type is defined and param_spec.type is iterable and param_spec.type is not string and param_spec.type is not mapping and param_spec.type[0] is defined -%}
39
+ {#- Handle array of types like ["object", "object"] from Union[dict, list] #}
40
+ {%- if param_spec.type | length > 1 -%}
41
+ {{- param_spec.type | join(" | ") }}
42
+ {%- else -%}
43
+ {{- param_spec.type[0] }}
44
+ {%- endif -%}
45
+ {%- elif param_spec.oneOf -%}
46
+ {#- Handle oneOf schemas - check for complex unions and fallback to any #}
47
+ {%- set has_object_variants = false -%}
48
+ {%- for variant in param_spec.oneOf -%}
49
+ {%- if variant.type == "object" -%}
50
+ {%- set has_object_variants = true -%}
51
+ {%- endif -%}
52
+ {%- endfor -%}
53
+ {%- if has_object_variants and param_spec.oneOf|length > 1 -%}
54
+ {{- "any" }}
55
+ {%- else -%}
56
+ {%- for variant in param_spec.oneOf -%}
57
+ {{- render_typescript_type(variant, required_params) -}}
58
+ {%- if variant.description %}
59
+ {{- "// " + variant.description }}
60
+ {%- endif -%}
61
+ {%- if variant.default is defined %}
62
+ {{ "// default: " + variant.default|tojson }}
63
+ {%- endif -%}
64
+ {%- if not loop.last %}
65
+ {{- " | " }}
66
+ {% endif -%}
67
+ {%- endfor -%}
68
+ {%- endif -%}
69
+ {%- elif param_spec.type == "string" -%}
70
+ {%- if param_spec.enum -%}
71
+ {{- '"' + param_spec.enum|join('" | "') + '"' -}}
72
+ {%- else -%}
73
+ {{- "string" }}
74
+ {%- if param_spec.nullable %}
75
+ {{- " | null" }}
76
+ {%- endif -%}
77
+ {%- endif -%}
78
+ {%- elif param_spec.type == "number" -%}
79
+ {{- "number" }}
80
+ {%- elif param_spec.type == "integer" -%}
81
+ {{- "number" }}
82
+ {%- elif param_spec.type == "boolean" -%}
83
+ {{- "boolean" }}
84
+
85
+ {%- elif param_spec.type == "object" -%}
86
+ {%- if param_spec.properties -%}
87
+ {{- "{
88
+ " }}
89
+ {%- for prop_name, prop_spec in param_spec.properties.items() -%}
90
+ {{- prop_name -}}
91
+ {%- if prop_name not in (param_spec.required or []) -%}
92
+ {{- "?" }}
93
+ {%- endif -%}
94
+ {{- ": " }}
95
+ {{ render_typescript_type(prop_spec, param_spec.required or []) }}
96
+ {%- if not loop.last -%}
97
+ {{-", " }}
98
+ {%- endif -%}
99
+ {%- endfor -%}
100
+ {{- "}" }}
101
+ {%- else -%}
102
+ {{- "object" }}
103
+ {%- endif -%}
104
+ {%- else -%}
105
+ {{- "any" }}
106
+ {%- endif -%}
107
+ {%- endmacro -%}
108
+
109
+ {%- macro render_tool_namespace(namespace_name, tools) -%}
110
+ {{- "## " + namespace_name + "
111
+
112
+ " }}
113
+ {{- "namespace " + namespace_name + " {
114
+
115
+ " }}
116
+ {%- for tool in tools %}
117
+ {%- set tool = tool.function %}
118
+ {{- "// " + tool.description + "
119
+ " }}
120
+ {{- "type "+ tool.name + " = " }}
121
+ {%- if tool.parameters and tool.parameters.properties %}
122
+ {{- "(_: {
123
+ " }}
124
+ {%- for param_name, param_spec in tool.parameters.properties.items() %}
125
+ {%- if param_spec.description %}
126
+ {{- "// " + param_spec.description + "
127
+ " }}
128
+ {%- endif %}
129
+ {{- param_name }}
130
+ {%- if param_name not in (tool.parameters.required or []) -%}
131
+ {{- "?" }}
132
+ {%- endif -%}
133
+ {{- ": " }}
134
+ {{- render_typescript_type(param_spec, tool.parameters.required or []) }}
135
+ {%- if param_spec.default is defined -%}
136
+ {%- if param_spec.enum %}
137
+ {{- ", // default: " + param_spec.default }}
138
+ {%- elif param_spec.oneOf %}
139
+ {{- "// default: " + param_spec.default }}
140
+ {%- else %}
141
+ {{- ", // default: " + param_spec.default|tojson }}
142
+ {%- endif -%}
143
+ {%- endif -%}
144
+ {%- if not loop.last %}
145
+ {{- ",
146
+ " }}
147
+ {%- else %}
148
+ {{- "
149
+ " }}
150
+ {%- endif -%}
151
+ {%- endfor %}
152
+ {{- "}) => any;
153
+
154
+ " }}
155
+ {%- else -%}
156
+ {{- "() => any;
157
+
158
+ " }}
159
+ {%- endif -%}
160
+ {%- endfor %}
161
+ {{- "} // namespace " + namespace_name }}
162
+ {%- endmacro -%}
163
+
164
+ {%- macro render_builtin_tools(browser_tool, python_tool) -%}
165
+ {%- if browser_tool %}
166
+ {{- "## browser
167
+
168
+ " }}
169
+ {{- "// Tool for browsing.
170
+ " }}
171
+ {{- "// The `cursor` appears in brackets before each browsing display: `[{cursor}]`.
172
+ " }}
173
+ {{- "// Cite information from the tool using the following format:
174
+ " }}
175
+ {{- "// `【{cursor}†L{line_start}(-L{line_end})?】`, for example: `【6†L9-L11】` or `【8†L3】`.
176
+ " }}
177
+ {{- "// Do not quote more than 10 words directly from the tool output.
178
+ " }}
179
+ {{- "// sources=web (default: web)
180
+ " }}
181
+ {{- "namespace browser {
182
+
183
+ " }}
184
+ {{- "// Searches for information related to `query` and displays `topn` results.
185
+ " }}
186
+ {{- "type search = (_: {
187
+ " }}
188
+ {{- "query: string,
189
+ " }}
190
+ {{- "topn?: number, // default: 10
191
+ " }}
192
+ {{- "source?: string,
193
+ " }}
194
+ {{- "}) => any;
195
+
196
+ " }}
197
+ {{- "// Opens the link `id` from the page indicated by `cursor` starting at line number `loc`, showing `num_lines` lines.
198
+ " }}
199
+ {{- "// Valid link ids are displayed with the formatting: `【{id}†.*】`.
200
+ " }}
201
+ {{- "// If `cursor` is not provided, the most recent page is implied.
202
+ " }}
203
+ {{- "// If `id` is a string, it is treated as a fully qualified URL associated with `source`.
204
+ " }}
205
+ {{- "// If `loc` is not provided, the viewport will be positioned at the beginning of the document or centered on the most relevant passage, if available.
206
+ " }}
207
+ {{- "// Use this function without `id` to scroll to a new location of an opened page.
208
+ " }}
209
+ {{- "type open = (_: {
210
+ " }}
211
+ {{- "id?: number | string, // default: -1
212
+ " }}
213
+ {{- "cursor?: number, // default: -1
214
+ " }}
215
+ {{- "loc?: number, // default: -1
216
+ " }}
217
+ {{- "num_lines?: number, // default: -1
218
+ " }}
219
+ {{- "view_source?: boolean, // default: false
220
+ " }}
221
+ {{- "source?: string,
222
+ " }}
223
+ {{- "}) => any;
224
+
225
+ " }}
226
+ {{- "// Finds exact matches of `pattern` in the current page, or the page given by `cursor`.
227
+ " }}
228
+ {{- "type find = (_: {
229
+ " }}
230
+ {{- "pattern: string,
231
+ " }}
232
+ {{- "cursor?: number, // default: -1
233
+ " }}
234
+ {{- "}) => any;
235
+
236
+ " }}
237
+ {{- "} // namespace browser
238
+
239
+ " }}
240
+ {%- endif -%}
241
+
242
+ {%- if python_tool %}
243
+ {{- "## python
244
+
245
+ " }}
246
+ {{- "Use this tool to execute Python code in your chain of thought. The code will not be shown to the user. This tool should be used for internal reasoning, but not for code that is intended to be visible to the user (e.g. when creating plots, tables, or files).
247
+
248
+ " }}
249
+ {{- "When you send a message containing Python code to python, it will be executed in a stateful Jupyter notebook environment. python will respond with the output of the execution or time out after 120.0 seconds. The drive at '/mnt/data' can be used to save and persist user files. Internet access for this session is UNKNOWN. Depends on the cluster.
250
+
251
+ " }}
252
+ {%- endif -%}
253
+ {%- endmacro -%}
254
+
255
+ {#- System Message Construction ============================================ #}
256
+ {%- macro build_system_message() -%}
257
+ {%- if model_identity is not defined %}
258
+ {%- set model_identity = "You are ChatGPT, a large language model trained by OpenAI." %}
259
+ {%- endif %}
260
+ {{- model_identity + "
261
+ " }}
262
+ {{- "Knowledge cutoff: 2024-06
263
+ " }}
264
+ {{- "Current date: " + strftime_now("%Y-%m-%d") + "
265
+
266
+ " }}
267
+ {%- if reasoning_effort is not defined %}
268
+ {%- set reasoning_effort = "medium" %}
269
+ {%- endif %}
270
+ {{- "Reasoning: " + reasoning_effort + "
271
+
272
+ " }}
273
+ {%- if builtin_tools %}
274
+ {{- "# Tools
275
+
276
+ " }}
277
+ {%- set available_builtin_tools = namespace(browser=false, python=false) %}
278
+ {%- for tool in builtin_tools %}
279
+ {%- if tool == "browser" %}
280
+ {%- set available_builtin_tools.browser = true %}
281
+ {%- elif tool == "python" %}
282
+ {%- set available_builtin_tools.python = true %}
283
+ {%- endif %}
284
+ {%- endfor %}
285
+ {{- render_builtin_tools(available_builtin_tools.browser, available_builtin_tools.python) }}
286
+ {%- endif -%}
287
+ {{- "# Valid channels: analysis, commentary, final. Channel must be included for every message." }}
288
+ {%- if tools -%}
289
+ {{- "
290
+ Calls to these tools must go to the commentary channel: 'functions'." }}
291
+ {%- endif -%}
292
+ {%- endmacro -%}
293
+
294
+ {#- Main Template Logic ================================================= #}
295
+ {#- Set defaults #}
296
+
297
+ {#- Render system message #}
298
+ {{- "<|start|>system<|message|>" }}
299
+ {{- build_system_message() }}
300
+ {{- "<|end|>" }}
301
+
302
+ {#- Extract developer message #}
303
+ {%- if messages[0].role == "developer" or messages[0].role == "system" %}
304
+ {%- set developer_message = messages[0].content %}
305
+ {%- set loop_messages = messages[1:] %}
306
+ {%- else %}
307
+ {%- set developer_message = "" %}
308
+ {%- set loop_messages = messages %}
309
+ {%- endif %}
310
+
311
+ {#- Render developer message #}
312
+ {%- if developer_message or tools %}
313
+ {{- "<|start|>developer<|message|>" }}
314
+ {%- if developer_message %}
315
+ {{- "# Instructions
316
+
317
+ " }}
318
+ {{- developer_message }}
319
+ {%- endif %}
320
+ {%- if tools -%}
321
+ {{- "
322
+
323
+ " }}
324
+ {{- "# Tools
325
+
326
+ " }}
327
+ {{- render_tool_namespace("functions", tools) }}
328
+ {%- endif -%}
329
+ {{- "<|end|>" }}
330
+ {%- endif %}
331
+
332
+ {#- Render messages #}
333
+ {%- set last_tool_call = namespace(name=none) %}
334
+ {%- for message in loop_messages -%}
335
+ {#- At this point only assistant/user/tool messages should remain #}
336
+ {%- if message.role == 'assistant' -%}
337
+ {#- Checks to ensure the messages are being passed in the format we expect #}
338
+ {%- if "content" in message %}
339
+ {%- if "<|channel|>analysis<|message|>" in message.content or "<|channel|>final<|message|>" in message.content %}
340
+ {{- raise_exception("You have passed a message containing <|channel|> tags in the content field. Instead of doing this, you should pass analysis messages (the string between '<|message|>' and '<|end|>') in the 'thinking' field, and final messages (the string between '<|message|>' and '<|end|>') in the 'content' field.") }}
341
+ {%- endif %}
342
+ {%- endif %}
343
+ {%- if "thinking" in message %}
344
+ {%- if "<|channel|>analysis<|message|>" in message.thinking or "<|channel|>final<|message|>" in message.thinking %}
345
+ {{- raise_exception("You have passed a message containing <|channel|> tags in the thinking field. Instead of doing this, you should pass analysis messages (the string between '<|message|>' and '<|end|>') in the 'thinking' field, and final messages (the string between '<|message|>' and '<|end|>') in the 'content' field.") }}
346
+ {%- endif %}
347
+ {%- endif %}
348
+ {%- if "tool_calls" in message %}
349
+ {#- We assume max 1 tool call per message, and so we infer the tool call name #}
350
+ {#- in "tool" messages from the most recent assistant tool call name #}
351
+ {%- set tool_call = message.tool_calls[0] %}
352
+ {%- if tool_call.function %}
353
+ {%- set tool_call = tool_call.function %}
354
+ {%- endif %}
355
+ {%- if message.content and message.thinking %}
356
+ {{- raise_exception("Cannot pass both content and thinking in an assistant message with tool calls! Put the analysis message in one or the other, but not both.") }}
357
+ {%- elif message.content %}
358
+ {{- "<|start|>assistant<|channel|>analysis<|message|>" + message.content + "<|end|>" }}
359
+ {%- elif message.thinking %}
360
+ {{- "<|start|>assistant<|channel|>analysis<|message|>" + message.thinking + "<|end|>" }}
361
+ {%- endif %}
362
+ {{- "<|start|>assistant to=" }}
363
+ {{- "functions." + tool_call.name + "<|channel|>commentary " }}
364
+ {{- (tool_call.content_type if tool_call.content_type is defined else "json") + "<|message|>" }}
365
+ {{- tool_call.arguments|tojson }}
366
+ {{- "<|call|>" }}
367
+ {%- set last_tool_call.name = tool_call.name %}
368
+ {%- elif loop.last and not add_generation_prompt %}
369
+ {#- Only render the CoT if the final turn is an assistant turn and add_generation_prompt is false #}
370
+ {#- This is a situation that should only occur in training, never in inference. #}
371
+ {%- if "thinking" in message %}
372
+ {{- "<|start|>assistant<|channel|>analysis<|message|>" + message.thinking + "<|end|>" }}
373
+ {%- endif %}
374
+ {#- <|return|> indicates the end of generation, but <|end|> does not #}
375
+ {#- <|return|> should never be an input to the model, but we include it as the final token #}
376
+ {#- when training, so the model learns to emit it. #}
377
+ {{- "<|start|>assistant<|channel|>final<|message|>" + message.content + "<|return|>" }}
378
+ {%- else %}
379
+ {#- CoT is dropped during all previous turns, so we never render it for inference #}
380
+ {{- "<|start|>assistant<|channel|>final<|message|>" + message.content + "<|end|>" }}
381
+ {%- set last_tool_call.name = none %}
382
+ {%- endif %}
383
+ {%- elif message.role == 'tool' -%}
384
+ {%- if last_tool_call.name is none %}
385
+ {{- raise_exception("Message has tool role, but there was no previous assistant message with a tool call!") }}
386
+ {%- endif %}
387
+ {{- "<|start|>functions." + last_tool_call.name }}
388
+ {{- " to=assistant<|channel|>commentary<|message|>" + message.content|tojson + "<|end|>" }}
389
+ {%- elif message.role == 'user' -%}
390
+ {{- "<|start|>user<|message|>" + message.content + "<|end|>" }}
391
+ {%- endif -%}
392
+ {%- endfor -%}
393
+
394
+ {#- Generation prompt #}
395
+ {%- if add_generation_prompt -%}
396
+ <|start|>assistant
397
+ {%- endif -%}
checkpoint-150/README.md ADDED
@@ -0,0 +1,208 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: tiny-random/gpt-oss
3
+ library_name: peft
4
+ pipeline_tag: text-generation
5
+ tags:
6
+ - axolotl
7
+ - base_model:adapter:tiny-random/gpt-oss
8
+ - lora
9
+ - transformers
10
+ ---
11
+
12
+ # Model Card for Model ID
13
+
14
+ <!-- Provide a quick summary of what the model is/does. -->
15
+
16
+
17
+
18
+ ## Model Details
19
+
20
+ ### Model Description
21
+
22
+ <!-- Provide a longer summary of what this model is. -->
23
+
24
+
25
+
26
+ - **Developed by:** [More Information Needed]
27
+ - **Funded by [optional]:** [More Information Needed]
28
+ - **Shared by [optional]:** [More Information Needed]
29
+ - **Model type:** [More Information Needed]
30
+ - **Language(s) (NLP):** [More Information Needed]
31
+ - **License:** [More Information Needed]
32
+ - **Finetuned from model [optional]:** [More Information Needed]
33
+
34
+ ### Model Sources [optional]
35
+
36
+ <!-- Provide the basic links for the model. -->
37
+
38
+ - **Repository:** [More Information Needed]
39
+ - **Paper [optional]:** [More Information Needed]
40
+ - **Demo [optional]:** [More Information Needed]
41
+
42
+ ## Uses
43
+
44
+ <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
45
+
46
+ ### Direct Use
47
+
48
+ <!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
49
+
50
+ [More Information Needed]
51
+
52
+ ### Downstream Use [optional]
53
+
54
+ <!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
55
+
56
+ [More Information Needed]
57
+
58
+ ### Out-of-Scope Use
59
+
60
+ <!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
61
+
62
+ [More Information Needed]
63
+
64
+ ## Bias, Risks, and Limitations
65
+
66
+ <!-- This section is meant to convey both technical and sociotechnical limitations. -->
67
+
68
+ [More Information Needed]
69
+
70
+ ### Recommendations
71
+
72
+ <!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
73
+
74
+ Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
75
+
76
+ ## How to Get Started with the Model
77
+
78
+ Use the code below to get started with the model.
79
+
80
+ [More Information Needed]
81
+
82
+ ## Training Details
83
+
84
+ ### Training Data
85
+
86
+ <!-- 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. -->
87
+
88
+ [More Information Needed]
89
+
90
+ ### Training Procedure
91
+
92
+ <!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
93
+
94
+ #### Preprocessing [optional]
95
+
96
+ [More Information Needed]
97
+
98
+
99
+ #### Training Hyperparameters
100
+
101
+ - **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
102
+
103
+ #### Speeds, Sizes, Times [optional]
104
+
105
+ <!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
106
+
107
+ [More Information Needed]
108
+
109
+ ## Evaluation
110
+
111
+ <!-- This section describes the evaluation protocols and provides the results. -->
112
+
113
+ ### Testing Data, Factors & Metrics
114
+
115
+ #### Testing Data
116
+
117
+ <!-- This should link to a Dataset Card if possible. -->
118
+
119
+ [More Information Needed]
120
+
121
+ #### Factors
122
+
123
+ <!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
124
+
125
+ [More Information Needed]
126
+
127
+ #### Metrics
128
+
129
+ <!-- These are the evaluation metrics being used, ideally with a description of why. -->
130
+
131
+ [More Information Needed]
132
+
133
+ ### Results
134
+
135
+ [More Information Needed]
136
+
137
+ #### Summary
138
+
139
+
140
+
141
+ ## Model Examination [optional]
142
+
143
+ <!-- Relevant interpretability work for the model goes here -->
144
+
145
+ [More Information Needed]
146
+
147
+ ## Environmental Impact
148
+
149
+ <!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
150
+
151
+ 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).
152
+
153
+ - **Hardware Type:** [More Information Needed]
154
+ - **Hours used:** [More Information Needed]
155
+ - **Cloud Provider:** [More Information Needed]
156
+ - **Compute Region:** [More Information Needed]
157
+ - **Carbon Emitted:** [More Information Needed]
158
+
159
+ ## Technical Specifications [optional]
160
+
161
+ ### Model Architecture and Objective
162
+
163
+ [More Information Needed]
164
+
165
+ ### Compute Infrastructure
166
+
167
+ [More Information Needed]
168
+
169
+ #### Hardware
170
+
171
+ [More Information Needed]
172
+
173
+ #### Software
174
+
175
+ [More Information Needed]
176
+
177
+ ## Citation [optional]
178
+
179
+ <!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
180
+
181
+ **BibTeX:**
182
+
183
+ [More Information Needed]
184
+
185
+ **APA:**
186
+
187
+ [More Information Needed]
188
+
189
+ ## Glossary [optional]
190
+
191
+ <!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
192
+
193
+ [More Information Needed]
194
+
195
+ ## More Information [optional]
196
+
197
+ [More Information Needed]
198
+
199
+ ## Model Card Authors [optional]
200
+
201
+ [More Information Needed]
202
+
203
+ ## Model Card Contact
204
+
205
+ [More Information Needed]
206
+ ### Framework versions
207
+
208
+ - PEFT 0.17.0
checkpoint-150/adapter_config.json ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alpha_pattern": {},
3
+ "auto_mapping": null,
4
+ "base_model_name_or_path": "tiny-random/gpt-oss",
5
+ "bias": "none",
6
+ "corda_config": null,
7
+ "eva_config": null,
8
+ "exclude_modules": null,
9
+ "fan_in_fan_out": null,
10
+ "inference_mode": true,
11
+ "init_lora_weights": true,
12
+ "layer_replication": null,
13
+ "layers_pattern": null,
14
+ "layers_to_transform": null,
15
+ "loftq_config": {},
16
+ "lora_alpha": 32,
17
+ "lora_bias": false,
18
+ "lora_dropout": 0.1,
19
+ "megatron_config": null,
20
+ "megatron_core": "megatron.core",
21
+ "modules_to_save": null,
22
+ "peft_type": "LORA",
23
+ "qalora_group_size": 16,
24
+ "r": 16,
25
+ "rank_pattern": {},
26
+ "revision": null,
27
+ "target_modules": [
28
+ "v_proj",
29
+ "k_proj",
30
+ "o_proj",
31
+ "gate_proj",
32
+ "q_proj",
33
+ "up_proj",
34
+ "down_proj"
35
+ ],
36
+ "target_parameters": [],
37
+ "task_type": "CAUSAL_LM",
38
+ "trainable_token_indices": null,
39
+ "use_dora": false,
40
+ "use_qalora": false,
41
+ "use_rslora": false
42
+ }
checkpoint-150/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:60bdcacf338c78bba188274dcfcca483b497ddd4fca3300910c8c37922990ff5
3
+ size 43008
checkpoint-150/chat_template.jinja ADDED
@@ -0,0 +1,397 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {#-
2
+ In addition to the normal inputs of `messages` and `tools`, this template also accepts the
3
+ following kwargs:
4
+ - "builtin_tools": A list, can contain "browser" and/or "python".
5
+ - "model_identity": A string that optionally describes the model identity.
6
+ - "reasoning_effort": A string that describes the reasoning effort, defaults to "medium".
7
+ #}
8
+
9
+ {#- Tool Definition Rendering ============================================== #}
10
+ {%- macro render_typescript_type(param_spec, required_params, is_nullable=false) -%}
11
+ {%- if param_spec.type == "array" -%}
12
+ {%- if param_spec['items'] -%}
13
+ {%- if param_spec['items']['type'] == "string" -%}
14
+ {{- "string[]" }}
15
+ {%- elif param_spec['items']['type'] == "number" -%}
16
+ {{- "number[]" }}
17
+ {%- elif param_spec['items']['type'] == "integer" -%}
18
+ {{- "number[]" }}
19
+ {%- elif param_spec['items']['type'] == "boolean" -%}
20
+ {{- "boolean[]" }}
21
+ {%- else -%}
22
+ {%- set inner_type = render_typescript_type(param_spec['items'], required_params) -%}
23
+ {%- if inner_type == "object | object" or inner_type|length > 50 -%}
24
+ {{- "any[]" }}
25
+ {%- else -%}
26
+ {{- inner_type + "[]" }}
27
+ {%- endif -%}
28
+ {%- endif -%}
29
+ {%- if param_spec.nullable -%}
30
+ {{- " | null" }}
31
+ {%- endif -%}
32
+ {%- else -%}
33
+ {{- "any[]" }}
34
+ {%- if param_spec.nullable -%}
35
+ {{- " | null" }}
36
+ {%- endif -%}
37
+ {%- endif -%}
38
+ {%- elif param_spec.type is defined and param_spec.type is iterable and param_spec.type is not string and param_spec.type is not mapping and param_spec.type[0] is defined -%}
39
+ {#- Handle array of types like ["object", "object"] from Union[dict, list] #}
40
+ {%- if param_spec.type | length > 1 -%}
41
+ {{- param_spec.type | join(" | ") }}
42
+ {%- else -%}
43
+ {{- param_spec.type[0] }}
44
+ {%- endif -%}
45
+ {%- elif param_spec.oneOf -%}
46
+ {#- Handle oneOf schemas - check for complex unions and fallback to any #}
47
+ {%- set has_object_variants = false -%}
48
+ {%- for variant in param_spec.oneOf -%}
49
+ {%- if variant.type == "object" -%}
50
+ {%- set has_object_variants = true -%}
51
+ {%- endif -%}
52
+ {%- endfor -%}
53
+ {%- if has_object_variants and param_spec.oneOf|length > 1 -%}
54
+ {{- "any" }}
55
+ {%- else -%}
56
+ {%- for variant in param_spec.oneOf -%}
57
+ {{- render_typescript_type(variant, required_params) -}}
58
+ {%- if variant.description %}
59
+ {{- "// " + variant.description }}
60
+ {%- endif -%}
61
+ {%- if variant.default is defined %}
62
+ {{ "// default: " + variant.default|tojson }}
63
+ {%- endif -%}
64
+ {%- if not loop.last %}
65
+ {{- " | " }}
66
+ {% endif -%}
67
+ {%- endfor -%}
68
+ {%- endif -%}
69
+ {%- elif param_spec.type == "string" -%}
70
+ {%- if param_spec.enum -%}
71
+ {{- '"' + param_spec.enum|join('" | "') + '"' -}}
72
+ {%- else -%}
73
+ {{- "string" }}
74
+ {%- if param_spec.nullable %}
75
+ {{- " | null" }}
76
+ {%- endif -%}
77
+ {%- endif -%}
78
+ {%- elif param_spec.type == "number" -%}
79
+ {{- "number" }}
80
+ {%- elif param_spec.type == "integer" -%}
81
+ {{- "number" }}
82
+ {%- elif param_spec.type == "boolean" -%}
83
+ {{- "boolean" }}
84
+
85
+ {%- elif param_spec.type == "object" -%}
86
+ {%- if param_spec.properties -%}
87
+ {{- "{
88
+ " }}
89
+ {%- for prop_name, prop_spec in param_spec.properties.items() -%}
90
+ {{- prop_name -}}
91
+ {%- if prop_name not in (param_spec.required or []) -%}
92
+ {{- "?" }}
93
+ {%- endif -%}
94
+ {{- ": " }}
95
+ {{ render_typescript_type(prop_spec, param_spec.required or []) }}
96
+ {%- if not loop.last -%}
97
+ {{-", " }}
98
+ {%- endif -%}
99
+ {%- endfor -%}
100
+ {{- "}" }}
101
+ {%- else -%}
102
+ {{- "object" }}
103
+ {%- endif -%}
104
+ {%- else -%}
105
+ {{- "any" }}
106
+ {%- endif -%}
107
+ {%- endmacro -%}
108
+
109
+ {%- macro render_tool_namespace(namespace_name, tools) -%}
110
+ {{- "## " + namespace_name + "
111
+
112
+ " }}
113
+ {{- "namespace " + namespace_name + " {
114
+
115
+ " }}
116
+ {%- for tool in tools %}
117
+ {%- set tool = tool.function %}
118
+ {{- "// " + tool.description + "
119
+ " }}
120
+ {{- "type "+ tool.name + " = " }}
121
+ {%- if tool.parameters and tool.parameters.properties %}
122
+ {{- "(_: {
123
+ " }}
124
+ {%- for param_name, param_spec in tool.parameters.properties.items() %}
125
+ {%- if param_spec.description %}
126
+ {{- "// " + param_spec.description + "
127
+ " }}
128
+ {%- endif %}
129
+ {{- param_name }}
130
+ {%- if param_name not in (tool.parameters.required or []) -%}
131
+ {{- "?" }}
132
+ {%- endif -%}
133
+ {{- ": " }}
134
+ {{- render_typescript_type(param_spec, tool.parameters.required or []) }}
135
+ {%- if param_spec.default is defined -%}
136
+ {%- if param_spec.enum %}
137
+ {{- ", // default: " + param_spec.default }}
138
+ {%- elif param_spec.oneOf %}
139
+ {{- "// default: " + param_spec.default }}
140
+ {%- else %}
141
+ {{- ", // default: " + param_spec.default|tojson }}
142
+ {%- endif -%}
143
+ {%- endif -%}
144
+ {%- if not loop.last %}
145
+ {{- ",
146
+ " }}
147
+ {%- else %}
148
+ {{- "
149
+ " }}
150
+ {%- endif -%}
151
+ {%- endfor %}
152
+ {{- "}) => any;
153
+
154
+ " }}
155
+ {%- else -%}
156
+ {{- "() => any;
157
+
158
+ " }}
159
+ {%- endif -%}
160
+ {%- endfor %}
161
+ {{- "} // namespace " + namespace_name }}
162
+ {%- endmacro -%}
163
+
164
+ {%- macro render_builtin_tools(browser_tool, python_tool) -%}
165
+ {%- if browser_tool %}
166
+ {{- "## browser
167
+
168
+ " }}
169
+ {{- "// Tool for browsing.
170
+ " }}
171
+ {{- "// The `cursor` appears in brackets before each browsing display: `[{cursor}]`.
172
+ " }}
173
+ {{- "// Cite information from the tool using the following format:
174
+ " }}
175
+ {{- "// `【{cursor}†L{line_start}(-L{line_end})?】`, for example: `【6†L9-L11】` or `【8†L3】`.
176
+ " }}
177
+ {{- "// Do not quote more than 10 words directly from the tool output.
178
+ " }}
179
+ {{- "// sources=web (default: web)
180
+ " }}
181
+ {{- "namespace browser {
182
+
183
+ " }}
184
+ {{- "// Searches for information related to `query` and displays `topn` results.
185
+ " }}
186
+ {{- "type search = (_: {
187
+ " }}
188
+ {{- "query: string,
189
+ " }}
190
+ {{- "topn?: number, // default: 10
191
+ " }}
192
+ {{- "source?: string,
193
+ " }}
194
+ {{- "}) => any;
195
+
196
+ " }}
197
+ {{- "// Opens the link `id` from the page indicated by `cursor` starting at line number `loc`, showing `num_lines` lines.
198
+ " }}
199
+ {{- "// Valid link ids are displayed with the formatting: `【{id}†.*】`.
200
+ " }}
201
+ {{- "// If `cursor` is not provided, the most recent page is implied.
202
+ " }}
203
+ {{- "// If `id` is a string, it is treated as a fully qualified URL associated with `source`.
204
+ " }}
205
+ {{- "// If `loc` is not provided, the viewport will be positioned at the beginning of the document or centered on the most relevant passage, if available.
206
+ " }}
207
+ {{- "// Use this function without `id` to scroll to a new location of an opened page.
208
+ " }}
209
+ {{- "type open = (_: {
210
+ " }}
211
+ {{- "id?: number | string, // default: -1
212
+ " }}
213
+ {{- "cursor?: number, // default: -1
214
+ " }}
215
+ {{- "loc?: number, // default: -1
216
+ " }}
217
+ {{- "num_lines?: number, // default: -1
218
+ " }}
219
+ {{- "view_source?: boolean, // default: false
220
+ " }}
221
+ {{- "source?: string,
222
+ " }}
223
+ {{- "}) => any;
224
+
225
+ " }}
226
+ {{- "// Finds exact matches of `pattern` in the current page, or the page given by `cursor`.
227
+ " }}
228
+ {{- "type find = (_: {
229
+ " }}
230
+ {{- "pattern: string,
231
+ " }}
232
+ {{- "cursor?: number, // default: -1
233
+ " }}
234
+ {{- "}) => any;
235
+
236
+ " }}
237
+ {{- "} // namespace browser
238
+
239
+ " }}
240
+ {%- endif -%}
241
+
242
+ {%- if python_tool %}
243
+ {{- "## python
244
+
245
+ " }}
246
+ {{- "Use this tool to execute Python code in your chain of thought. The code will not be shown to the user. This tool should be used for internal reasoning, but not for code that is intended to be visible to the user (e.g. when creating plots, tables, or files).
247
+
248
+ " }}
249
+ {{- "When you send a message containing Python code to python, it will be executed in a stateful Jupyter notebook environment. python will respond with the output of the execution or time out after 120.0 seconds. The drive at '/mnt/data' can be used to save and persist user files. Internet access for this session is UNKNOWN. Depends on the cluster.
250
+
251
+ " }}
252
+ {%- endif -%}
253
+ {%- endmacro -%}
254
+
255
+ {#- System Message Construction ============================================ #}
256
+ {%- macro build_system_message() -%}
257
+ {%- if model_identity is not defined %}
258
+ {%- set model_identity = "You are ChatGPT, a large language model trained by OpenAI." %}
259
+ {%- endif %}
260
+ {{- model_identity + "
261
+ " }}
262
+ {{- "Knowledge cutoff: 2024-06
263
+ " }}
264
+ {{- "Current date: " + strftime_now("%Y-%m-%d") + "
265
+
266
+ " }}
267
+ {%- if reasoning_effort is not defined %}
268
+ {%- set reasoning_effort = "medium" %}
269
+ {%- endif %}
270
+ {{- "Reasoning: " + reasoning_effort + "
271
+
272
+ " }}
273
+ {%- if builtin_tools %}
274
+ {{- "# Tools
275
+
276
+ " }}
277
+ {%- set available_builtin_tools = namespace(browser=false, python=false) %}
278
+ {%- for tool in builtin_tools %}
279
+ {%- if tool == "browser" %}
280
+ {%- set available_builtin_tools.browser = true %}
281
+ {%- elif tool == "python" %}
282
+ {%- set available_builtin_tools.python = true %}
283
+ {%- endif %}
284
+ {%- endfor %}
285
+ {{- render_builtin_tools(available_builtin_tools.browser, available_builtin_tools.python) }}
286
+ {%- endif -%}
287
+ {{- "# Valid channels: analysis, commentary, final. Channel must be included for every message." }}
288
+ {%- if tools -%}
289
+ {{- "
290
+ Calls to these tools must go to the commentary channel: 'functions'." }}
291
+ {%- endif -%}
292
+ {%- endmacro -%}
293
+
294
+ {#- Main Template Logic ================================================= #}
295
+ {#- Set defaults #}
296
+
297
+ {#- Render system message #}
298
+ {{- "<|start|>system<|message|>" }}
299
+ {{- build_system_message() }}
300
+ {{- "<|end|>" }}
301
+
302
+ {#- Extract developer message #}
303
+ {%- if messages[0].role == "developer" or messages[0].role == "system" %}
304
+ {%- set developer_message = messages[0].content %}
305
+ {%- set loop_messages = messages[1:] %}
306
+ {%- else %}
307
+ {%- set developer_message = "" %}
308
+ {%- set loop_messages = messages %}
309
+ {%- endif %}
310
+
311
+ {#- Render developer message #}
312
+ {%- if developer_message or tools %}
313
+ {{- "<|start|>developer<|message|>" }}
314
+ {%- if developer_message %}
315
+ {{- "# Instructions
316
+
317
+ " }}
318
+ {{- developer_message }}
319
+ {%- endif %}
320
+ {%- if tools -%}
321
+ {{- "
322
+
323
+ " }}
324
+ {{- "# Tools
325
+
326
+ " }}
327
+ {{- render_tool_namespace("functions", tools) }}
328
+ {%- endif -%}
329
+ {{- "<|end|>" }}
330
+ {%- endif %}
331
+
332
+ {#- Render messages #}
333
+ {%- set last_tool_call = namespace(name=none) %}
334
+ {%- for message in loop_messages -%}
335
+ {#- At this point only assistant/user/tool messages should remain #}
336
+ {%- if message.role == 'assistant' -%}
337
+ {#- Checks to ensure the messages are being passed in the format we expect #}
338
+ {%- if "content" in message %}
339
+ {%- if "<|channel|>analysis<|message|>" in message.content or "<|channel|>final<|message|>" in message.content %}
340
+ {{- raise_exception("You have passed a message containing <|channel|> tags in the content field. Instead of doing this, you should pass analysis messages (the string between '<|message|>' and '<|end|>') in the 'thinking' field, and final messages (the string between '<|message|>' and '<|end|>') in the 'content' field.") }}
341
+ {%- endif %}
342
+ {%- endif %}
343
+ {%- if "thinking" in message %}
344
+ {%- if "<|channel|>analysis<|message|>" in message.thinking or "<|channel|>final<|message|>" in message.thinking %}
345
+ {{- raise_exception("You have passed a message containing <|channel|> tags in the thinking field. Instead of doing this, you should pass analysis messages (the string between '<|message|>' and '<|end|>') in the 'thinking' field, and final messages (the string between '<|message|>' and '<|end|>') in the 'content' field.") }}
346
+ {%- endif %}
347
+ {%- endif %}
348
+ {%- if "tool_calls" in message %}
349
+ {#- We assume max 1 tool call per message, and so we infer the tool call name #}
350
+ {#- in "tool" messages from the most recent assistant tool call name #}
351
+ {%- set tool_call = message.tool_calls[0] %}
352
+ {%- if tool_call.function %}
353
+ {%- set tool_call = tool_call.function %}
354
+ {%- endif %}
355
+ {%- if message.content and message.thinking %}
356
+ {{- raise_exception("Cannot pass both content and thinking in an assistant message with tool calls! Put the analysis message in one or the other, but not both.") }}
357
+ {%- elif message.content %}
358
+ {{- "<|start|>assistant<|channel|>analysis<|message|>" + message.content + "<|end|>" }}
359
+ {%- elif message.thinking %}
360
+ {{- "<|start|>assistant<|channel|>analysis<|message|>" + message.thinking + "<|end|>" }}
361
+ {%- endif %}
362
+ {{- "<|start|>assistant to=" }}
363
+ {{- "functions." + tool_call.name + "<|channel|>commentary " }}
364
+ {{- (tool_call.content_type if tool_call.content_type is defined else "json") + "<|message|>" }}
365
+ {{- tool_call.arguments|tojson }}
366
+ {{- "<|call|>" }}
367
+ {%- set last_tool_call.name = tool_call.name %}
368
+ {%- elif loop.last and not add_generation_prompt %}
369
+ {#- Only render the CoT if the final turn is an assistant turn and add_generation_prompt is false #}
370
+ {#- This is a situation that should only occur in training, never in inference. #}
371
+ {%- if "thinking" in message %}
372
+ {{- "<|start|>assistant<|channel|>analysis<|message|>" + message.thinking + "<|end|>" }}
373
+ {%- endif %}
374
+ {#- <|return|> indicates the end of generation, but <|end|> does not #}
375
+ {#- <|return|> should never be an input to the model, but we include it as the final token #}
376
+ {#- when training, so the model learns to emit it. #}
377
+ {{- "<|start|>assistant<|channel|>final<|message|>" + message.content + "<|return|>" }}
378
+ {%- else %}
379
+ {#- CoT is dropped during all previous turns, so we never render it for inference #}
380
+ {{- "<|start|>assistant<|channel|>final<|message|>" + message.content + "<|end|>" }}
381
+ {%- set last_tool_call.name = none %}
382
+ {%- endif %}
383
+ {%- elif message.role == 'tool' -%}
384
+ {%- if last_tool_call.name is none %}
385
+ {{- raise_exception("Message has tool role, but there was no previous assistant message with a tool call!") }}
386
+ {%- endif %}
387
+ {{- "<|start|>functions." + last_tool_call.name }}
388
+ {{- " to=assistant<|channel|>commentary<|message|>" + message.content|tojson + "<|end|>" }}
389
+ {%- elif message.role == 'user' -%}
390
+ {{- "<|start|>user<|message|>" + message.content + "<|end|>" }}
391
+ {%- endif -%}
392
+ {%- endfor -%}
393
+
394
+ {#- Generation prompt #}
395
+ {%- if add_generation_prompt -%}
396
+ <|start|>assistant
397
+ {%- endif -%}
checkpoint-150/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e0c8b207642e60ccaa18f6b87430a6013cd74cc22a13966fda4e9c27793bb325
3
+ size 96746
checkpoint-150/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:aa4e9ea328bdf2792816bf9a77ed4a4c18f5cb18a68ea1062e85334c33207f9a
3
+ size 14244
checkpoint-150/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:da800fd5ac660f36778294f9df633c4f039b5f7bde42ca6a92dea5ac7d9666b1
3
+ size 988
checkpoint-150/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a66b2297c2db113bfe44f16fa19c080bc9b78ae5e0a5199d83ce1faacaaa20ab
3
+ size 2080
checkpoint-150/special_tokens_map.json ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "<|startoftext|>",
4
+ "lstrip": false,
5
+ "normalized": false,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "eos_token": {
10
+ "content": "<|return|>",
11
+ "lstrip": false,
12
+ "normalized": false,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "pad_token": {
17
+ "content": "<|endoftext|>",
18
+ "lstrip": false,
19
+ "normalized": false,
20
+ "rstrip": false,
21
+ "single_word": false
22
+ }
23
+ }
checkpoint-150/tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0614fe83cadab421296e664e1f48f4261fa8fef6e03e63bb75c20f38e37d07d3
3
+ size 27868174
checkpoint-150/tokenizer_config.json ADDED
@@ -0,0 +1,183 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "added_tokens_decoder": {
3
+ "199998": {
4
+ "content": "<|startoftext|>",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false,
9
+ "special": true
10
+ },
11
+ "199999": {
12
+ "content": "<|endoftext|>",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false,
17
+ "special": true
18
+ },
19
+ "200000": {
20
+ "content": "<|reserved_200000|>",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false,
25
+ "special": true
26
+ },
27
+ "200001": {
28
+ "content": "<|reserved_200001|>",
29
+ "lstrip": false,
30
+ "normalized": false,
31
+ "rstrip": false,
32
+ "single_word": false,
33
+ "special": true
34
+ },
35
+ "200002": {
36
+ "content": "<|return|>",
37
+ "lstrip": false,
38
+ "normalized": false,
39
+ "rstrip": false,
40
+ "single_word": false,
41
+ "special": true
42
+ },
43
+ "200003": {
44
+ "content": "<|constrain|>",
45
+ "lstrip": false,
46
+ "normalized": false,
47
+ "rstrip": false,
48
+ "single_word": false,
49
+ "special": true
50
+ },
51
+ "200004": {
52
+ "content": "<|reserved_200004|>",
53
+ "lstrip": false,
54
+ "normalized": false,
55
+ "rstrip": false,
56
+ "single_word": false,
57
+ "special": true
58
+ },
59
+ "200005": {
60
+ "content": "<|channel|>",
61
+ "lstrip": false,
62
+ "normalized": false,
63
+ "rstrip": false,
64
+ "single_word": false,
65
+ "special": true
66
+ },
67
+ "200006": {
68
+ "content": "<|start|>",
69
+ "lstrip": false,
70
+ "normalized": false,
71
+ "rstrip": false,
72
+ "single_word": false,
73
+ "special": true
74
+ },
75
+ "200007": {
76
+ "content": "<|end|>",
77
+ "lstrip": false,
78
+ "normalized": false,
79
+ "rstrip": false,
80
+ "single_word": false,
81
+ "special": true
82
+ },
83
+ "200008": {
84
+ "content": "<|message|>",
85
+ "lstrip": false,
86
+ "normalized": false,
87
+ "rstrip": false,
88
+ "single_word": false,
89
+ "special": true
90
+ },
91
+ "200009": {
92
+ "content": "<|reserved_200009|>",
93
+ "lstrip": false,
94
+ "normalized": false,
95
+ "rstrip": false,
96
+ "single_word": false,
97
+ "special": true
98
+ },
99
+ "200010": {
100
+ "content": "<|reserved_200010|>",
101
+ "lstrip": false,
102
+ "normalized": false,
103
+ "rstrip": false,
104
+ "single_word": false,
105
+ "special": true
106
+ },
107
+ "200011": {
108
+ "content": "<|reserved_200011|>",
109
+ "lstrip": false,
110
+ "normalized": false,
111
+ "rstrip": false,
112
+ "single_word": false,
113
+ "special": true
114
+ },
115
+ "200012": {
116
+ "content": "<|call|>",
117
+ "lstrip": false,
118
+ "normalized": false,
119
+ "rstrip": false,
120
+ "single_word": false,
121
+ "special": true
122
+ },
123
+ "200013": {
124
+ "content": "<|reserved_200013|>",
125
+ "lstrip": false,
126
+ "normalized": false,
127
+ "rstrip": false,
128
+ "single_word": false,
129
+ "special": true
130
+ },
131
+ "200014": {
132
+ "content": "<|reserved_200014|>",
133
+ "lstrip": false,
134
+ "normalized": false,
135
+ "rstrip": false,
136
+ "single_word": false,
137
+ "special": true
138
+ },
139
+ "200015": {
140
+ "content": "<|reserved_200015|>",
141
+ "lstrip": false,
142
+ "normalized": false,
143
+ "rstrip": false,
144
+ "single_word": false,
145
+ "special": true
146
+ },
147
+ "200016": {
148
+ "content": "<|reserved_200016|>",
149
+ "lstrip": false,
150
+ "normalized": false,
151
+ "rstrip": false,
152
+ "single_word": false,
153
+ "special": true
154
+ },
155
+ "200017": {
156
+ "content": "<|reserved_200017|>",
157
+ "lstrip": false,
158
+ "normalized": false,
159
+ "rstrip": false,
160
+ "single_word": false,
161
+ "special": true
162
+ },
163
+ "200018": {
164
+ "content": "<|endofprompt|>",
165
+ "lstrip": false,
166
+ "normalized": false,
167
+ "rstrip": false,
168
+ "single_word": false,
169
+ "special": true
170
+ }
171
+ },
172
+ "bos_token": "<|startoftext|>",
173
+ "clean_up_tokenization_spaces": false,
174
+ "eos_token": "<|return|>",
175
+ "extra_special_tokens": {},
176
+ "model_input_names": [
177
+ "input_ids",
178
+ "attention_mask"
179
+ ],
180
+ "model_max_length": 1000000000000000019884624838656,
181
+ "pad_token": "<|endoftext|>",
182
+ "tokenizer_class": "PreTrainedTokenizerFast"
183
+ }
checkpoint-150/trainer_state.json ADDED
@@ -0,0 +1,184 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 1.316831683168317,
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.08800880088008801,
14
+ "grad_norm": 0.008909749798476696,
15
+ "learning_rate": 8.18181797512807e-05,
16
+ "loss": 12.2157,
17
+ "memory/device_mem_reserved(gib)": 2.73,
18
+ "memory/max_mem_active(gib)": 2.34,
19
+ "memory/max_mem_allocated(gib)": 2.34,
20
+ "step": 10
21
+ },
22
+ {
23
+ "epoch": 0.17601760176017603,
24
+ "grad_norm": 0.007974537089467049,
25
+ "learning_rate": 9.999999747378752e-05,
26
+ "loss": 12.2154,
27
+ "memory/device_mem_reserved(gib)": 2.73,
28
+ "memory/max_mem_active(gib)": 2.34,
29
+ "memory/max_mem_allocated(gib)": 2.34,
30
+ "step": 20
31
+ },
32
+ {
33
+ "epoch": 0.264026402640264,
34
+ "grad_norm": 0.010009167715907097,
35
+ "learning_rate": 9.999999747378752e-05,
36
+ "loss": 12.2152,
37
+ "memory/device_mem_reserved(gib)": 2.73,
38
+ "memory/max_mem_active(gib)": 2.34,
39
+ "memory/max_mem_allocated(gib)": 2.34,
40
+ "step": 30
41
+ },
42
+ {
43
+ "epoch": 0.35203520352035206,
44
+ "grad_norm": 0.010353786870837212,
45
+ "learning_rate": 9.999999747378752e-05,
46
+ "loss": 12.2155,
47
+ "memory/device_mem_reserved(gib)": 2.73,
48
+ "memory/max_mem_active(gib)": 2.34,
49
+ "memory/max_mem_allocated(gib)": 2.34,
50
+ "step": 40
51
+ },
52
+ {
53
+ "epoch": 0.44004400440044006,
54
+ "grad_norm": 0.01487517450004816,
55
+ "learning_rate": 9.999999747378752e-05,
56
+ "loss": 12.2151,
57
+ "memory/device_mem_reserved(gib)": 2.73,
58
+ "memory/max_mem_active(gib)": 2.34,
59
+ "memory/max_mem_allocated(gib)": 2.34,
60
+ "step": 50
61
+ },
62
+ {
63
+ "epoch": 0.528052805280528,
64
+ "grad_norm": 0.011968914419412613,
65
+ "learning_rate": 9.999999747378752e-05,
66
+ "loss": 12.215,
67
+ "memory/device_mem_reserved(gib)": 2.73,
68
+ "memory/max_mem_active(gib)": 2.34,
69
+ "memory/max_mem_allocated(gib)": 2.34,
70
+ "step": 60
71
+ },
72
+ {
73
+ "epoch": 0.6160616061606161,
74
+ "grad_norm": 0.015747424215078354,
75
+ "learning_rate": 9.999999747378752e-05,
76
+ "loss": 12.2146,
77
+ "memory/device_mem_reserved(gib)": 2.73,
78
+ "memory/max_mem_active(gib)": 2.34,
79
+ "memory/max_mem_allocated(gib)": 2.34,
80
+ "step": 70
81
+ },
82
+ {
83
+ "epoch": 0.7040704070407041,
84
+ "grad_norm": 0.01620076224207878,
85
+ "learning_rate": 9.999999747378752e-05,
86
+ "loss": 12.2132,
87
+ "memory/device_mem_reserved(gib)": 2.73,
88
+ "memory/max_mem_active(gib)": 2.34,
89
+ "memory/max_mem_allocated(gib)": 2.34,
90
+ "step": 80
91
+ },
92
+ {
93
+ "epoch": 0.7920792079207921,
94
+ "grad_norm": 0.016200672835111618,
95
+ "learning_rate": 9.999999747378752e-05,
96
+ "loss": 12.2131,
97
+ "memory/device_mem_reserved(gib)": 2.73,
98
+ "memory/max_mem_active(gib)": 2.34,
99
+ "memory/max_mem_allocated(gib)": 2.34,
100
+ "step": 90
101
+ },
102
+ {
103
+ "epoch": 0.8800880088008801,
104
+ "grad_norm": 0.01671593263745308,
105
+ "learning_rate": 9.999999747378752e-05,
106
+ "loss": 12.2126,
107
+ "memory/device_mem_reserved(gib)": 2.73,
108
+ "memory/max_mem_active(gib)": 2.34,
109
+ "memory/max_mem_allocated(gib)": 2.34,
110
+ "step": 100
111
+ },
112
+ {
113
+ "epoch": 0.9680968096809681,
114
+ "grad_norm": 0.019221968948841095,
115
+ "learning_rate": 9.999999747378752e-05,
116
+ "loss": 12.2124,
117
+ "memory/device_mem_reserved(gib)": 2.73,
118
+ "memory/max_mem_active(gib)": 2.34,
119
+ "memory/max_mem_allocated(gib)": 2.34,
120
+ "step": 110
121
+ },
122
+ {
123
+ "epoch": 1.0528052805280528,
124
+ "grad_norm": 0.02141665853559971,
125
+ "learning_rate": 9.999999747378752e-05,
126
+ "loss": 12.2117,
127
+ "memory/device_mem_reserved(gib)": 2.73,
128
+ "memory/max_mem_active(gib)": 2.34,
129
+ "memory/max_mem_allocated(gib)": 2.34,
130
+ "step": 120
131
+ },
132
+ {
133
+ "epoch": 1.140814081408141,
134
+ "grad_norm": 0.015679990872740746,
135
+ "learning_rate": 9.999999747378752e-05,
136
+ "loss": 12.2111,
137
+ "memory/device_mem_reserved(gib)": 2.74,
138
+ "memory/max_mem_active(gib)": 2.34,
139
+ "memory/max_mem_allocated(gib)": 2.34,
140
+ "step": 130
141
+ },
142
+ {
143
+ "epoch": 1.2288228822882288,
144
+ "grad_norm": 0.022765284404158592,
145
+ "learning_rate": 9.999999747378752e-05,
146
+ "loss": 12.2093,
147
+ "memory/device_mem_reserved(gib)": 2.74,
148
+ "memory/max_mem_active(gib)": 2.34,
149
+ "memory/max_mem_allocated(gib)": 2.34,
150
+ "step": 140
151
+ },
152
+ {
153
+ "epoch": 1.316831683168317,
154
+ "grad_norm": 0.01766524650156498,
155
+ "learning_rate": 9.999999747378752e-05,
156
+ "loss": 12.2088,
157
+ "memory/device_mem_reserved(gib)": 2.74,
158
+ "memory/max_mem_active(gib)": 2.34,
159
+ "memory/max_mem_allocated(gib)": 2.34,
160
+ "step": 150
161
+ }
162
+ ],
163
+ "logging_steps": 10,
164
+ "max_steps": 300,
165
+ "num_input_tokens_seen": 0,
166
+ "num_train_epochs": 3,
167
+ "save_steps": 150,
168
+ "stateful_callbacks": {
169
+ "TrainerControl": {
170
+ "args": {
171
+ "should_epoch_stop": false,
172
+ "should_evaluate": false,
173
+ "should_log": false,
174
+ "should_save": true,
175
+ "should_training_stop": false
176
+ },
177
+ "attributes": {}
178
+ }
179
+ },
180
+ "total_flos": 3064545861120.0,
181
+ "train_batch_size": 4,
182
+ "trial_name": null,
183
+ "trial_params": null
184
+ }
checkpoint-150/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a33123adb3cce6429a0192902873aaa084cd85cfcdd34c6b726552ee2bb1fa77
3
+ size 6776
checkpoint-300/README.md ADDED
@@ -0,0 +1,208 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ base_model: tiny-random/gpt-oss
3
+ library_name: peft
4
+ pipeline_tag: text-generation
5
+ tags:
6
+ - axolotl
7
+ - base_model:adapter:tiny-random/gpt-oss
8
+ - lora
9
+ - transformers
10
+ ---
11
+
12
+ # Model Card for Model ID
13
+
14
+ <!-- Provide a quick summary of what the model is/does. -->
15
+
16
+
17
+
18
+ ## Model Details
19
+
20
+ ### Model Description
21
+
22
+ <!-- Provide a longer summary of what this model is. -->
23
+
24
+
25
+
26
+ - **Developed by:** [More Information Needed]
27
+ - **Funded by [optional]:** [More Information Needed]
28
+ - **Shared by [optional]:** [More Information Needed]
29
+ - **Model type:** [More Information Needed]
30
+ - **Language(s) (NLP):** [More Information Needed]
31
+ - **License:** [More Information Needed]
32
+ - **Finetuned from model [optional]:** [More Information Needed]
33
+
34
+ ### Model Sources [optional]
35
+
36
+ <!-- Provide the basic links for the model. -->
37
+
38
+ - **Repository:** [More Information Needed]
39
+ - **Paper [optional]:** [More Information Needed]
40
+ - **Demo [optional]:** [More Information Needed]
41
+
42
+ ## Uses
43
+
44
+ <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
45
+
46
+ ### Direct Use
47
+
48
+ <!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
49
+
50
+ [More Information Needed]
51
+
52
+ ### Downstream Use [optional]
53
+
54
+ <!-- This section is for the model use when fine-tuned for a task, or when plugged into a larger ecosystem/app -->
55
+
56
+ [More Information Needed]
57
+
58
+ ### Out-of-Scope Use
59
+
60
+ <!-- This section addresses misuse, malicious use, and uses that the model will not work well for. -->
61
+
62
+ [More Information Needed]
63
+
64
+ ## Bias, Risks, and Limitations
65
+
66
+ <!-- This section is meant to convey both technical and sociotechnical limitations. -->
67
+
68
+ [More Information Needed]
69
+
70
+ ### Recommendations
71
+
72
+ <!-- This section is meant to convey recommendations with respect to the bias, risk, and technical limitations. -->
73
+
74
+ Users (both direct and downstream) should be made aware of the risks, biases and limitations of the model. More information needed for further recommendations.
75
+
76
+ ## How to Get Started with the Model
77
+
78
+ Use the code below to get started with the model.
79
+
80
+ [More Information Needed]
81
+
82
+ ## Training Details
83
+
84
+ ### Training Data
85
+
86
+ <!-- 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. -->
87
+
88
+ [More Information Needed]
89
+
90
+ ### Training Procedure
91
+
92
+ <!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
93
+
94
+ #### Preprocessing [optional]
95
+
96
+ [More Information Needed]
97
+
98
+
99
+ #### Training Hyperparameters
100
+
101
+ - **Training regime:** [More Information Needed] <!--fp32, fp16 mixed precision, bf16 mixed precision, bf16 non-mixed precision, fp16 non-mixed precision, fp8 mixed precision -->
102
+
103
+ #### Speeds, Sizes, Times [optional]
104
+
105
+ <!-- This section provides information about throughput, start/end time, checkpoint size if relevant, etc. -->
106
+
107
+ [More Information Needed]
108
+
109
+ ## Evaluation
110
+
111
+ <!-- This section describes the evaluation protocols and provides the results. -->
112
+
113
+ ### Testing Data, Factors & Metrics
114
+
115
+ #### Testing Data
116
+
117
+ <!-- This should link to a Dataset Card if possible. -->
118
+
119
+ [More Information Needed]
120
+
121
+ #### Factors
122
+
123
+ <!-- These are the things the evaluation is disaggregating by, e.g., subpopulations or domains. -->
124
+
125
+ [More Information Needed]
126
+
127
+ #### Metrics
128
+
129
+ <!-- These are the evaluation metrics being used, ideally with a description of why. -->
130
+
131
+ [More Information Needed]
132
+
133
+ ### Results
134
+
135
+ [More Information Needed]
136
+
137
+ #### Summary
138
+
139
+
140
+
141
+ ## Model Examination [optional]
142
+
143
+ <!-- Relevant interpretability work for the model goes here -->
144
+
145
+ [More Information Needed]
146
+
147
+ ## Environmental Impact
148
+
149
+ <!-- Total emissions (in grams of CO2eq) and additional considerations, such as electricity usage, go here. Edit the suggested text below accordingly -->
150
+
151
+ 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).
152
+
153
+ - **Hardware Type:** [More Information Needed]
154
+ - **Hours used:** [More Information Needed]
155
+ - **Cloud Provider:** [More Information Needed]
156
+ - **Compute Region:** [More Information Needed]
157
+ - **Carbon Emitted:** [More Information Needed]
158
+
159
+ ## Technical Specifications [optional]
160
+
161
+ ### Model Architecture and Objective
162
+
163
+ [More Information Needed]
164
+
165
+ ### Compute Infrastructure
166
+
167
+ [More Information Needed]
168
+
169
+ #### Hardware
170
+
171
+ [More Information Needed]
172
+
173
+ #### Software
174
+
175
+ [More Information Needed]
176
+
177
+ ## Citation [optional]
178
+
179
+ <!-- If there is a paper or blog post introducing the model, the APA and Bibtex information for that should go in this section. -->
180
+
181
+ **BibTeX:**
182
+
183
+ [More Information Needed]
184
+
185
+ **APA:**
186
+
187
+ [More Information Needed]
188
+
189
+ ## Glossary [optional]
190
+
191
+ <!-- If relevant, include terms and calculations in this section that can help readers understand the model or model card. -->
192
+
193
+ [More Information Needed]
194
+
195
+ ## More Information [optional]
196
+
197
+ [More Information Needed]
198
+
199
+ ## Model Card Authors [optional]
200
+
201
+ [More Information Needed]
202
+
203
+ ## Model Card Contact
204
+
205
+ [More Information Needed]
206
+ ### Framework versions
207
+
208
+ - PEFT 0.17.0
checkpoint-300/adapter_config.json ADDED
@@ -0,0 +1,42 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "alpha_pattern": {},
3
+ "auto_mapping": null,
4
+ "base_model_name_or_path": "tiny-random/gpt-oss",
5
+ "bias": "none",
6
+ "corda_config": null,
7
+ "eva_config": null,
8
+ "exclude_modules": null,
9
+ "fan_in_fan_out": null,
10
+ "inference_mode": true,
11
+ "init_lora_weights": true,
12
+ "layer_replication": null,
13
+ "layers_pattern": null,
14
+ "layers_to_transform": null,
15
+ "loftq_config": {},
16
+ "lora_alpha": 32,
17
+ "lora_bias": false,
18
+ "lora_dropout": 0.1,
19
+ "megatron_config": null,
20
+ "megatron_core": "megatron.core",
21
+ "modules_to_save": null,
22
+ "peft_type": "LORA",
23
+ "qalora_group_size": 16,
24
+ "r": 16,
25
+ "rank_pattern": {},
26
+ "revision": null,
27
+ "target_modules": [
28
+ "v_proj",
29
+ "k_proj",
30
+ "o_proj",
31
+ "gate_proj",
32
+ "q_proj",
33
+ "up_proj",
34
+ "down_proj"
35
+ ],
36
+ "target_parameters": [],
37
+ "task_type": "CAUSAL_LM",
38
+ "trainable_token_indices": null,
39
+ "use_dora": false,
40
+ "use_qalora": false,
41
+ "use_rslora": false
42
+ }
checkpoint-300/adapter_model.safetensors ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6fe39862f113460291406bcad1362d191b55143cfb6b365ca9dcdc304ea5a47e
3
+ size 43008
checkpoint-300/chat_template.jinja ADDED
@@ -0,0 +1,397 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {#-
2
+ In addition to the normal inputs of `messages` and `tools`, this template also accepts the
3
+ following kwargs:
4
+ - "builtin_tools": A list, can contain "browser" and/or "python".
5
+ - "model_identity": A string that optionally describes the model identity.
6
+ - "reasoning_effort": A string that describes the reasoning effort, defaults to "medium".
7
+ #}
8
+
9
+ {#- Tool Definition Rendering ============================================== #}
10
+ {%- macro render_typescript_type(param_spec, required_params, is_nullable=false) -%}
11
+ {%- if param_spec.type == "array" -%}
12
+ {%- if param_spec['items'] -%}
13
+ {%- if param_spec['items']['type'] == "string" -%}
14
+ {{- "string[]" }}
15
+ {%- elif param_spec['items']['type'] == "number" -%}
16
+ {{- "number[]" }}
17
+ {%- elif param_spec['items']['type'] == "integer" -%}
18
+ {{- "number[]" }}
19
+ {%- elif param_spec['items']['type'] == "boolean" -%}
20
+ {{- "boolean[]" }}
21
+ {%- else -%}
22
+ {%- set inner_type = render_typescript_type(param_spec['items'], required_params) -%}
23
+ {%- if inner_type == "object | object" or inner_type|length > 50 -%}
24
+ {{- "any[]" }}
25
+ {%- else -%}
26
+ {{- inner_type + "[]" }}
27
+ {%- endif -%}
28
+ {%- endif -%}
29
+ {%- if param_spec.nullable -%}
30
+ {{- " | null" }}
31
+ {%- endif -%}
32
+ {%- else -%}
33
+ {{- "any[]" }}
34
+ {%- if param_spec.nullable -%}
35
+ {{- " | null" }}
36
+ {%- endif -%}
37
+ {%- endif -%}
38
+ {%- elif param_spec.type is defined and param_spec.type is iterable and param_spec.type is not string and param_spec.type is not mapping and param_spec.type[0] is defined -%}
39
+ {#- Handle array of types like ["object", "object"] from Union[dict, list] #}
40
+ {%- if param_spec.type | length > 1 -%}
41
+ {{- param_spec.type | join(" | ") }}
42
+ {%- else -%}
43
+ {{- param_spec.type[0] }}
44
+ {%- endif -%}
45
+ {%- elif param_spec.oneOf -%}
46
+ {#- Handle oneOf schemas - check for complex unions and fallback to any #}
47
+ {%- set has_object_variants = false -%}
48
+ {%- for variant in param_spec.oneOf -%}
49
+ {%- if variant.type == "object" -%}
50
+ {%- set has_object_variants = true -%}
51
+ {%- endif -%}
52
+ {%- endfor -%}
53
+ {%- if has_object_variants and param_spec.oneOf|length > 1 -%}
54
+ {{- "any" }}
55
+ {%- else -%}
56
+ {%- for variant in param_spec.oneOf -%}
57
+ {{- render_typescript_type(variant, required_params) -}}
58
+ {%- if variant.description %}
59
+ {{- "// " + variant.description }}
60
+ {%- endif -%}
61
+ {%- if variant.default is defined %}
62
+ {{ "// default: " + variant.default|tojson }}
63
+ {%- endif -%}
64
+ {%- if not loop.last %}
65
+ {{- " | " }}
66
+ {% endif -%}
67
+ {%- endfor -%}
68
+ {%- endif -%}
69
+ {%- elif param_spec.type == "string" -%}
70
+ {%- if param_spec.enum -%}
71
+ {{- '"' + param_spec.enum|join('" | "') + '"' -}}
72
+ {%- else -%}
73
+ {{- "string" }}
74
+ {%- if param_spec.nullable %}
75
+ {{- " | null" }}
76
+ {%- endif -%}
77
+ {%- endif -%}
78
+ {%- elif param_spec.type == "number" -%}
79
+ {{- "number" }}
80
+ {%- elif param_spec.type == "integer" -%}
81
+ {{- "number" }}
82
+ {%- elif param_spec.type == "boolean" -%}
83
+ {{- "boolean" }}
84
+
85
+ {%- elif param_spec.type == "object" -%}
86
+ {%- if param_spec.properties -%}
87
+ {{- "{
88
+ " }}
89
+ {%- for prop_name, prop_spec in param_spec.properties.items() -%}
90
+ {{- prop_name -}}
91
+ {%- if prop_name not in (param_spec.required or []) -%}
92
+ {{- "?" }}
93
+ {%- endif -%}
94
+ {{- ": " }}
95
+ {{ render_typescript_type(prop_spec, param_spec.required or []) }}
96
+ {%- if not loop.last -%}
97
+ {{-", " }}
98
+ {%- endif -%}
99
+ {%- endfor -%}
100
+ {{- "}" }}
101
+ {%- else -%}
102
+ {{- "object" }}
103
+ {%- endif -%}
104
+ {%- else -%}
105
+ {{- "any" }}
106
+ {%- endif -%}
107
+ {%- endmacro -%}
108
+
109
+ {%- macro render_tool_namespace(namespace_name, tools) -%}
110
+ {{- "## " + namespace_name + "
111
+
112
+ " }}
113
+ {{- "namespace " + namespace_name + " {
114
+
115
+ " }}
116
+ {%- for tool in tools %}
117
+ {%- set tool = tool.function %}
118
+ {{- "// " + tool.description + "
119
+ " }}
120
+ {{- "type "+ tool.name + " = " }}
121
+ {%- if tool.parameters and tool.parameters.properties %}
122
+ {{- "(_: {
123
+ " }}
124
+ {%- for param_name, param_spec in tool.parameters.properties.items() %}
125
+ {%- if param_spec.description %}
126
+ {{- "// " + param_spec.description + "
127
+ " }}
128
+ {%- endif %}
129
+ {{- param_name }}
130
+ {%- if param_name not in (tool.parameters.required or []) -%}
131
+ {{- "?" }}
132
+ {%- endif -%}
133
+ {{- ": " }}
134
+ {{- render_typescript_type(param_spec, tool.parameters.required or []) }}
135
+ {%- if param_spec.default is defined -%}
136
+ {%- if param_spec.enum %}
137
+ {{- ", // default: " + param_spec.default }}
138
+ {%- elif param_spec.oneOf %}
139
+ {{- "// default: " + param_spec.default }}
140
+ {%- else %}
141
+ {{- ", // default: " + param_spec.default|tojson }}
142
+ {%- endif -%}
143
+ {%- endif -%}
144
+ {%- if not loop.last %}
145
+ {{- ",
146
+ " }}
147
+ {%- else %}
148
+ {{- "
149
+ " }}
150
+ {%- endif -%}
151
+ {%- endfor %}
152
+ {{- "}) => any;
153
+
154
+ " }}
155
+ {%- else -%}
156
+ {{- "() => any;
157
+
158
+ " }}
159
+ {%- endif -%}
160
+ {%- endfor %}
161
+ {{- "} // namespace " + namespace_name }}
162
+ {%- endmacro -%}
163
+
164
+ {%- macro render_builtin_tools(browser_tool, python_tool) -%}
165
+ {%- if browser_tool %}
166
+ {{- "## browser
167
+
168
+ " }}
169
+ {{- "// Tool for browsing.
170
+ " }}
171
+ {{- "// The `cursor` appears in brackets before each browsing display: `[{cursor}]`.
172
+ " }}
173
+ {{- "// Cite information from the tool using the following format:
174
+ " }}
175
+ {{- "// `【{cursor}†L{line_start}(-L{line_end})?】`, for example: `【6†L9-L11】` or `【8†L3】`.
176
+ " }}
177
+ {{- "// Do not quote more than 10 words directly from the tool output.
178
+ " }}
179
+ {{- "// sources=web (default: web)
180
+ " }}
181
+ {{- "namespace browser {
182
+
183
+ " }}
184
+ {{- "// Searches for information related to `query` and displays `topn` results.
185
+ " }}
186
+ {{- "type search = (_: {
187
+ " }}
188
+ {{- "query: string,
189
+ " }}
190
+ {{- "topn?: number, // default: 10
191
+ " }}
192
+ {{- "source?: string,
193
+ " }}
194
+ {{- "}) => any;
195
+
196
+ " }}
197
+ {{- "// Opens the link `id` from the page indicated by `cursor` starting at line number `loc`, showing `num_lines` lines.
198
+ " }}
199
+ {{- "// Valid link ids are displayed with the formatting: `【{id}†.*】`.
200
+ " }}
201
+ {{- "// If `cursor` is not provided, the most recent page is implied.
202
+ " }}
203
+ {{- "// If `id` is a string, it is treated as a fully qualified URL associated with `source`.
204
+ " }}
205
+ {{- "// If `loc` is not provided, the viewport will be positioned at the beginning of the document or centered on the most relevant passage, if available.
206
+ " }}
207
+ {{- "// Use this function without `id` to scroll to a new location of an opened page.
208
+ " }}
209
+ {{- "type open = (_: {
210
+ " }}
211
+ {{- "id?: number | string, // default: -1
212
+ " }}
213
+ {{- "cursor?: number, // default: -1
214
+ " }}
215
+ {{- "loc?: number, // default: -1
216
+ " }}
217
+ {{- "num_lines?: number, // default: -1
218
+ " }}
219
+ {{- "view_source?: boolean, // default: false
220
+ " }}
221
+ {{- "source?: string,
222
+ " }}
223
+ {{- "}) => any;
224
+
225
+ " }}
226
+ {{- "// Finds exact matches of `pattern` in the current page, or the page given by `cursor`.
227
+ " }}
228
+ {{- "type find = (_: {
229
+ " }}
230
+ {{- "pattern: string,
231
+ " }}
232
+ {{- "cursor?: number, // default: -1
233
+ " }}
234
+ {{- "}) => any;
235
+
236
+ " }}
237
+ {{- "} // namespace browser
238
+
239
+ " }}
240
+ {%- endif -%}
241
+
242
+ {%- if python_tool %}
243
+ {{- "## python
244
+
245
+ " }}
246
+ {{- "Use this tool to execute Python code in your chain of thought. The code will not be shown to the user. This tool should be used for internal reasoning, but not for code that is intended to be visible to the user (e.g. when creating plots, tables, or files).
247
+
248
+ " }}
249
+ {{- "When you send a message containing Python code to python, it will be executed in a stateful Jupyter notebook environment. python will respond with the output of the execution or time out after 120.0 seconds. The drive at '/mnt/data' can be used to save and persist user files. Internet access for this session is UNKNOWN. Depends on the cluster.
250
+
251
+ " }}
252
+ {%- endif -%}
253
+ {%- endmacro -%}
254
+
255
+ {#- System Message Construction ============================================ #}
256
+ {%- macro build_system_message() -%}
257
+ {%- if model_identity is not defined %}
258
+ {%- set model_identity = "You are ChatGPT, a large language model trained by OpenAI." %}
259
+ {%- endif %}
260
+ {{- model_identity + "
261
+ " }}
262
+ {{- "Knowledge cutoff: 2024-06
263
+ " }}
264
+ {{- "Current date: " + strftime_now("%Y-%m-%d") + "
265
+
266
+ " }}
267
+ {%- if reasoning_effort is not defined %}
268
+ {%- set reasoning_effort = "medium" %}
269
+ {%- endif %}
270
+ {{- "Reasoning: " + reasoning_effort + "
271
+
272
+ " }}
273
+ {%- if builtin_tools %}
274
+ {{- "# Tools
275
+
276
+ " }}
277
+ {%- set available_builtin_tools = namespace(browser=false, python=false) %}
278
+ {%- for tool in builtin_tools %}
279
+ {%- if tool == "browser" %}
280
+ {%- set available_builtin_tools.browser = true %}
281
+ {%- elif tool == "python" %}
282
+ {%- set available_builtin_tools.python = true %}
283
+ {%- endif %}
284
+ {%- endfor %}
285
+ {{- render_builtin_tools(available_builtin_tools.browser, available_builtin_tools.python) }}
286
+ {%- endif -%}
287
+ {{- "# Valid channels: analysis, commentary, final. Channel must be included for every message." }}
288
+ {%- if tools -%}
289
+ {{- "
290
+ Calls to these tools must go to the commentary channel: 'functions'." }}
291
+ {%- endif -%}
292
+ {%- endmacro -%}
293
+
294
+ {#- Main Template Logic ================================================= #}
295
+ {#- Set defaults #}
296
+
297
+ {#- Render system message #}
298
+ {{- "<|start|>system<|message|>" }}
299
+ {{- build_system_message() }}
300
+ {{- "<|end|>" }}
301
+
302
+ {#- Extract developer message #}
303
+ {%- if messages[0].role == "developer" or messages[0].role == "system" %}
304
+ {%- set developer_message = messages[0].content %}
305
+ {%- set loop_messages = messages[1:] %}
306
+ {%- else %}
307
+ {%- set developer_message = "" %}
308
+ {%- set loop_messages = messages %}
309
+ {%- endif %}
310
+
311
+ {#- Render developer message #}
312
+ {%- if developer_message or tools %}
313
+ {{- "<|start|>developer<|message|>" }}
314
+ {%- if developer_message %}
315
+ {{- "# Instructions
316
+
317
+ " }}
318
+ {{- developer_message }}
319
+ {%- endif %}
320
+ {%- if tools -%}
321
+ {{- "
322
+
323
+ " }}
324
+ {{- "# Tools
325
+
326
+ " }}
327
+ {{- render_tool_namespace("functions", tools) }}
328
+ {%- endif -%}
329
+ {{- "<|end|>" }}
330
+ {%- endif %}
331
+
332
+ {#- Render messages #}
333
+ {%- set last_tool_call = namespace(name=none) %}
334
+ {%- for message in loop_messages -%}
335
+ {#- At this point only assistant/user/tool messages should remain #}
336
+ {%- if message.role == 'assistant' -%}
337
+ {#- Checks to ensure the messages are being passed in the format we expect #}
338
+ {%- if "content" in message %}
339
+ {%- if "<|channel|>analysis<|message|>" in message.content or "<|channel|>final<|message|>" in message.content %}
340
+ {{- raise_exception("You have passed a message containing <|channel|> tags in the content field. Instead of doing this, you should pass analysis messages (the string between '<|message|>' and '<|end|>') in the 'thinking' field, and final messages (the string between '<|message|>' and '<|end|>') in the 'content' field.") }}
341
+ {%- endif %}
342
+ {%- endif %}
343
+ {%- if "thinking" in message %}
344
+ {%- if "<|channel|>analysis<|message|>" in message.thinking or "<|channel|>final<|message|>" in message.thinking %}
345
+ {{- raise_exception("You have passed a message containing <|channel|> tags in the thinking field. Instead of doing this, you should pass analysis messages (the string between '<|message|>' and '<|end|>') in the 'thinking' field, and final messages (the string between '<|message|>' and '<|end|>') in the 'content' field.") }}
346
+ {%- endif %}
347
+ {%- endif %}
348
+ {%- if "tool_calls" in message %}
349
+ {#- We assume max 1 tool call per message, and so we infer the tool call name #}
350
+ {#- in "tool" messages from the most recent assistant tool call name #}
351
+ {%- set tool_call = message.tool_calls[0] %}
352
+ {%- if tool_call.function %}
353
+ {%- set tool_call = tool_call.function %}
354
+ {%- endif %}
355
+ {%- if message.content and message.thinking %}
356
+ {{- raise_exception("Cannot pass both content and thinking in an assistant message with tool calls! Put the analysis message in one or the other, but not both.") }}
357
+ {%- elif message.content %}
358
+ {{- "<|start|>assistant<|channel|>analysis<|message|>" + message.content + "<|end|>" }}
359
+ {%- elif message.thinking %}
360
+ {{- "<|start|>assistant<|channel|>analysis<|message|>" + message.thinking + "<|end|>" }}
361
+ {%- endif %}
362
+ {{- "<|start|>assistant to=" }}
363
+ {{- "functions." + tool_call.name + "<|channel|>commentary " }}
364
+ {{- (tool_call.content_type if tool_call.content_type is defined else "json") + "<|message|>" }}
365
+ {{- tool_call.arguments|tojson }}
366
+ {{- "<|call|>" }}
367
+ {%- set last_tool_call.name = tool_call.name %}
368
+ {%- elif loop.last and not add_generation_prompt %}
369
+ {#- Only render the CoT if the final turn is an assistant turn and add_generation_prompt is false #}
370
+ {#- This is a situation that should only occur in training, never in inference. #}
371
+ {%- if "thinking" in message %}
372
+ {{- "<|start|>assistant<|channel|>analysis<|message|>" + message.thinking + "<|end|>" }}
373
+ {%- endif %}
374
+ {#- <|return|> indicates the end of generation, but <|end|> does not #}
375
+ {#- <|return|> should never be an input to the model, but we include it as the final token #}
376
+ {#- when training, so the model learns to emit it. #}
377
+ {{- "<|start|>assistant<|channel|>final<|message|>" + message.content + "<|return|>" }}
378
+ {%- else %}
379
+ {#- CoT is dropped during all previous turns, so we never render it for inference #}
380
+ {{- "<|start|>assistant<|channel|>final<|message|>" + message.content + "<|end|>" }}
381
+ {%- set last_tool_call.name = none %}
382
+ {%- endif %}
383
+ {%- elif message.role == 'tool' -%}
384
+ {%- if last_tool_call.name is none %}
385
+ {{- raise_exception("Message has tool role, but there was no previous assistant message with a tool call!") }}
386
+ {%- endif %}
387
+ {{- "<|start|>functions." + last_tool_call.name }}
388
+ {{- " to=assistant<|channel|>commentary<|message|>" + message.content|tojson + "<|end|>" }}
389
+ {%- elif message.role == 'user' -%}
390
+ {{- "<|start|>user<|message|>" + message.content + "<|end|>" }}
391
+ {%- endif -%}
392
+ {%- endfor -%}
393
+
394
+ {#- Generation prompt #}
395
+ {%- if add_generation_prompt -%}
396
+ <|start|>assistant
397
+ {%- endif -%}
checkpoint-300/optimizer.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:850fc9caf277fa5ca7750db2a5e9c52b0d4d48ef6b7a6cc781dbfa9afb500be8
3
+ size 96746
checkpoint-300/rng_state.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1cde4ff942c9187cde9c7e789189e5b9e792025ecdf1ab53cad44f4f79990ad3
3
+ size 14244
checkpoint-300/scaler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:92873c8c89778fe11b8eeb338a181eefdf056f2f8096c36bf259c3fd791afb34
3
+ size 988
checkpoint-300/scheduler.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:8d6d0bfbfa66de948314c68639369ead3b124f435ab14172d1445517028ef8f2
3
+ size 2080
checkpoint-300/special_tokens_map.json ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "<|startoftext|>",
4
+ "lstrip": false,
5
+ "normalized": false,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "eos_token": {
10
+ "content": "<|return|>",
11
+ "lstrip": false,
12
+ "normalized": false,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "pad_token": {
17
+ "content": "<|endoftext|>",
18
+ "lstrip": false,
19
+ "normalized": false,
20
+ "rstrip": false,
21
+ "single_word": false
22
+ }
23
+ }
checkpoint-300/tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0614fe83cadab421296e664e1f48f4261fa8fef6e03e63bb75c20f38e37d07d3
3
+ size 27868174
checkpoint-300/tokenizer_config.json ADDED
@@ -0,0 +1,183 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "added_tokens_decoder": {
3
+ "199998": {
4
+ "content": "<|startoftext|>",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false,
9
+ "special": true
10
+ },
11
+ "199999": {
12
+ "content": "<|endoftext|>",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false,
17
+ "special": true
18
+ },
19
+ "200000": {
20
+ "content": "<|reserved_200000|>",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false,
25
+ "special": true
26
+ },
27
+ "200001": {
28
+ "content": "<|reserved_200001|>",
29
+ "lstrip": false,
30
+ "normalized": false,
31
+ "rstrip": false,
32
+ "single_word": false,
33
+ "special": true
34
+ },
35
+ "200002": {
36
+ "content": "<|return|>",
37
+ "lstrip": false,
38
+ "normalized": false,
39
+ "rstrip": false,
40
+ "single_word": false,
41
+ "special": true
42
+ },
43
+ "200003": {
44
+ "content": "<|constrain|>",
45
+ "lstrip": false,
46
+ "normalized": false,
47
+ "rstrip": false,
48
+ "single_word": false,
49
+ "special": true
50
+ },
51
+ "200004": {
52
+ "content": "<|reserved_200004|>",
53
+ "lstrip": false,
54
+ "normalized": false,
55
+ "rstrip": false,
56
+ "single_word": false,
57
+ "special": true
58
+ },
59
+ "200005": {
60
+ "content": "<|channel|>",
61
+ "lstrip": false,
62
+ "normalized": false,
63
+ "rstrip": false,
64
+ "single_word": false,
65
+ "special": true
66
+ },
67
+ "200006": {
68
+ "content": "<|start|>",
69
+ "lstrip": false,
70
+ "normalized": false,
71
+ "rstrip": false,
72
+ "single_word": false,
73
+ "special": true
74
+ },
75
+ "200007": {
76
+ "content": "<|end|>",
77
+ "lstrip": false,
78
+ "normalized": false,
79
+ "rstrip": false,
80
+ "single_word": false,
81
+ "special": true
82
+ },
83
+ "200008": {
84
+ "content": "<|message|>",
85
+ "lstrip": false,
86
+ "normalized": false,
87
+ "rstrip": false,
88
+ "single_word": false,
89
+ "special": true
90
+ },
91
+ "200009": {
92
+ "content": "<|reserved_200009|>",
93
+ "lstrip": false,
94
+ "normalized": false,
95
+ "rstrip": false,
96
+ "single_word": false,
97
+ "special": true
98
+ },
99
+ "200010": {
100
+ "content": "<|reserved_200010|>",
101
+ "lstrip": false,
102
+ "normalized": false,
103
+ "rstrip": false,
104
+ "single_word": false,
105
+ "special": true
106
+ },
107
+ "200011": {
108
+ "content": "<|reserved_200011|>",
109
+ "lstrip": false,
110
+ "normalized": false,
111
+ "rstrip": false,
112
+ "single_word": false,
113
+ "special": true
114
+ },
115
+ "200012": {
116
+ "content": "<|call|>",
117
+ "lstrip": false,
118
+ "normalized": false,
119
+ "rstrip": false,
120
+ "single_word": false,
121
+ "special": true
122
+ },
123
+ "200013": {
124
+ "content": "<|reserved_200013|>",
125
+ "lstrip": false,
126
+ "normalized": false,
127
+ "rstrip": false,
128
+ "single_word": false,
129
+ "special": true
130
+ },
131
+ "200014": {
132
+ "content": "<|reserved_200014|>",
133
+ "lstrip": false,
134
+ "normalized": false,
135
+ "rstrip": false,
136
+ "single_word": false,
137
+ "special": true
138
+ },
139
+ "200015": {
140
+ "content": "<|reserved_200015|>",
141
+ "lstrip": false,
142
+ "normalized": false,
143
+ "rstrip": false,
144
+ "single_word": false,
145
+ "special": true
146
+ },
147
+ "200016": {
148
+ "content": "<|reserved_200016|>",
149
+ "lstrip": false,
150
+ "normalized": false,
151
+ "rstrip": false,
152
+ "single_word": false,
153
+ "special": true
154
+ },
155
+ "200017": {
156
+ "content": "<|reserved_200017|>",
157
+ "lstrip": false,
158
+ "normalized": false,
159
+ "rstrip": false,
160
+ "single_word": false,
161
+ "special": true
162
+ },
163
+ "200018": {
164
+ "content": "<|endofprompt|>",
165
+ "lstrip": false,
166
+ "normalized": false,
167
+ "rstrip": false,
168
+ "single_word": false,
169
+ "special": true
170
+ }
171
+ },
172
+ "bos_token": "<|startoftext|>",
173
+ "clean_up_tokenization_spaces": false,
174
+ "eos_token": "<|return|>",
175
+ "extra_special_tokens": {},
176
+ "model_input_names": [
177
+ "input_ids",
178
+ "attention_mask"
179
+ ],
180
+ "model_max_length": 1000000000000000019884624838656,
181
+ "pad_token": "<|endoftext|>",
182
+ "tokenizer_class": "PreTrainedTokenizerFast"
183
+ }
checkpoint-300/trainer_state.json ADDED
@@ -0,0 +1,334 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "best_global_step": null,
3
+ "best_metric": null,
4
+ "best_model_checkpoint": null,
5
+ "epoch": 2.633663366336634,
6
+ "eval_steps": 500,
7
+ "global_step": 300,
8
+ "is_hyper_param_search": false,
9
+ "is_local_process_zero": true,
10
+ "is_world_process_zero": true,
11
+ "log_history": [
12
+ {
13
+ "epoch": 0.08800880088008801,
14
+ "grad_norm": 0.008909749798476696,
15
+ "learning_rate": 8.18181797512807e-05,
16
+ "loss": 12.2157,
17
+ "memory/device_mem_reserved(gib)": 2.73,
18
+ "memory/max_mem_active(gib)": 2.34,
19
+ "memory/max_mem_allocated(gib)": 2.34,
20
+ "step": 10
21
+ },
22
+ {
23
+ "epoch": 0.17601760176017603,
24
+ "grad_norm": 0.007974537089467049,
25
+ "learning_rate": 9.999999747378752e-05,
26
+ "loss": 12.2154,
27
+ "memory/device_mem_reserved(gib)": 2.73,
28
+ "memory/max_mem_active(gib)": 2.34,
29
+ "memory/max_mem_allocated(gib)": 2.34,
30
+ "step": 20
31
+ },
32
+ {
33
+ "epoch": 0.264026402640264,
34
+ "grad_norm": 0.010009167715907097,
35
+ "learning_rate": 9.999999747378752e-05,
36
+ "loss": 12.2152,
37
+ "memory/device_mem_reserved(gib)": 2.73,
38
+ "memory/max_mem_active(gib)": 2.34,
39
+ "memory/max_mem_allocated(gib)": 2.34,
40
+ "step": 30
41
+ },
42
+ {
43
+ "epoch": 0.35203520352035206,
44
+ "grad_norm": 0.010353786870837212,
45
+ "learning_rate": 9.999999747378752e-05,
46
+ "loss": 12.2155,
47
+ "memory/device_mem_reserved(gib)": 2.73,
48
+ "memory/max_mem_active(gib)": 2.34,
49
+ "memory/max_mem_allocated(gib)": 2.34,
50
+ "step": 40
51
+ },
52
+ {
53
+ "epoch": 0.44004400440044006,
54
+ "grad_norm": 0.01487517450004816,
55
+ "learning_rate": 9.999999747378752e-05,
56
+ "loss": 12.2151,
57
+ "memory/device_mem_reserved(gib)": 2.73,
58
+ "memory/max_mem_active(gib)": 2.34,
59
+ "memory/max_mem_allocated(gib)": 2.34,
60
+ "step": 50
61
+ },
62
+ {
63
+ "epoch": 0.528052805280528,
64
+ "grad_norm": 0.011968914419412613,
65
+ "learning_rate": 9.999999747378752e-05,
66
+ "loss": 12.215,
67
+ "memory/device_mem_reserved(gib)": 2.73,
68
+ "memory/max_mem_active(gib)": 2.34,
69
+ "memory/max_mem_allocated(gib)": 2.34,
70
+ "step": 60
71
+ },
72
+ {
73
+ "epoch": 0.6160616061606161,
74
+ "grad_norm": 0.015747424215078354,
75
+ "learning_rate": 9.999999747378752e-05,
76
+ "loss": 12.2146,
77
+ "memory/device_mem_reserved(gib)": 2.73,
78
+ "memory/max_mem_active(gib)": 2.34,
79
+ "memory/max_mem_allocated(gib)": 2.34,
80
+ "step": 70
81
+ },
82
+ {
83
+ "epoch": 0.7040704070407041,
84
+ "grad_norm": 0.01620076224207878,
85
+ "learning_rate": 9.999999747378752e-05,
86
+ "loss": 12.2132,
87
+ "memory/device_mem_reserved(gib)": 2.73,
88
+ "memory/max_mem_active(gib)": 2.34,
89
+ "memory/max_mem_allocated(gib)": 2.34,
90
+ "step": 80
91
+ },
92
+ {
93
+ "epoch": 0.7920792079207921,
94
+ "grad_norm": 0.016200672835111618,
95
+ "learning_rate": 9.999999747378752e-05,
96
+ "loss": 12.2131,
97
+ "memory/device_mem_reserved(gib)": 2.73,
98
+ "memory/max_mem_active(gib)": 2.34,
99
+ "memory/max_mem_allocated(gib)": 2.34,
100
+ "step": 90
101
+ },
102
+ {
103
+ "epoch": 0.8800880088008801,
104
+ "grad_norm": 0.01671593263745308,
105
+ "learning_rate": 9.999999747378752e-05,
106
+ "loss": 12.2126,
107
+ "memory/device_mem_reserved(gib)": 2.73,
108
+ "memory/max_mem_active(gib)": 2.34,
109
+ "memory/max_mem_allocated(gib)": 2.34,
110
+ "step": 100
111
+ },
112
+ {
113
+ "epoch": 0.9680968096809681,
114
+ "grad_norm": 0.019221968948841095,
115
+ "learning_rate": 9.999999747378752e-05,
116
+ "loss": 12.2124,
117
+ "memory/device_mem_reserved(gib)": 2.73,
118
+ "memory/max_mem_active(gib)": 2.34,
119
+ "memory/max_mem_allocated(gib)": 2.34,
120
+ "step": 110
121
+ },
122
+ {
123
+ "epoch": 1.0528052805280528,
124
+ "grad_norm": 0.02141665853559971,
125
+ "learning_rate": 9.999999747378752e-05,
126
+ "loss": 12.2117,
127
+ "memory/device_mem_reserved(gib)": 2.73,
128
+ "memory/max_mem_active(gib)": 2.34,
129
+ "memory/max_mem_allocated(gib)": 2.34,
130
+ "step": 120
131
+ },
132
+ {
133
+ "epoch": 1.140814081408141,
134
+ "grad_norm": 0.015679990872740746,
135
+ "learning_rate": 9.999999747378752e-05,
136
+ "loss": 12.2111,
137
+ "memory/device_mem_reserved(gib)": 2.74,
138
+ "memory/max_mem_active(gib)": 2.34,
139
+ "memory/max_mem_allocated(gib)": 2.34,
140
+ "step": 130
141
+ },
142
+ {
143
+ "epoch": 1.2288228822882288,
144
+ "grad_norm": 0.022765284404158592,
145
+ "learning_rate": 9.999999747378752e-05,
146
+ "loss": 12.2093,
147
+ "memory/device_mem_reserved(gib)": 2.74,
148
+ "memory/max_mem_active(gib)": 2.34,
149
+ "memory/max_mem_allocated(gib)": 2.34,
150
+ "step": 140
151
+ },
152
+ {
153
+ "epoch": 1.316831683168317,
154
+ "grad_norm": 0.01766524650156498,
155
+ "learning_rate": 9.999999747378752e-05,
156
+ "loss": 12.2088,
157
+ "memory/device_mem_reserved(gib)": 2.74,
158
+ "memory/max_mem_active(gib)": 2.34,
159
+ "memory/max_mem_allocated(gib)": 2.34,
160
+ "step": 150
161
+ },
162
+ {
163
+ "epoch": 1.4048404840484048,
164
+ "grad_norm": 0.02177940309047699,
165
+ "learning_rate": 9.999999747378752e-05,
166
+ "loss": 12.2081,
167
+ "memory/device_mem_reserved(gib)": 2.74,
168
+ "memory/max_mem_active(gib)": 2.34,
169
+ "memory/max_mem_allocated(gib)": 2.34,
170
+ "step": 160
171
+ },
172
+ {
173
+ "epoch": 1.492849284928493,
174
+ "grad_norm": 0.018786178901791573,
175
+ "learning_rate": 9.999999747378752e-05,
176
+ "loss": 12.207,
177
+ "memory/device_mem_reserved(gib)": 2.74,
178
+ "memory/max_mem_active(gib)": 2.34,
179
+ "memory/max_mem_allocated(gib)": 2.34,
180
+ "step": 170
181
+ },
182
+ {
183
+ "epoch": 1.5808580858085808,
184
+ "grad_norm": 0.0304368045181036,
185
+ "learning_rate": 9.999999747378752e-05,
186
+ "loss": 12.2067,
187
+ "memory/device_mem_reserved(gib)": 2.74,
188
+ "memory/max_mem_active(gib)": 2.34,
189
+ "memory/max_mem_allocated(gib)": 2.34,
190
+ "step": 180
191
+ },
192
+ {
193
+ "epoch": 1.668866886688669,
194
+ "grad_norm": 0.02242487668991089,
195
+ "learning_rate": 9.999999747378752e-05,
196
+ "loss": 12.2058,
197
+ "memory/device_mem_reserved(gib)": 2.74,
198
+ "memory/max_mem_active(gib)": 2.34,
199
+ "memory/max_mem_allocated(gib)": 2.34,
200
+ "step": 190
201
+ },
202
+ {
203
+ "epoch": 1.756875687568757,
204
+ "grad_norm": 0.024080732837319374,
205
+ "learning_rate": 9.999999747378752e-05,
206
+ "loss": 12.2046,
207
+ "memory/device_mem_reserved(gib)": 2.74,
208
+ "memory/max_mem_active(gib)": 2.34,
209
+ "memory/max_mem_allocated(gib)": 2.34,
210
+ "step": 200
211
+ },
212
+ {
213
+ "epoch": 1.844884488448845,
214
+ "grad_norm": 0.020711150020360947,
215
+ "learning_rate": 9.999999747378752e-05,
216
+ "loss": 12.2049,
217
+ "memory/device_mem_reserved(gib)": 2.74,
218
+ "memory/max_mem_active(gib)": 2.34,
219
+ "memory/max_mem_allocated(gib)": 2.34,
220
+ "step": 210
221
+ },
222
+ {
223
+ "epoch": 1.9328932893289328,
224
+ "grad_norm": 0.017322849482297897,
225
+ "learning_rate": 9.999999747378752e-05,
226
+ "loss": 12.204,
227
+ "memory/device_mem_reserved(gib)": 2.74,
228
+ "memory/max_mem_active(gib)": 2.34,
229
+ "memory/max_mem_allocated(gib)": 2.34,
230
+ "step": 220
231
+ },
232
+ {
233
+ "epoch": 2.0176017601760177,
234
+ "grad_norm": 0.022822650149464607,
235
+ "learning_rate": 9.999999747378752e-05,
236
+ "loss": 12.2028,
237
+ "memory/device_mem_reserved(gib)": 2.74,
238
+ "memory/max_mem_active(gib)": 2.34,
239
+ "memory/max_mem_allocated(gib)": 2.34,
240
+ "step": 230
241
+ },
242
+ {
243
+ "epoch": 2.1056105610561056,
244
+ "grad_norm": 0.01951930858194828,
245
+ "learning_rate": 9.999999747378752e-05,
246
+ "loss": 12.2027,
247
+ "memory/device_mem_reserved(gib)": 2.74,
248
+ "memory/max_mem_active(gib)": 2.34,
249
+ "memory/max_mem_allocated(gib)": 2.34,
250
+ "step": 240
251
+ },
252
+ {
253
+ "epoch": 2.1936193619361934,
254
+ "grad_norm": 0.01794363372027874,
255
+ "learning_rate": 9.999999747378752e-05,
256
+ "loss": 12.2025,
257
+ "memory/device_mem_reserved(gib)": 2.74,
258
+ "memory/max_mem_active(gib)": 2.34,
259
+ "memory/max_mem_allocated(gib)": 2.34,
260
+ "step": 250
261
+ },
262
+ {
263
+ "epoch": 2.281628162816282,
264
+ "grad_norm": 0.014412708580493927,
265
+ "learning_rate": 9.999999747378752e-05,
266
+ "loss": 12.2021,
267
+ "memory/device_mem_reserved(gib)": 2.74,
268
+ "memory/max_mem_active(gib)": 2.34,
269
+ "memory/max_mem_allocated(gib)": 2.34,
270
+ "step": 260
271
+ },
272
+ {
273
+ "epoch": 2.3696369636963697,
274
+ "grad_norm": 0.012650610879063606,
275
+ "learning_rate": 9.999999747378752e-05,
276
+ "loss": 12.2017,
277
+ "memory/device_mem_reserved(gib)": 2.74,
278
+ "memory/max_mem_active(gib)": 2.34,
279
+ "memory/max_mem_allocated(gib)": 2.34,
280
+ "step": 270
281
+ },
282
+ {
283
+ "epoch": 2.4576457645764576,
284
+ "grad_norm": 0.015798937529325485,
285
+ "learning_rate": 9.999999747378752e-05,
286
+ "loss": 12.2013,
287
+ "memory/device_mem_reserved(gib)": 2.74,
288
+ "memory/max_mem_active(gib)": 2.34,
289
+ "memory/max_mem_allocated(gib)": 2.34,
290
+ "step": 280
291
+ },
292
+ {
293
+ "epoch": 2.5456545654565454,
294
+ "grad_norm": 0.01763707585632801,
295
+ "learning_rate": 9.999999747378752e-05,
296
+ "loss": 12.2013,
297
+ "memory/device_mem_reserved(gib)": 2.74,
298
+ "memory/max_mem_active(gib)": 2.34,
299
+ "memory/max_mem_allocated(gib)": 2.34,
300
+ "step": 290
301
+ },
302
+ {
303
+ "epoch": 2.633663366336634,
304
+ "grad_norm": 0.016535457223653793,
305
+ "learning_rate": 9.999999747378752e-05,
306
+ "loss": 12.2006,
307
+ "memory/device_mem_reserved(gib)": 2.74,
308
+ "memory/max_mem_active(gib)": 2.34,
309
+ "memory/max_mem_allocated(gib)": 2.34,
310
+ "step": 300
311
+ }
312
+ ],
313
+ "logging_steps": 10,
314
+ "max_steps": 300,
315
+ "num_input_tokens_seen": 0,
316
+ "num_train_epochs": 3,
317
+ "save_steps": 150,
318
+ "stateful_callbacks": {
319
+ "TrainerControl": {
320
+ "args": {
321
+ "should_epoch_stop": false,
322
+ "should_evaluate": false,
323
+ "should_log": false,
324
+ "should_save": true,
325
+ "should_training_stop": true
326
+ },
327
+ "attributes": {}
328
+ }
329
+ },
330
+ "total_flos": 6141434826240.0,
331
+ "train_batch_size": 4,
332
+ "trial_name": null,
333
+ "trial_params": null
334
+ }
checkpoint-300/training_args.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a33123adb3cce6429a0192902873aaa084cd85cfcdd34c6b726552ee2bb1fa77
3
+ size 6776
config.json ADDED
@@ -0,0 +1,46 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "GptOssForCausalLM"
4
+ ],
5
+ "attention_bias": true,
6
+ "attention_dropout": 0.0,
7
+ "eos_token_id": 200002,
8
+ "experts_per_token": 4,
9
+ "head_dim": 32,
10
+ "hidden_act": "silu",
11
+ "hidden_size": 32,
12
+ "initial_context_length": 4096,
13
+ "initializer_range": 0.02,
14
+ "intermediate_size": 64,
15
+ "layer_types": [
16
+ "sliding_attention",
17
+ "full_attention"
18
+ ],
19
+ "max_position_embeddings": 131072,
20
+ "model_type": "gpt_oss",
21
+ "num_attention_heads": 2,
22
+ "num_experts_per_tok": 4,
23
+ "num_hidden_layers": 2,
24
+ "num_key_value_heads": 1,
25
+ "num_local_experts": 32,
26
+ "output_router_logits": false,
27
+ "pad_token_id": 199999,
28
+ "rms_norm_eps": 1e-05,
29
+ "rope_scaling": {
30
+ "beta_fast": 32.0,
31
+ "beta_slow": 1.0,
32
+ "factor": 32.0,
33
+ "original_max_position_embeddings": 4096,
34
+ "rope_type": "yarn",
35
+ "truncate": false
36
+ },
37
+ "rope_theta": 150000,
38
+ "router_aux_loss_coef": 0.9,
39
+ "sliding_window": 128,
40
+ "swiglu_limit": 7.0,
41
+ "tie_word_embeddings": true,
42
+ "torch_dtype": "float16",
43
+ "transformers_version": "4.55.2",
44
+ "use_cache": false,
45
+ "vocab_size": 201088
46
+ }
special_tokens_map.json ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "bos_token": {
3
+ "content": "<|startoftext|>",
4
+ "lstrip": false,
5
+ "normalized": false,
6
+ "rstrip": false,
7
+ "single_word": false
8
+ },
9
+ "eos_token": {
10
+ "content": "<|return|>",
11
+ "lstrip": false,
12
+ "normalized": false,
13
+ "rstrip": false,
14
+ "single_word": false
15
+ },
16
+ "pad_token": {
17
+ "content": "<|endoftext|>",
18
+ "lstrip": false,
19
+ "normalized": false,
20
+ "rstrip": false,
21
+ "single_word": false
22
+ }
23
+ }
tokenizer.json ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0614fe83cadab421296e664e1f48f4261fa8fef6e03e63bb75c20f38e37d07d3
3
+ size 27868174
tokenizer_config.json ADDED
@@ -0,0 +1,183 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "added_tokens_decoder": {
3
+ "199998": {
4
+ "content": "<|startoftext|>",
5
+ "lstrip": false,
6
+ "normalized": false,
7
+ "rstrip": false,
8
+ "single_word": false,
9
+ "special": true
10
+ },
11
+ "199999": {
12
+ "content": "<|endoftext|>",
13
+ "lstrip": false,
14
+ "normalized": false,
15
+ "rstrip": false,
16
+ "single_word": false,
17
+ "special": true
18
+ },
19
+ "200000": {
20
+ "content": "<|reserved_200000|>",
21
+ "lstrip": false,
22
+ "normalized": false,
23
+ "rstrip": false,
24
+ "single_word": false,
25
+ "special": true
26
+ },
27
+ "200001": {
28
+ "content": "<|reserved_200001|>",
29
+ "lstrip": false,
30
+ "normalized": false,
31
+ "rstrip": false,
32
+ "single_word": false,
33
+ "special": true
34
+ },
35
+ "200002": {
36
+ "content": "<|return|>",
37
+ "lstrip": false,
38
+ "normalized": false,
39
+ "rstrip": false,
40
+ "single_word": false,
41
+ "special": true
42
+ },
43
+ "200003": {
44
+ "content": "<|constrain|>",
45
+ "lstrip": false,
46
+ "normalized": false,
47
+ "rstrip": false,
48
+ "single_word": false,
49
+ "special": true
50
+ },
51
+ "200004": {
52
+ "content": "<|reserved_200004|>",
53
+ "lstrip": false,
54
+ "normalized": false,
55
+ "rstrip": false,
56
+ "single_word": false,
57
+ "special": true
58
+ },
59
+ "200005": {
60
+ "content": "<|channel|>",
61
+ "lstrip": false,
62
+ "normalized": false,
63
+ "rstrip": false,
64
+ "single_word": false,
65
+ "special": true
66
+ },
67
+ "200006": {
68
+ "content": "<|start|>",
69
+ "lstrip": false,
70
+ "normalized": false,
71
+ "rstrip": false,
72
+ "single_word": false,
73
+ "special": true
74
+ },
75
+ "200007": {
76
+ "content": "<|end|>",
77
+ "lstrip": false,
78
+ "normalized": false,
79
+ "rstrip": false,
80
+ "single_word": false,
81
+ "special": true
82
+ },
83
+ "200008": {
84
+ "content": "<|message|>",
85
+ "lstrip": false,
86
+ "normalized": false,
87
+ "rstrip": false,
88
+ "single_word": false,
89
+ "special": true
90
+ },
91
+ "200009": {
92
+ "content": "<|reserved_200009|>",
93
+ "lstrip": false,
94
+ "normalized": false,
95
+ "rstrip": false,
96
+ "single_word": false,
97
+ "special": true
98
+ },
99
+ "200010": {
100
+ "content": "<|reserved_200010|>",
101
+ "lstrip": false,
102
+ "normalized": false,
103
+ "rstrip": false,
104
+ "single_word": false,
105
+ "special": true
106
+ },
107
+ "200011": {
108
+ "content": "<|reserved_200011|>",
109
+ "lstrip": false,
110
+ "normalized": false,
111
+ "rstrip": false,
112
+ "single_word": false,
113
+ "special": true
114
+ },
115
+ "200012": {
116
+ "content": "<|call|>",
117
+ "lstrip": false,
118
+ "normalized": false,
119
+ "rstrip": false,
120
+ "single_word": false,
121
+ "special": true
122
+ },
123
+ "200013": {
124
+ "content": "<|reserved_200013|>",
125
+ "lstrip": false,
126
+ "normalized": false,
127
+ "rstrip": false,
128
+ "single_word": false,
129
+ "special": true
130
+ },
131
+ "200014": {
132
+ "content": "<|reserved_200014|>",
133
+ "lstrip": false,
134
+ "normalized": false,
135
+ "rstrip": false,
136
+ "single_word": false,
137
+ "special": true
138
+ },
139
+ "200015": {
140
+ "content": "<|reserved_200015|>",
141
+ "lstrip": false,
142
+ "normalized": false,
143
+ "rstrip": false,
144
+ "single_word": false,
145
+ "special": true
146
+ },
147
+ "200016": {
148
+ "content": "<|reserved_200016|>",
149
+ "lstrip": false,
150
+ "normalized": false,
151
+ "rstrip": false,
152
+ "single_word": false,
153
+ "special": true
154
+ },
155
+ "200017": {
156
+ "content": "<|reserved_200017|>",
157
+ "lstrip": false,
158
+ "normalized": false,
159
+ "rstrip": false,
160
+ "single_word": false,
161
+ "special": true
162
+ },
163
+ "200018": {
164
+ "content": "<|endofprompt|>",
165
+ "lstrip": false,
166
+ "normalized": false,
167
+ "rstrip": false,
168
+ "single_word": false,
169
+ "special": true
170
+ }
171
+ },
172
+ "bos_token": "<|startoftext|>",
173
+ "clean_up_tokenization_spaces": false,
174
+ "eos_token": "<|return|>",
175
+ "extra_special_tokens": {},
176
+ "model_input_names": [
177
+ "input_ids",
178
+ "attention_mask"
179
+ ],
180
+ "model_max_length": 1000000000000000019884624838656,
181
+ "pad_token": "<|endoftext|>",
182
+ "tokenizer_class": "PreTrainedTokenizerFast"
183
+ }