devops-thiago commited on
Commit
882f550
·
verified ·
1 Parent(s): 8e7b1d3

Publish merged standalone ClassOne model

Browse files
chat_template.jinja CHANGED
@@ -1,386 +1,386 @@
1
- {#
2
- Template: Google Gemma 4 Canonical Chat Template
3
- Author: Google Gemma Engineering Team
4
- Published: 2026-07-09
5
- Context: Fixed tool-calling loops, turn closures, and thinking content-ordering.
6
- #}
7
- {%- macro format_parameters(properties, required, filter_keys=false) -%}
8
- {%- set standard_keys = ['description', 'type', 'properties', 'required', 'nullable'] -%}
9
- {%- set ns = namespace(found_first=false) -%}
10
- {%- for key, value in properties | dictsort -%}
11
- {%- set add_comma = false -%}
12
- {%- if not filter_keys or key not in standard_keys -%}
13
- {%- if ns.found_first %},{% endif -%}
14
- {%- set ns.found_first = true -%}
15
- {{ key }}:{
16
- {%- if value['description'] -%}
17
- description:<|"|>{{ value['description'] }}<|"|>
18
- {%- set add_comma = true -%}
19
- {%- endif -%}
20
- {%- if value['type'] | upper == 'STRING' -%}
21
- {%- if value['enum'] -%}
22
- {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
23
- enum:{{ format_argument(value['enum']) }}
24
- {%- endif -%}
25
- {%- elif value['type'] | upper == 'ARRAY' -%}
26
- {%- if value['items'] is mapping and value['items'] -%}
27
- {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
28
- items:{
29
- {%- set ns_items = namespace(found_first=false) -%}
30
- {%- for item_key, item_value in value['items'] | dictsort -%}
31
- {%- if item_value is not none -%}
32
- {%- if ns_items.found_first %},{% endif -%}
33
- {%- set ns_items.found_first = true -%}
34
- {%- if item_key == 'properties' -%}
35
- properties:{
36
- {%- if item_value is mapping -%}
37
- {{- format_parameters(item_value, value['items']['required'] | default([])) -}}
38
- {%- endif -%}
39
- }
40
- {%- elif item_key == 'required' -%}
41
- required:[
42
- {%- for req_item in item_value -%}
43
- <|"|>{{- req_item -}}<|"|>
44
- {%- if not loop.last %},{% endif -%}
45
- {%- endfor -%}
46
- ]
47
- {%- elif item_key == 'type' -%}
48
- {%- if item_value is string -%}
49
- type:{{ format_argument(item_value | upper) }}
50
- {%- else -%}
51
- type:{{ format_argument(item_value | map('upper') | list) }}
52
- {%- endif -%}
53
- {%- else -%}
54
- {{ item_key }}:{{ format_argument(item_value) }}
55
- {%- endif -%}
56
- {%- endif -%}
57
- {%- endfor -%}
58
- }
59
- {%- endif -%}
60
- {%- endif -%}
61
- {%- if value['nullable'] %}
62
- {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
63
- nullable:true
64
- {%- endif -%}
65
- {%- if value['type'] | upper == 'OBJECT' -%}
66
- {%- if value['properties'] is defined and value['properties'] is mapping -%}
67
- {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
68
- properties:{
69
- {{- format_parameters(value['properties'], value['required'] | default([])) -}}
70
- }
71
- {%- elif value is mapping -%}
72
- {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
73
- properties:{
74
- {{- format_parameters(value, value['required'] | default([]), filter_keys=true) -}}
75
- }
76
- {%- endif -%}
77
- {%- if value['required'] -%}
78
- {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
79
- required:[
80
- {%- for item in value['required'] | default([]) -%}
81
- <|"|>{{- item -}}<|"|>
82
- {%- if not loop.last %},{% endif -%}
83
- {%- endfor -%}
84
- ]
85
- {%- endif -%}
86
- {%- endif -%}
87
- {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
88
- type:<|"|>{{ value['type'] | upper }}<|"|>}
89
- {%- endif -%}
90
- {%- endfor -%}
91
- {%- endmacro -%}
92
- {%- macro format_function_declaration(tool_data) -%}
93
- declaration:{{- tool_data['function']['name'] -}}{description:<|"|>{{- tool_data['function']['description'] -}}<|"|>
94
- {%- set params = tool_data['function']['parameters'] -%}
95
- {%- if params -%}
96
- ,parameters:{
97
- {%- if params['properties'] -%}
98
- properties:{ {{- format_parameters(params['properties'], params['required']) -}} },
99
- {%- endif -%}
100
- {%- if params['required'] -%}
101
- required:[
102
- {%- for item in params['required'] -%}
103
- <|"|>{{- item -}}<|"|>
104
- {{- ',' if not loop.last -}}
105
- {%- endfor -%}
106
- ],
107
- {%- endif -%}
108
- {%- if params['type'] -%}
109
- type:<|"|>{{- params['type'] | upper -}}<|"|>}
110
- {%- endif -%}
111
- {%- endif -%}
112
- {%- if 'response' in tool_data['function'] -%}
113
- {%- set response_declaration = tool_data['function']['response'] -%}
114
- ,response:{
115
- {%- if response_declaration['description'] -%}
116
- description:<|"|>{{- response_declaration['description'] -}}<|"|>,
117
- {%- endif -%}
118
- {%- if response_declaration['type'] | upper == 'OBJECT' -%}
119
- type:<|"|>{{- response_declaration['type'] | upper -}}<|"|>}
120
- {%- endif -%}
121
- {%- endif -%}
122
- }
123
- {%- endmacro -%}
124
- {%- macro format_argument(argument, escape_keys=True) -%}
125
- {%- if argument is none -%}
126
- {{- 'null' -}}
127
- {%- elif argument is string -%}
128
- {{- '<|"|>' + argument + '<|"|>' -}}
129
- {%- elif argument is boolean -%}
130
- {{- 'true' if argument else 'false' -}}
131
- {%- elif argument is mapping -%}
132
- {{- '{' -}}
133
- {%- set ns = namespace(found_first=false) -%}
134
- {%- for key, value in argument | dictsort -%}
135
- {%- if ns.found_first %},{% endif -%}
136
- {%- set ns.found_first = true -%}
137
- {%- if escape_keys -%}
138
- {{- '<|"|>' + key + '<|"|>' -}}
139
- {%- else -%}
140
- {{- key -}}
141
- {%- endif -%}
142
- :{{- format_argument(value, escape_keys=escape_keys) -}}
143
- {%- endfor -%}
144
- {{- '}' -}}
145
- {%- elif argument is sequence -%}
146
- {{- '[' -}}
147
- {%- for item in argument -%}
148
- {{- format_argument(item, escape_keys=escape_keys) -}}
149
- {%- if not loop.last %},{% endif -%}
150
- {%- endfor -%}
151
- {{- ']' -}}
152
- {%- else -%}
153
- {{- argument -}}
154
- {%- endif -%}
155
- {%- endmacro -%}
156
- {%- macro strip_thinking(text) -%}
157
- {%- set ns = namespace(result='') -%}
158
- {%- for part in text.split('<channel|>') -%}
159
- {%- if '<|channel>' in part -%}
160
- {%- set ns.result = ns.result + part.split('<|channel>')[0] -%}
161
- {%- else -%}
162
- {%- set ns.result = ns.result + part -%}
163
- {%- endif -%}
164
- {%- endfor -%}
165
- {{- ns.result | trim -}}
166
- {%- endmacro -%}
167
-
168
- {%- macro format_tool_response_block(tool_name, response) -%}
169
- {{- '<|tool_response>' -}}
170
- {%- if response is mapping -%}
171
- {{- 'response:' + tool_name + '{' -}}
172
- {%- for key, value in response | dictsort -%}
173
- {{- key -}}:{{- format_argument(value, escape_keys=False) -}}
174
- {%- if not loop.last %},{% endif -%}
175
- {%- endfor -%}
176
- {{- '}' -}}
177
- {%- else -%}
178
- {{- 'response:' + tool_name + '{value:' + format_argument(response, escape_keys=False) + '}' -}}
179
- {%- endif -%}
180
- {{- '<tool_response|>' -}}
181
- {%- endmacro -%}
182
-
183
- {#- ===== SETUP ===== -#}
184
- {%- set ns = namespace(prev_message_type=None, prev_non_tool_role=None) -%}
185
- {%- set loop_messages = messages -%}
186
- {%- set enable_thinking = enable_thinking | default(false) -%}
187
- {%- set preserve_thinking = preserve_thinking | default(false) -%}
188
- {{- bos_token -}}
189
- {#- Handle System/Tool Definitions Block -#}
190
- {%- if enable_thinking or tools or (messages and messages[0]['role'] in ['system', 'developer']) -%}
191
- {{- '<|turn>system\n' -}}
192
- {#- Inject Thinking token at the very top of the FIRST system turn -#}
193
- {%- if enable_thinking -%}
194
- {{- '<|think|>\n' -}}
195
- {%- set ns.prev_message_type = 'think' -%}
196
- {%- endif -%}
197
- {%- if messages and messages[0]['role'] in ['system', 'developer'] -%}
198
- {%- if messages[0]['content'] is string -%}
199
- {{- messages[0]['content'] | trim -}}
200
- {%- elif messages[0]['content'] is sequence -%}
201
- {%- for item in messages[0]['content'] -%}
202
- {{- item['text'] | trim + ' '-}}
203
- {%- endfor -%}
204
- {%- endif -%}
205
- {%- set loop_messages = messages[1:] -%}
206
- {%- endif -%}
207
- {%- if tools -%}
208
- {%- for tool in tools %}
209
- {{- '<|tool>' -}}
210
- {{- format_function_declaration(tool) | trim -}}
211
- {{- '<tool|>' -}}
212
- {%- endfor %}
213
- {%- set ns.prev_message_type = 'tool' -%}
214
- {%- endif -%}
215
- {{- '<turn|>\n' -}}
216
- {%- endif %}
217
-
218
- {#- Pre-scan: find last user message index for reasoning guard -#}
219
- {%- set ns_turn = namespace(last_user_idx=-1) -%}
220
- {%- for i in range(loop_messages | length) -%}
221
- {%- if loop_messages[i]['role'] == 'user' -%}
222
- {%- set ns_turn.last_user_idx = i -%}
223
- {%- endif -%}
224
- {%- endfor -%}
225
-
226
- {#- Loop through messages -#}
227
- {%- for message in loop_messages -%}
228
- {%- if message['role'] != 'tool' -%}
229
- {%- set ns.prev_message_type = None -%}
230
- {%- set role = 'model' if message['role'] == 'assistant' else message['role'] -%}
231
- {#- Detect continuation using tracked state — O(1) instead of O(n) backward scan -#}
232
- {%- set continue_same_model_turn = (role == 'model' and ns.prev_non_tool_role == 'assistant') -%}
233
- {%- if not continue_same_model_turn -%}
234
- {{- '<|turn>' + role + '\n' }}
235
- {%- endif -%}
236
-
237
- {#- Render reasoning/reasoning_content as thinking channel -#}
238
- {%- set thinking_text = message.get('reasoning') or message.get('reasoning_content') -%}
239
- {%- set thinking_gate = (loop.index0 > ns_turn.last_user_idx) or (preserve_thinking and message.get('tool_calls')) -%}
240
- {%- if thinking_text and thinking_gate -%}
241
- {{- '<|channel>thought\n' + thinking_text + '\n<channel|>' -}}
242
- {%- endif -%}
243
-
244
- {%- if message.get('tool_calls') -%}
245
- {%- for tool_call in message.get('tool_calls') -%}
246
- {%- set function = tool_call['function'] -%}
247
- {{- '<|tool_call>call:' + function['name'] + '{' -}}
248
- {%- if function['arguments'] is mapping -%}
249
- {%- set ns_args = namespace(found_first=false) -%}
250
- {%- for key, value in function['arguments'] | dictsort -%}
251
- {%- if ns_args.found_first %},{% endif -%}
252
- {%- set ns_args.found_first = true -%}
253
- {{- key -}}:{{- format_argument(value, escape_keys=False) -}}
254
- {%- endfor -%}
255
- {%- elif function['arguments'] is none -%}
256
- {%- else -%}
257
- {{- raise_exception(
258
- "chat_template: tool_calls[].function.arguments must be a "
259
- "JSON object (mapping), not a string. Deserialize arguments "
260
- "before passing to the template."
261
- ) -}}
262
- {%- endif -%}
263
- {{- '}<tool_call|>' -}}
264
- {%- endfor -%}
265
- {%- set ns.prev_message_type = 'tool_call' -%}
266
- {%- endif -%}
267
-
268
- {%- set ns_tr_out = namespace(flag=false) -%}
269
- {%- if message.get('tool_responses') -%}
270
- {#- Legacy: tool_responses embedded on the assistant message (Google/Gemma native) -#}
271
- {%- for tool_response in message.get('tool_responses') -%}
272
- {{- format_tool_response_block(tool_response['name'] | default('unknown', true), tool_response['response']) -}}
273
- {%- set ns_tr_out.flag = true -%}
274
- {%- set ns.prev_message_type = 'tool_response' -%}
275
- {%- endfor -%}
276
- {%- elif message.get('tool_calls') -%}
277
- {#- OpenAI Chat Completions: forward-scan consecutive role:tool messages -#}
278
- {%- set ns_tool_scan = namespace(stopped=false) -%}
279
- {%- for k in range(loop.index0 + 1, loop_messages | length) -%}
280
- {%- if ns_tool_scan.stopped -%}
281
- {%- elif loop_messages[k]['role'] != 'tool' -%}
282
- {%- set ns_tool_scan.stopped = true -%}
283
- {%- else -%}
284
- {%- set follow = loop_messages[k] -%}
285
- {#- Resolve tool_call_id to function name -#}
286
- {%- set ns_tname = namespace(name=follow.get('name') or 'unknown') -%}
287
- {%- for tc in message.get('tool_calls') -%}
288
- {%- if tc.get('id') == follow.get('tool_call_id') -%}
289
- {%- set ns_tname.name = tc['function']['name'] -%}
290
- {%- endif -%}
291
- {%- endfor -%}
292
- {#- Handle content as string or content-parts array -#}
293
- {%- set tool_body = follow.get('content') -%}
294
- {%- if tool_body is string -%}
295
- {{- format_tool_response_block(ns_tname.name, tool_body) -}}
296
- {%- elif tool_body is sequence and tool_body is not string -%}
297
- {%- set ns_txt = namespace(s='') -%}
298
- {%- for part in tool_body -%}
299
- {%- if part.get('type') == 'text' -%}
300
- {%- set ns_txt.s = ns_txt.s + (part.get('text') | default('')) -%}
301
- {%- endif -%}
302
- {%- endfor -%}
303
- {{- format_tool_response_block(ns_tname.name, ns_txt.s) -}}
304
- {%- for part in tool_body -%}
305
- {%- if part.get('type') in ['image', 'image_url'] -%}
306
- {{- '<|image|>' -}}
307
- {%- elif part.get('type') in ['audio', 'input_audio'] -%}
308
- {{- '<|audio|>' -}}
309
- {%- elif part.get('type') == 'video' -%}
310
- {{- '<|video|>' -}}
311
- {%- endif -%}
312
- {%- endfor -%}
313
- {%- else -%}
314
- {{- format_tool_response_block(ns_tname.name, tool_body) -}}
315
- {%- endif -%}
316
- {%- set ns_tr_out.flag = true -%}
317
- {%- set ns.prev_message_type = 'tool_response' -%}
318
- {%- endif -%}
319
- {%- endfor -%}
320
- {%- endif -%}
321
-
322
- {%- set captured_content -%}
323
- {%- if message.get('content') is string -%}
324
- {%- if role == 'model' -%}
325
- {{- strip_thinking(message['content']) -}}
326
- {%- else -%}
327
- {{- message['content'] | trim -}}
328
- {%- endif -%}
329
- {%- elif message.get('content') is sequence -%}
330
- {%- for item in message['content'] -%}
331
- {%- if item.get('type') == 'text' -%}
332
- {%- if role == 'model' -%}
333
- {{- strip_thinking(item['text']) -}}
334
- {%- else -%}
335
- {{- item['text'] | trim -}}
336
- {%- endif -%}
337
- {%- elif item.get('type') in ['image', 'image_url'] -%}
338
- {{- '<|image|>' -}}
339
- {%- elif item.get('type') in ['audio', 'input_audio'] -%}
340
- {{- '<|audio|>' -}}
341
- {%- elif item.get('type') == 'video' -%}
342
- {{- '<|video|>' -}}
343
- {%- endif -%}
344
- {%- endfor -%}
345
- {%- endif -%}
346
- {%- endset -%}
347
-
348
- {{- captured_content -}}
349
- {%- set has_content = captured_content | trim | length > 0 -%}
350
-
351
- {#- Forward-scan: find next non-tool message role for continuation detection -#}
352
- {%- set next_nt = namespace(role=None, found=false) -%}
353
- {%- for j in range(loop.index0 + 1, loop_messages | length) -%}
354
- {%- if not next_nt.found -%}
355
- {%- if loop_messages[j]['role'] != 'tool' -%}
356
- {%- set next_nt.role = loop_messages[j]['role'] -%}
357
- {%- set next_nt.found = true -%}
358
- {%- endif -%}
359
- {%- endif -%}
360
- {%- endfor -%}
361
-
362
- {%- set continues_into_next = (
363
- role == 'model'
364
- and next_nt.role == 'assistant'
365
- and (not message.get('tool_calls') or ns_tr_out.flag)
366
- ) -%}
367
-
368
- {%- if ns.prev_message_type == 'tool_call' and not ns_tr_out.flag -%}
369
- {{- '<|tool_response>' -}}
370
- {%- elif continues_into_next -%}
371
- {%- elif not (ns_tr_out.flag and not has_content and not next_nt.found) -%}
372
- {{- '<turn|>\n' -}}
373
- {%- endif -%}
374
-
375
- {#- Track previous non-tool role for next iteration (avoids O(n) backward scan) -#}
376
- {%- set ns.prev_non_tool_role = message['role'] -%}
377
- {%- endif -%}
378
- {%- endfor -%}
379
-
380
- {%- if add_generation_prompt -%}
381
- {%- if ns.prev_message_type != 'tool_response' and ns.prev_message_type != 'tool_call' -%}
382
- {{- '<|turn>model\n' -}}
383
- {%- elif ns.prev_message_type == 'tool_response' and enable_thinking -%}
384
- {{- '<|channel>thought\n' -}}
385
- {%- endif -%}
386
- {%- endif -%}
 
1
+ {#
2
+ Template: Google Gemma 4 Canonical Chat Template
3
+ Author: Google Gemma Engineering Team
4
+ Published: 2026-07-09
5
+ Context: Fixed tool-calling loops, turn closures, and thinking content-ordering.
6
+ #}
7
+ {%- macro format_parameters(properties, required, filter_keys=false) -%}
8
+ {%- set standard_keys = ['description', 'type', 'properties', 'required', 'nullable'] -%}
9
+ {%- set ns = namespace(found_first=false) -%}
10
+ {%- for key, value in properties | dictsort -%}
11
+ {%- set add_comma = false -%}
12
+ {%- if not filter_keys or key not in standard_keys -%}
13
+ {%- if ns.found_first %},{% endif -%}
14
+ {%- set ns.found_first = true -%}
15
+ {{ key }}:{
16
+ {%- if value['description'] -%}
17
+ description:<|"|>{{ value['description'] }}<|"|>
18
+ {%- set add_comma = true -%}
19
+ {%- endif -%}
20
+ {%- if value['type'] | upper == 'STRING' -%}
21
+ {%- if value['enum'] -%}
22
+ {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
23
+ enum:{{ format_argument(value['enum']) }}
24
+ {%- endif -%}
25
+ {%- elif value['type'] | upper == 'ARRAY' -%}
26
+ {%- if value['items'] is mapping and value['items'] -%}
27
+ {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
28
+ items:{
29
+ {%- set ns_items = namespace(found_first=false) -%}
30
+ {%- for item_key, item_value in value['items'] | dictsort -%}
31
+ {%- if item_value is not none -%}
32
+ {%- if ns_items.found_first %},{% endif -%}
33
+ {%- set ns_items.found_first = true -%}
34
+ {%- if item_key == 'properties' -%}
35
+ properties:{
36
+ {%- if item_value is mapping -%}
37
+ {{- format_parameters(item_value, value['items']['required'] | default([])) -}}
38
+ {%- endif -%}
39
+ }
40
+ {%- elif item_key == 'required' -%}
41
+ required:[
42
+ {%- for req_item in item_value -%}
43
+ <|"|>{{- req_item -}}<|"|>
44
+ {%- if not loop.last %},{% endif -%}
45
+ {%- endfor -%}
46
+ ]
47
+ {%- elif item_key == 'type' -%}
48
+ {%- if item_value is string -%}
49
+ type:{{ format_argument(item_value | upper) }}
50
+ {%- else -%}
51
+ type:{{ format_argument(item_value | map('upper') | list) }}
52
+ {%- endif -%}
53
+ {%- else -%}
54
+ {{ item_key }}:{{ format_argument(item_value) }}
55
+ {%- endif -%}
56
+ {%- endif -%}
57
+ {%- endfor -%}
58
+ }
59
+ {%- endif -%}
60
+ {%- endif -%}
61
+ {%- if value['nullable'] %}
62
+ {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
63
+ nullable:true
64
+ {%- endif -%}
65
+ {%- if value['type'] | upper == 'OBJECT' -%}
66
+ {%- if value['properties'] is defined and value['properties'] is mapping -%}
67
+ {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
68
+ properties:{
69
+ {{- format_parameters(value['properties'], value['required'] | default([])) -}}
70
+ }
71
+ {%- elif value is mapping -%}
72
+ {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
73
+ properties:{
74
+ {{- format_parameters(value, value['required'] | default([]), filter_keys=true) -}}
75
+ }
76
+ {%- endif -%}
77
+ {%- if value['required'] -%}
78
+ {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
79
+ required:[
80
+ {%- for item in value['required'] | default([]) -%}
81
+ <|"|>{{- item -}}<|"|>
82
+ {%- if not loop.last %},{% endif -%}
83
+ {%- endfor -%}
84
+ ]
85
+ {%- endif -%}
86
+ {%- endif -%}
87
+ {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
88
+ type:<|"|>{{ value['type'] | upper }}<|"|>}
89
+ {%- endif -%}
90
+ {%- endfor -%}
91
+ {%- endmacro -%}
92
+ {%- macro format_function_declaration(tool_data) -%}
93
+ declaration:{{- tool_data['function']['name'] -}}{description:<|"|>{{- tool_data['function']['description'] -}}<|"|>
94
+ {%- set params = tool_data['function']['parameters'] -%}
95
+ {%- if params -%}
96
+ ,parameters:{
97
+ {%- if params['properties'] -%}
98
+ properties:{ {{- format_parameters(params['properties'], params['required']) -}} },
99
+ {%- endif -%}
100
+ {%- if params['required'] -%}
101
+ required:[
102
+ {%- for item in params['required'] -%}
103
+ <|"|>{{- item -}}<|"|>
104
+ {{- ',' if not loop.last -}}
105
+ {%- endfor -%}
106
+ ],
107
+ {%- endif -%}
108
+ {%- if params['type'] -%}
109
+ type:<|"|>{{- params['type'] | upper -}}<|"|>}
110
+ {%- endif -%}
111
+ {%- endif -%}
112
+ {%- if 'response' in tool_data['function'] -%}
113
+ {%- set response_declaration = tool_data['function']['response'] -%}
114
+ ,response:{
115
+ {%- if response_declaration['description'] -%}
116
+ description:<|"|>{{- response_declaration['description'] -}}<|"|>,
117
+ {%- endif -%}
118
+ {%- if response_declaration['type'] | upper == 'OBJECT' -%}
119
+ type:<|"|>{{- response_declaration['type'] | upper -}}<|"|>}
120
+ {%- endif -%}
121
+ {%- endif -%}
122
+ }
123
+ {%- endmacro -%}
124
+ {%- macro format_argument(argument, escape_keys=True) -%}
125
+ {%- if argument is none -%}
126
+ {{- 'null' -}}
127
+ {%- elif argument is string -%}
128
+ {{- '<|"|>' + argument + '<|"|>' -}}
129
+ {%- elif argument is boolean -%}
130
+ {{- 'true' if argument else 'false' -}}
131
+ {%- elif argument is mapping -%}
132
+ {{- '{' -}}
133
+ {%- set ns = namespace(found_first=false) -%}
134
+ {%- for key, value in argument | dictsort -%}
135
+ {%- if ns.found_first %},{% endif -%}
136
+ {%- set ns.found_first = true -%}
137
+ {%- if escape_keys -%}
138
+ {{- '<|"|>' + key + '<|"|>' -}}
139
+ {%- else -%}
140
+ {{- key -}}
141
+ {%- endif -%}
142
+ :{{- format_argument(value, escape_keys=escape_keys) -}}
143
+ {%- endfor -%}
144
+ {{- '}' -}}
145
+ {%- elif argument is sequence -%}
146
+ {{- '[' -}}
147
+ {%- for item in argument -%}
148
+ {{- format_argument(item, escape_keys=escape_keys) -}}
149
+ {%- if not loop.last %},{% endif -%}
150
+ {%- endfor -%}
151
+ {{- ']' -}}
152
+ {%- else -%}
153
+ {{- argument -}}
154
+ {%- endif -%}
155
+ {%- endmacro -%}
156
+ {%- macro strip_thinking(text) -%}
157
+ {%- set ns = namespace(result='') -%}
158
+ {%- for part in text.split('<channel|>') -%}
159
+ {%- if '<|channel>' in part -%}
160
+ {%- set ns.result = ns.result + part.split('<|channel>')[0] -%}
161
+ {%- else -%}
162
+ {%- set ns.result = ns.result + part -%}
163
+ {%- endif -%}
164
+ {%- endfor -%}
165
+ {{- ns.result | trim -}}
166
+ {%- endmacro -%}
167
+
168
+ {%- macro format_tool_response_block(tool_name, response) -%}
169
+ {{- '<|tool_response>' -}}
170
+ {%- if response is mapping -%}
171
+ {{- 'response:' + tool_name + '{' -}}
172
+ {%- for key, value in response | dictsort -%}
173
+ {{- key -}}:{{- format_argument(value, escape_keys=False) -}}
174
+ {%- if not loop.last %},{% endif -%}
175
+ {%- endfor -%}
176
+ {{- '}' -}}
177
+ {%- else -%}
178
+ {{- 'response:' + tool_name + '{value:' + format_argument(response, escape_keys=False) + '}' -}}
179
+ {%- endif -%}
180
+ {{- '<tool_response|>' -}}
181
+ {%- endmacro -%}
182
+
183
+ {#- ===== SETUP ===== -#}
184
+ {%- set ns = namespace(prev_message_type=None, prev_non_tool_role=None) -%}
185
+ {%- set loop_messages = messages -%}
186
+ {%- set enable_thinking = enable_thinking | default(false) -%}
187
+ {%- set preserve_thinking = preserve_thinking | default(false) -%}
188
+ {{- bos_token -}}
189
+ {#- Handle System/Tool Definitions Block -#}
190
+ {%- if enable_thinking or tools or (messages and messages[0]['role'] in ['system', 'developer']) -%}
191
+ {{- '<|turn>system\n' -}}
192
+ {#- Inject Thinking token at the very top of the FIRST system turn -#}
193
+ {%- if enable_thinking -%}
194
+ {{- '<|think|>\n' -}}
195
+ {%- set ns.prev_message_type = 'think' -%}
196
+ {%- endif -%}
197
+ {%- if messages and messages[0]['role'] in ['system', 'developer'] -%}
198
+ {%- if messages[0]['content'] is string -%}
199
+ {{- messages[0]['content'] | trim -}}
200
+ {%- elif messages[0]['content'] is sequence -%}
201
+ {%- for item in messages[0]['content'] -%}
202
+ {{- item['text'] | trim + ' '-}}
203
+ {%- endfor -%}
204
+ {%- endif -%}
205
+ {%- set loop_messages = messages[1:] -%}
206
+ {%- endif -%}
207
+ {%- if tools -%}
208
+ {%- for tool in tools %}
209
+ {{- '<|tool>' -}}
210
+ {{- format_function_declaration(tool) | trim -}}
211
+ {{- '<tool|>' -}}
212
+ {%- endfor %}
213
+ {%- set ns.prev_message_type = 'tool' -%}
214
+ {%- endif -%}
215
+ {{- '<turn|>\n' -}}
216
+ {%- endif %}
217
+
218
+ {#- Pre-scan: find last user message index for reasoning guard -#}
219
+ {%- set ns_turn = namespace(last_user_idx=-1) -%}
220
+ {%- for i in range(loop_messages | length) -%}
221
+ {%- if loop_messages[i]['role'] == 'user' -%}
222
+ {%- set ns_turn.last_user_idx = i -%}
223
+ {%- endif -%}
224
+ {%- endfor -%}
225
+
226
+ {#- Loop through messages -#}
227
+ {%- for message in loop_messages -%}
228
+ {%- if message['role'] != 'tool' -%}
229
+ {%- set ns.prev_message_type = None -%}
230
+ {%- set role = 'model' if message['role'] == 'assistant' else message['role'] -%}
231
+ {#- Detect continuation using tracked state — O(1) instead of O(n) backward scan -#}
232
+ {%- set continue_same_model_turn = (role == 'model' and ns.prev_non_tool_role == 'assistant') -%}
233
+ {%- if not continue_same_model_turn -%}
234
+ {{- '<|turn>' + role + '\n' }}
235
+ {%- endif -%}
236
+
237
+ {#- Render reasoning/reasoning_content as thinking channel -#}
238
+ {%- set thinking_text = message.get('reasoning') or message.get('reasoning_content') -%}
239
+ {%- set thinking_gate = (loop.index0 > ns_turn.last_user_idx) or (preserve_thinking and message.get('tool_calls')) -%}
240
+ {%- if thinking_text and thinking_gate -%}
241
+ {{- '<|channel>thought\n' + thinking_text + '\n<channel|>' -}}
242
+ {%- endif -%}
243
+
244
+ {%- if message.get('tool_calls') -%}
245
+ {%- for tool_call in message.get('tool_calls') -%}
246
+ {%- set function = tool_call['function'] -%}
247
+ {{- '<|tool_call>call:' + function['name'] + '{' -}}
248
+ {%- if function['arguments'] is mapping -%}
249
+ {%- set ns_args = namespace(found_first=false) -%}
250
+ {%- for key, value in function['arguments'] | dictsort -%}
251
+ {%- if ns_args.found_first %},{% endif -%}
252
+ {%- set ns_args.found_first = true -%}
253
+ {{- key -}}:{{- format_argument(value, escape_keys=False) -}}
254
+ {%- endfor -%}
255
+ {%- elif function['arguments'] is none -%}
256
+ {%- else -%}
257
+ {{- raise_exception(
258
+ "chat_template: tool_calls[].function.arguments must be a "
259
+ "JSON object (mapping), not a string. Deserialize arguments "
260
+ "before passing to the template."
261
+ ) -}}
262
+ {%- endif -%}
263
+ {{- '}<tool_call|>' -}}
264
+ {%- endfor -%}
265
+ {%- set ns.prev_message_type = 'tool_call' -%}
266
+ {%- endif -%}
267
+
268
+ {%- set ns_tr_out = namespace(flag=false) -%}
269
+ {%- if message.get('tool_responses') -%}
270
+ {#- Legacy: tool_responses embedded on the assistant message (Google/Gemma native) -#}
271
+ {%- for tool_response in message.get('tool_responses') -%}
272
+ {{- format_tool_response_block(tool_response['name'] | default('unknown', true), tool_response['response']) -}}
273
+ {%- set ns_tr_out.flag = true -%}
274
+ {%- set ns.prev_message_type = 'tool_response' -%}
275
+ {%- endfor -%}
276
+ {%- elif message.get('tool_calls') -%}
277
+ {#- OpenAI Chat Completions: forward-scan consecutive role:tool messages -#}
278
+ {%- set ns_tool_scan = namespace(stopped=false) -%}
279
+ {%- for k in range(loop.index0 + 1, loop_messages | length) -%}
280
+ {%- if ns_tool_scan.stopped -%}
281
+ {%- elif loop_messages[k]['role'] != 'tool' -%}
282
+ {%- set ns_tool_scan.stopped = true -%}
283
+ {%- else -%}
284
+ {%- set follow = loop_messages[k] -%}
285
+ {#- Resolve tool_call_id to function name -#}
286
+ {%- set ns_tname = namespace(name=follow.get('name') or 'unknown') -%}
287
+ {%- for tc in message.get('tool_calls') -%}
288
+ {%- if tc.get('id') == follow.get('tool_call_id') -%}
289
+ {%- set ns_tname.name = tc['function']['name'] -%}
290
+ {%- endif -%}
291
+ {%- endfor -%}
292
+ {#- Handle content as string or content-parts array -#}
293
+ {%- set tool_body = follow.get('content') -%}
294
+ {%- if tool_body is string -%}
295
+ {{- format_tool_response_block(ns_tname.name, tool_body) -}}
296
+ {%- elif tool_body is sequence and tool_body is not string -%}
297
+ {%- set ns_txt = namespace(s='') -%}
298
+ {%- for part in tool_body -%}
299
+ {%- if part.get('type') == 'text' -%}
300
+ {%- set ns_txt.s = ns_txt.s + (part.get('text') | default('')) -%}
301
+ {%- endif -%}
302
+ {%- endfor -%}
303
+ {{- format_tool_response_block(ns_tname.name, ns_txt.s) -}}
304
+ {%- for part in tool_body -%}
305
+ {%- if part.get('type') in ['image', 'image_url'] -%}
306
+ {{- '<|image|>' -}}
307
+ {%- elif part.get('type') in ['audio', 'input_audio'] -%}
308
+ {{- '<|audio|>' -}}
309
+ {%- elif part.get('type') == 'video' -%}
310
+ {{- '<|video|>' -}}
311
+ {%- endif -%}
312
+ {%- endfor -%}
313
+ {%- else -%}
314
+ {{- format_tool_response_block(ns_tname.name, tool_body) -}}
315
+ {%- endif -%}
316
+ {%- set ns_tr_out.flag = true -%}
317
+ {%- set ns.prev_message_type = 'tool_response' -%}
318
+ {%- endif -%}
319
+ {%- endfor -%}
320
+ {%- endif -%}
321
+
322
+ {%- set captured_content -%}
323
+ {%- if message.get('content') is string -%}
324
+ {%- if role == 'model' -%}
325
+ {{- strip_thinking(message['content']) -}}
326
+ {%- else -%}
327
+ {{- message['content'] | trim -}}
328
+ {%- endif -%}
329
+ {%- elif message.get('content') is sequence -%}
330
+ {%- for item in message['content'] -%}
331
+ {%- if item.get('type') == 'text' -%}
332
+ {%- if role == 'model' -%}
333
+ {{- strip_thinking(item['text']) -}}
334
+ {%- else -%}
335
+ {{- item['text'] | trim -}}
336
+ {%- endif -%}
337
+ {%- elif item.get('type') in ['image', 'image_url'] -%}
338
+ {{- '<|image|>' -}}
339
+ {%- elif item.get('type') in ['audio', 'input_audio'] -%}
340
+ {{- '<|audio|>' -}}
341
+ {%- elif item.get('type') == 'video' -%}
342
+ {{- '<|video|>' -}}
343
+ {%- endif -%}
344
+ {%- endfor -%}
345
+ {%- endif -%}
346
+ {%- endset -%}
347
+
348
+ {{- captured_content -}}
349
+ {%- set has_content = captured_content | trim | length > 0 -%}
350
+
351
+ {#- Forward-scan: find next non-tool message role for continuation detection -#}
352
+ {%- set next_nt = namespace(role=None, found=false) -%}
353
+ {%- for j in range(loop.index0 + 1, loop_messages | length) -%}
354
+ {%- if not next_nt.found -%}
355
+ {%- if loop_messages[j]['role'] != 'tool' -%}
356
+ {%- set next_nt.role = loop_messages[j]['role'] -%}
357
+ {%- set next_nt.found = true -%}
358
+ {%- endif -%}
359
+ {%- endif -%}
360
+ {%- endfor -%}
361
+
362
+ {%- set continues_into_next = (
363
+ role == 'model'
364
+ and next_nt.role == 'assistant'
365
+ and (not message.get('tool_calls') or ns_tr_out.flag)
366
+ ) -%}
367
+
368
+ {%- if ns.prev_message_type == 'tool_call' and not ns_tr_out.flag -%}
369
+ {{- '<|tool_response>' -}}
370
+ {%- elif continues_into_next -%}
371
+ {%- elif not (ns_tr_out.flag and not has_content and not next_nt.found) -%}
372
+ {{- '<turn|>\n' -}}
373
+ {%- endif -%}
374
+
375
+ {#- Track previous non-tool role for next iteration (avoids O(n) backward scan) -#}
376
+ {%- set ns.prev_non_tool_role = message['role'] -%}
377
+ {%- endif -%}
378
+ {%- endfor -%}
379
+
380
+ {%- if add_generation_prompt -%}
381
+ {%- if ns.prev_message_type != 'tool_response' and ns.prev_message_type != 'tool_call' -%}
382
+ {{- '<|turn>model\n' -}}
383
+ {%- elif ns.prev_message_type == 'tool_response' and enable_thinking -%}
384
+ {{- '<|channel>thought\n' -}}
385
+ {%- endif -%}
386
+ {%- endif -%}
config.json CHANGED
@@ -1,212 +1,212 @@
1
- {
2
- "architectures": [
3
- "Gemma4Model"
4
- ],
5
- "audio_config": {
6
- "_name_or_path": "",
7
- "architectures": null,
8
- "attention_chunk_size": 12,
9
- "attention_context_left": 13,
10
- "attention_context_right": 0,
11
- "attention_invalid_logits_value": -1000000000.0,
12
- "attention_logit_cap": 50.0,
13
- "chunk_size_feed_forward": 0,
14
- "conv_kernel_size": 5,
15
- "dtype": "bfloat16",
16
- "gradient_clipping": 10000000000.0,
17
- "hidden_act": "silu",
18
- "hidden_size": 1024,
19
- "id2label": {
20
- "0": "LABEL_0",
21
- "1": "LABEL_1"
22
- },
23
- "initializer_range": 0.02,
24
- "is_encoder_decoder": false,
25
- "label2id": {
26
- "LABEL_0": 0,
27
- "LABEL_1": 1
28
- },
29
- "model_type": "gemma4_audio",
30
- "num_attention_heads": 8,
31
- "num_hidden_layers": 12,
32
- "output_attentions": false,
33
- "output_hidden_states": false,
34
- "output_proj_dims": 1536,
35
- "problem_type": null,
36
- "residual_weight": 0.5,
37
- "return_dict": true,
38
- "rms_norm_eps": 1e-06,
39
- "subsampling_conv_channels": [
40
- 128,
41
- 32
42
- ],
43
- "use_clipped_linears": true
44
- },
45
- "audio_token_id": 258881,
46
- "boa_token_id": 256000,
47
- "boi_token_id": 255999,
48
- "dtype": "bfloat16",
49
- "eoa_token_id": 258883,
50
- "eoa_token_index": 258883,
51
- "eoi_token_id": 258882,
52
- "eos_token_id": [
53
- 1,
54
- 106
55
- ],
56
- "image_token_id": 258880,
57
- "initializer_range": 0.02,
58
- "model_type": "gemma4",
59
- "text_config": {
60
- "attention_bias": false,
61
- "attention_dropout": 0.0,
62
- "attention_k_eq_v": false,
63
- "bos_token_id": 2,
64
- "dtype": "bfloat16",
65
- "enable_moe_block": false,
66
- "eos_token_id": 1,
67
- "expert_intermediate_size": null,
68
- "final_logit_softcapping": 30.0,
69
- "head_dim": 256,
70
- "hidden_activation": "gelu_pytorch_tanh",
71
- "hidden_size": 1536,
72
- "hidden_size_per_layer_input": 256,
73
- "initializer_range": 0.02,
74
- "intermediate_size": 6144,
75
- "layer_types": [
76
- "sliding_attention",
77
- "sliding_attention",
78
- "sliding_attention",
79
- "sliding_attention",
80
- "full_attention",
81
- "sliding_attention",
82
- "sliding_attention",
83
- "sliding_attention",
84
- "sliding_attention",
85
- "full_attention",
86
- "sliding_attention",
87
- "sliding_attention",
88
- "sliding_attention",
89
- "sliding_attention",
90
- "full_attention",
91
- "sliding_attention",
92
- "sliding_attention",
93
- "sliding_attention",
94
- "sliding_attention",
95
- "full_attention",
96
- "sliding_attention",
97
- "sliding_attention",
98
- "sliding_attention",
99
- "sliding_attention",
100
- "full_attention",
101
- "sliding_attention",
102
- "sliding_attention",
103
- "sliding_attention",
104
- "sliding_attention",
105
- "full_attention",
106
- "sliding_attention",
107
- "sliding_attention",
108
- "sliding_attention",
109
- "sliding_attention",
110
- "full_attention"
111
- ],
112
- "max_position_embeddings": 131072,
113
- "model_type": "gemma4_text",
114
- "moe_intermediate_size": null,
115
- "num_attention_heads": 8,
116
- "num_experts": null,
117
- "num_hidden_layers": 35,
118
- "num_key_value_heads": 1,
119
- "num_kv_shared_layers": 20,
120
- "pad_token_id": 0,
121
- "per_layer_config": {
122
- "04": {
123
- "head_dim": 512
124
- },
125
- "09": {
126
- "head_dim": 512
127
- },
128
- "14": {
129
- "head_dim": 512
130
- },
131
- "19": {
132
- "head_dim": 512
133
- },
134
- "24": {
135
- "head_dim": 512
136
- },
137
- "29": {
138
- "head_dim": 512
139
- },
140
- "34": {
141
- "head_dim": 512
142
- }
143
- },
144
- "rms_norm_eps": 1e-06,
145
- "rope_parameters": {
146
- "full_attention": {
147
- "partial_rotary_factor": 0.25,
148
- "rope_theta": 1000000.0,
149
- "rope_type": "proportional"
150
- },
151
- "sliding_attention": {
152
- "rope_theta": 10000.0,
153
- "rope_type": "default"
154
- }
155
- },
156
- "sliding_window": 512,
157
- "tie_word_embeddings": true,
158
- "top_k_experts": null,
159
- "use_bidirectional_attention": null,
160
- "use_cache": true,
161
- "use_double_wide_mlp": true,
162
- "vocab_size": 262156,
163
- "vocab_size_per_layer_input": 262156
164
- },
165
- "tie_word_embeddings": true,
166
- "transformers_version": "5.17.0",
167
- "video_token_id": 258884,
168
- "vision_config": {
169
- "_name_or_path": "",
170
- "architectures": null,
171
- "attention_bias": false,
172
- "attention_dropout": 0.0,
173
- "chunk_size_feed_forward": 0,
174
- "default_output_length": 280,
175
- "dtype": "bfloat16",
176
- "global_head_dim": 64,
177
- "head_dim": 64,
178
- "hidden_activation": "gelu_pytorch_tanh",
179
- "hidden_size": 768,
180
- "id2label": {
181
- "0": "LABEL_0",
182
- "1": "LABEL_1"
183
- },
184
- "initializer_range": 0.02,
185
- "intermediate_size": 3072,
186
- "is_encoder_decoder": false,
187
- "label2id": {
188
- "LABEL_0": 0,
189
- "LABEL_1": 1
190
- },
191
- "max_position_embeddings": 131072,
192
- "model_type": "gemma4_vision",
193
- "num_attention_heads": 12,
194
- "num_hidden_layers": 16,
195
- "num_key_value_heads": 12,
196
- "output_attentions": false,
197
- "output_hidden_states": false,
198
- "patch_size": 16,
199
- "pooling_kernel_size": 3,
200
- "position_embedding_size": 10240,
201
- "problem_type": null,
202
- "return_dict": true,
203
- "rms_norm_eps": 1e-06,
204
- "rope_parameters": {
205
- "rope_theta": 100.0,
206
- "rope_type": "axial"
207
- },
208
- "standardize": false,
209
- "use_clipped_linears": true
210
- },
211
- "vision_soft_tokens_per_image": 280
212
- }
 
1
+ {
2
+ "architectures": [
3
+ "Gemma4Model"
4
+ ],
5
+ "audio_config": {
6
+ "_name_or_path": "",
7
+ "architectures": null,
8
+ "attention_chunk_size": 12,
9
+ "attention_context_left": 13,
10
+ "attention_context_right": 0,
11
+ "attention_invalid_logits_value": -1000000000.0,
12
+ "attention_logit_cap": 50.0,
13
+ "chunk_size_feed_forward": 0,
14
+ "conv_kernel_size": 5,
15
+ "dtype": "bfloat16",
16
+ "gradient_clipping": 10000000000.0,
17
+ "hidden_act": "silu",
18
+ "hidden_size": 1024,
19
+ "id2label": {
20
+ "0": "LABEL_0",
21
+ "1": "LABEL_1"
22
+ },
23
+ "initializer_range": 0.02,
24
+ "is_encoder_decoder": false,
25
+ "label2id": {
26
+ "LABEL_0": 0,
27
+ "LABEL_1": 1
28
+ },
29
+ "model_type": "gemma4_audio",
30
+ "num_attention_heads": 8,
31
+ "num_hidden_layers": 12,
32
+ "output_attentions": false,
33
+ "output_hidden_states": false,
34
+ "output_proj_dims": 1536,
35
+ "problem_type": null,
36
+ "residual_weight": 0.5,
37
+ "return_dict": true,
38
+ "rms_norm_eps": 1e-06,
39
+ "subsampling_conv_channels": [
40
+ 128,
41
+ 32
42
+ ],
43
+ "use_clipped_linears": true
44
+ },
45
+ "audio_token_id": 258881,
46
+ "boa_token_id": 256000,
47
+ "boi_token_id": 255999,
48
+ "dtype": "bfloat16",
49
+ "eoa_token_id": 258883,
50
+ "eoa_token_index": 258883,
51
+ "eoi_token_id": 258882,
52
+ "eos_token_id": [
53
+ 1,
54
+ 106
55
+ ],
56
+ "image_token_id": 258880,
57
+ "initializer_range": 0.02,
58
+ "model_type": "gemma4",
59
+ "text_config": {
60
+ "attention_bias": false,
61
+ "attention_dropout": 0.0,
62
+ "attention_k_eq_v": false,
63
+ "bos_token_id": 2,
64
+ "dtype": "bfloat16",
65
+ "enable_moe_block": false,
66
+ "eos_token_id": 1,
67
+ "expert_intermediate_size": null,
68
+ "final_logit_softcapping": 30.0,
69
+ "head_dim": 256,
70
+ "hidden_activation": "gelu_pytorch_tanh",
71
+ "hidden_size": 1536,
72
+ "hidden_size_per_layer_input": 256,
73
+ "initializer_range": 0.02,
74
+ "intermediate_size": 6144,
75
+ "layer_types": [
76
+ "sliding_attention",
77
+ "sliding_attention",
78
+ "sliding_attention",
79
+ "sliding_attention",
80
+ "full_attention",
81
+ "sliding_attention",
82
+ "sliding_attention",
83
+ "sliding_attention",
84
+ "sliding_attention",
85
+ "full_attention",
86
+ "sliding_attention",
87
+ "sliding_attention",
88
+ "sliding_attention",
89
+ "sliding_attention",
90
+ "full_attention",
91
+ "sliding_attention",
92
+ "sliding_attention",
93
+ "sliding_attention",
94
+ "sliding_attention",
95
+ "full_attention",
96
+ "sliding_attention",
97
+ "sliding_attention",
98
+ "sliding_attention",
99
+ "sliding_attention",
100
+ "full_attention",
101
+ "sliding_attention",
102
+ "sliding_attention",
103
+ "sliding_attention",
104
+ "sliding_attention",
105
+ "full_attention",
106
+ "sliding_attention",
107
+ "sliding_attention",
108
+ "sliding_attention",
109
+ "sliding_attention",
110
+ "full_attention"
111
+ ],
112
+ "max_position_embeddings": 131072,
113
+ "model_type": "gemma4_text",
114
+ "moe_intermediate_size": null,
115
+ "num_attention_heads": 8,
116
+ "num_experts": null,
117
+ "num_hidden_layers": 35,
118
+ "num_key_value_heads": 1,
119
+ "num_kv_shared_layers": 20,
120
+ "pad_token_id": 0,
121
+ "per_layer_config": {
122
+ "04": {
123
+ "head_dim": 512
124
+ },
125
+ "09": {
126
+ "head_dim": 512
127
+ },
128
+ "14": {
129
+ "head_dim": 512
130
+ },
131
+ "19": {
132
+ "head_dim": 512
133
+ },
134
+ "24": {
135
+ "head_dim": 512
136
+ },
137
+ "29": {
138
+ "head_dim": 512
139
+ },
140
+ "34": {
141
+ "head_dim": 512
142
+ }
143
+ },
144
+ "rms_norm_eps": 1e-06,
145
+ "rope_parameters": {
146
+ "full_attention": {
147
+ "partial_rotary_factor": 0.25,
148
+ "rope_theta": 1000000.0,
149
+ "rope_type": "proportional"
150
+ },
151
+ "sliding_attention": {
152
+ "rope_theta": 10000.0,
153
+ "rope_type": "default"
154
+ }
155
+ },
156
+ "sliding_window": 512,
157
+ "tie_word_embeddings": true,
158
+ "top_k_experts": null,
159
+ "use_bidirectional_attention": null,
160
+ "use_cache": true,
161
+ "use_double_wide_mlp": true,
162
+ "vocab_size": 262156,
163
+ "vocab_size_per_layer_input": 262156
164
+ },
165
+ "tie_word_embeddings": true,
166
+ "transformers_version": "5.17.0",
167
+ "video_token_id": 258884,
168
+ "vision_config": {
169
+ "_name_or_path": "",
170
+ "architectures": null,
171
+ "attention_bias": false,
172
+ "attention_dropout": 0.0,
173
+ "chunk_size_feed_forward": 0,
174
+ "default_output_length": 280,
175
+ "dtype": "bfloat16",
176
+ "global_head_dim": 64,
177
+ "head_dim": 64,
178
+ "hidden_activation": "gelu_pytorch_tanh",
179
+ "hidden_size": 768,
180
+ "id2label": {
181
+ "0": "LABEL_0",
182
+ "1": "LABEL_1"
183
+ },
184
+ "initializer_range": 0.02,
185
+ "intermediate_size": 3072,
186
+ "is_encoder_decoder": false,
187
+ "label2id": {
188
+ "LABEL_0": 0,
189
+ "LABEL_1": 1
190
+ },
191
+ "max_position_embeddings": 131072,
192
+ "model_type": "gemma4_vision",
193
+ "num_attention_heads": 12,
194
+ "num_hidden_layers": 16,
195
+ "num_key_value_heads": 12,
196
+ "output_attentions": false,
197
+ "output_hidden_states": false,
198
+ "patch_size": 16,
199
+ "pooling_kernel_size": 3,
200
+ "position_embedding_size": 10240,
201
+ "problem_type": null,
202
+ "return_dict": true,
203
+ "rms_norm_eps": 1e-06,
204
+ "rope_parameters": {
205
+ "rope_theta": 100.0,
206
+ "rope_type": "axial"
207
+ },
208
+ "standardize": false,
209
+ "use_clipped_linears": true
210
+ },
211
+ "vision_soft_tokens_per_image": 280
212
+ }
model-00001-of-00003.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:edf8ee463b0d15927d3284d05c280a19c892755b540e498a79fa4a06903a6081
3
  size 1422163120
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:6de31fb3d2eba8b9091e5b9dfacac63fef4be2a4c46526249bc91009b3228b1e
3
  size 1422163120
model-00002-of-00003.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:5ff7df9b018c3a96e5344eaea2acbd8b2c786b7c1df21029c611efc183d0766f
3
  size 4987319954
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f4ccf6ad6d71c111042b42ea4b42b1ef5c0824d157660450def8df5f604503dc
3
  size 4987319954
model-00003-of-00003.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:bb95204a82f4568ac7460b89cb415feca73594a0c77ce020cfbef57c2d34c7b0
3
  size 3799608956
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cf679bf3a2f7809590a333c8d82da775996ff9bdb6fb550cb82949943b5cdee9
3
  size 3799608956
model.safetensors.index.json CHANGED
The diff for this file is too large to render. See raw diff
 
tokenizer_config.json CHANGED
@@ -1,153 +1,153 @@
1
- {
2
- "audio_token": "<|audio|>",
3
- "backend": "tokenizers",
4
- "boa_token": "<|audio>",
5
- "boi_token": "<|image>",
6
- "bos_token": "<bos>",
7
- "eoa_token": "<audio|>",
8
- "eoc_token": "<channel|>",
9
- "eoi_token": "<image|>",
10
- "eos_token": "<eos>",
11
- "eot_token": "<turn|>",
12
- "escape_token": "<|\"|>",
13
- "etc_token": "<tool_call|>",
14
- "etd_token": "<tool|>",
15
- "etr_token": "<tool_response|>",
16
- "extra_special_tokens": [
17
- "<|state_start|>",
18
- "<|state_end|>",
19
- "<|noul_start|>",
20
- "<|noul_end|>",
21
- "<|choice_start|>",
22
- "<|choice_end|>",
23
- "<|opt_start|>",
24
- "<|opt_end|>",
25
- "<|score_start|>",
26
- "<|score_end|>",
27
- "<|level_start|>",
28
- "<|level_end|>"
29
- ],
30
- "image_token": "<|image|>",
31
- "is_local": false,
32
- "local_files_only": false,
33
- "mask_token": "<mask>",
34
- "model_max_length": 1000000000000000019884624838656,
35
- "model_specific_special_tokens": {
36
- "audio_token": "<|audio|>",
37
- "boa_token": "<|audio>",
38
- "boi_token": "<|image>",
39
- "eoa_token": "<audio|>",
40
- "eoc_token": "<channel|>",
41
- "eoi_token": "<image|>",
42
- "eot_token": "<turn|>",
43
- "escape_token": "<|\"|>",
44
- "etc_token": "<tool_call|>",
45
- "etd_token": "<tool|>",
46
- "etr_token": "<tool_response|>",
47
- "image_token": "<|image|>",
48
- "soc_token": "<|channel>",
49
- "sot_token": "<|turn>",
50
- "stc_token": "<|tool_call>",
51
- "std_token": "<|tool>",
52
- "str_token": "<|tool_response>",
53
- "think_token": "<|think|>"
54
- },
55
- "pad_token": "<pad>",
56
- "padding_side": "left",
57
- "processor_class": "Gemma4Processor",
58
- "response_schema": {
59
- "properties": {
60
- "content": {
61
- "type": "string"
62
- },
63
- "role": {
64
- "const": "assistant"
65
- },
66
- "thinking": {
67
- "type": "string"
68
- },
69
- "tool_calls": {
70
- "items": {
71
- "properties": {
72
- "function": {
73
- "properties": {
74
- "arguments": {
75
- "additionalProperties": {},
76
- "type": "object",
77
- "x-parser": "gemma4-tool-call"
78
- },
79
- "name": {
80
- "type": "string"
81
- }
82
- },
83
- "type": "object",
84
- "x-regex": "call\\:(?P<name>\\w+)(?P<arguments>\\{.*\\})"
85
- },
86
- "type": {
87
- "const": "function"
88
- }
89
- },
90
- "type": "object"
91
- },
92
- "type": "array",
93
- "x-regex-iterator": "<\\|tool_call>(.*?)<tool_call\\|>"
94
- }
95
- },
96
- "type": "object",
97
- "x-regex": "(\\<\\|channel\\>thought\\n(?P<thinking>.*?)\\<channel\\|\\>)?(?P<tool_calls>\\<\\|tool_call\\>.*\\<tool_call\\|\\>)?(?P<content>(?:(?!\\<turn\\|\\>)(?!\\<\\|tool_response\\>).)+)?(?:\\<turn\\|\\>|\\<\\|tool_response\\>)?"
98
- },
99
- "response_template": {
100
- "defaults": {
101
- "role": "assistant"
102
- },
103
- "fields": {
104
- "content": {
105
- "close": [
106
- "<turn|>",
107
- "<|tool_response>",
108
- "<eos>"
109
- ],
110
- "content": "text"
111
- },
112
- "thinking": {
113
- "close": "<channel|>",
114
- "content": "text",
115
- "open": "<|channel>thought\n"
116
- },
117
- "tool_calls": {
118
- "close": "<tool_call|>",
119
- "content": "json",
120
- "content_args": {
121
- "string_delims": [
122
- [
123
- "<|\"|>",
124
- "<|\"|>"
125
- ]
126
- ],
127
- "unquoted_keys": true
128
- },
129
- "open_pattern": "<\\|tool_call>call:(?P<name>\\w+)",
130
- "repeats": true,
131
- "transform": {
132
- "function": {
133
- "arguments": "{content}",
134
- "name": "{name}"
135
- },
136
- "type": "function"
137
- }
138
- }
139
- },
140
- "start_anchor": [
141
- "<|turn>model\n",
142
- "<tool_response|>"
143
- ]
144
- },
145
- "soc_token": "<|channel>",
146
- "sot_token": "<|turn>",
147
- "stc_token": "<|tool_call>",
148
- "std_token": "<|tool>",
149
- "str_token": "<|tool_response>",
150
- "think_token": "<|think|>",
151
- "tokenizer_class": "GemmaTokenizer",
152
- "unk_token": "<unk>"
153
- }
 
1
+ {
2
+ "audio_token": "<|audio|>",
3
+ "backend": "tokenizers",
4
+ "boa_token": "<|audio>",
5
+ "boi_token": "<|image>",
6
+ "bos_token": "<bos>",
7
+ "eoa_token": "<audio|>",
8
+ "eoc_token": "<channel|>",
9
+ "eoi_token": "<image|>",
10
+ "eos_token": "<eos>",
11
+ "eot_token": "<turn|>",
12
+ "escape_token": "<|\"|>",
13
+ "etc_token": "<tool_call|>",
14
+ "etd_token": "<tool|>",
15
+ "etr_token": "<tool_response|>",
16
+ "extra_special_tokens": [
17
+ "<|state_start|>",
18
+ "<|state_end|>",
19
+ "<|noul_start|>",
20
+ "<|noul_end|>",
21
+ "<|choice_start|>",
22
+ "<|choice_end|>",
23
+ "<|opt_start|>",
24
+ "<|opt_end|>",
25
+ "<|score_start|>",
26
+ "<|score_end|>",
27
+ "<|level_start|>",
28
+ "<|level_end|>"
29
+ ],
30
+ "image_token": "<|image|>",
31
+ "is_local": false,
32
+ "local_files_only": false,
33
+ "mask_token": "<mask>",
34
+ "model_max_length": 1000000000000000019884624838656,
35
+ "model_specific_special_tokens": {
36
+ "audio_token": "<|audio|>",
37
+ "boa_token": "<|audio>",
38
+ "boi_token": "<|image>",
39
+ "eoa_token": "<audio|>",
40
+ "eoc_token": "<channel|>",
41
+ "eoi_token": "<image|>",
42
+ "eot_token": "<turn|>",
43
+ "escape_token": "<|\"|>",
44
+ "etc_token": "<tool_call|>",
45
+ "etd_token": "<tool|>",
46
+ "etr_token": "<tool_response|>",
47
+ "image_token": "<|image|>",
48
+ "soc_token": "<|channel>",
49
+ "sot_token": "<|turn>",
50
+ "stc_token": "<|tool_call>",
51
+ "std_token": "<|tool>",
52
+ "str_token": "<|tool_response>",
53
+ "think_token": "<|think|>"
54
+ },
55
+ "pad_token": "<pad>",
56
+ "padding_side": "left",
57
+ "processor_class": "Gemma4Processor",
58
+ "response_schema": {
59
+ "properties": {
60
+ "content": {
61
+ "type": "string"
62
+ },
63
+ "role": {
64
+ "const": "assistant"
65
+ },
66
+ "thinking": {
67
+ "type": "string"
68
+ },
69
+ "tool_calls": {
70
+ "items": {
71
+ "properties": {
72
+ "function": {
73
+ "properties": {
74
+ "arguments": {
75
+ "additionalProperties": {},
76
+ "type": "object",
77
+ "x-parser": "gemma4-tool-call"
78
+ },
79
+ "name": {
80
+ "type": "string"
81
+ }
82
+ },
83
+ "type": "object",
84
+ "x-regex": "call\\:(?P<name>\\w+)(?P<arguments>\\{.*\\})"
85
+ },
86
+ "type": {
87
+ "const": "function"
88
+ }
89
+ },
90
+ "type": "object"
91
+ },
92
+ "type": "array",
93
+ "x-regex-iterator": "<\\|tool_call>(.*?)<tool_call\\|>"
94
+ }
95
+ },
96
+ "type": "object",
97
+ "x-regex": "(\\<\\|channel\\>thought\\n(?P<thinking>.*?)\\<channel\\|\\>)?(?P<tool_calls>\\<\\|tool_call\\>.*\\<tool_call\\|\\>)?(?P<content>(?:(?!\\<turn\\|\\>)(?!\\<\\|tool_response\\>).)+)?(?:\\<turn\\|\\>|\\<\\|tool_response\\>)?"
98
+ },
99
+ "response_template": {
100
+ "defaults": {
101
+ "role": "assistant"
102
+ },
103
+ "fields": {
104
+ "content": {
105
+ "close": [
106
+ "<turn|>",
107
+ "<|tool_response>",
108
+ "<eos>"
109
+ ],
110
+ "content": "text"
111
+ },
112
+ "thinking": {
113
+ "close": "<channel|>",
114
+ "content": "text",
115
+ "open": "<|channel>thought\n"
116
+ },
117
+ "tool_calls": {
118
+ "close": "<tool_call|>",
119
+ "content": "json",
120
+ "content_args": {
121
+ "string_delims": [
122
+ [
123
+ "<|\"|>",
124
+ "<|\"|>"
125
+ ]
126
+ ],
127
+ "unquoted_keys": true
128
+ },
129
+ "open_pattern": "<\\|tool_call>call:(?P<name>\\w+)",
130
+ "repeats": true,
131
+ "transform": {
132
+ "function": {
133
+ "arguments": "{content}",
134
+ "name": "{name}"
135
+ },
136
+ "type": "function"
137
+ }
138
+ }
139
+ },
140
+ "start_anchor": [
141
+ "<|turn>model\n",
142
+ "<tool_response|>"
143
+ ]
144
+ },
145
+ "soc_token": "<|channel>",
146
+ "sot_token": "<|turn>",
147
+ "stc_token": "<|tool_call>",
148
+ "std_token": "<|tool>",
149
+ "str_token": "<|tool_response>",
150
+ "think_token": "<|think|>",
151
+ "tokenizer_class": "GemmaTokenizer",
152
+ "unk_token": "<unk>"
153
+ }