Add files using upload-large-folder tool

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+tokenizer.json filter=lfs diff=lfs merge=lfs -text

README.md

@@ -0,0 +1,132 @@
+---
+license: gemma
+language:
+  - en
+base_model: google/gemma-4-31b-it
+tags:
+  - mlx
+  - mlx-node
+  - quantized
+  - awq
+  - mxfp4
+  - micro-scaling-fp
+  - gemma4
+  - sliding-window-attention
+  - vision-language
+  - apple-silicon
+  - unsloth-dynamic
+library_name: mlx-node
+quantized_by: mlx-node
+pipeline_tag: text-generation
+model_type: gemma4
+---
+
+# Gemma-4-31B-IT — UD-MXFP4_K_XL (mlx-node)
+
+MXFP4 (OCP micro-scaling FP4) quantization of [google/gemma-4-31b-it](https://huggingface.co/google/gemma-4-31b-it) for Apple Silicon, using the [**Unsloth Dynamic** quantization strategy](https://unsloth.ai/docs/models/qwen3.5/gguf-benchmarks) via [mlx-node](https://github.com/mlx-node/mlx-node).
+
+| | Original (BF16) | UD-MXFP4_K_XL (this model) |
+|---|---|---|
+| **Size** | ~62 GB | **24 GB** |
+| **Format** | SafeTensors | SafeTensors |
+| **Precision** | BF16 uniform | MXFP4 (E8M0 scales) + mixed affine + BF16 |
+| **FFN group size** | — | **32** |
+| **Biases** | — | no (MLP); yes (affine layers) |
+
+## What is MXFP4?
+
+MXFP4 is the [Open Compute Project (OCP) micro-scaling FP4](https://www.opencompute.org/documents/ocp-microscaling-formats-mx-v1-0-spec-final-pdf) format. Each group of 32 elements shares a single 8-bit E8M0 scale (a power-of-two exponent), and elements themselves are stored as E2M1 FP4 values. Compared to 4-bit affine:
+
+- **Half the scale storage**: uint8 E8M0 vs. fp16/fp32 affine scales
+- **No biases**: zero-point implicit (FP4 covers ±range)
+- **Hardware-friendly**: scale is just an exponent shift, no FP multiply on the scale path
+
+For typical LLM weight distributions, MXFP4 retains quality on par with 4-bit affine while shrinking the metadata footprint.
+
+## All Variants
+
+| Repo | Bit budget | Size | Decode (tok/s) |
+|---|---|---|---|
+| [Brooooooklyn/Gemma-4-31B-IT-UD-Q2_K_XL-mlx](https://huggingface.co/Brooooooklyn/Gemma-4-31B-IT-UD-Q2_K_XL-mlx) | 2-bit base | 18 GB | 12.1 |
+| [Brooooooklyn/Gemma-4-31B-IT-UD-Q3_K_XL-mlx](https://huggingface.co/Brooooooklyn/Gemma-4-31B-IT-UD-Q3_K_XL-mlx) | 3-bit base | 21 GB | 10.4 |
+| [Brooooooklyn/Gemma-4-31B-IT-UD-Q4_K_XL-mlx](https://huggingface.co/Brooooooklyn/Gemma-4-31B-IT-UD-Q4_K_XL-mlx) | 4-bit base | 24 GB | 9.2 |
+| **[Brooooooklyn/Gemma-4-31B-IT-UD-MXFP4_K_XL-mlx](https://huggingface.co/Brooooooklyn/Gemma-4-31B-IT-UD-MXFP4_K_XL-mlx) (this model)** | **mxfp4** | **24 GB** | **10.6** |
+| [Brooooooklyn/Gemma-4-31B-IT-UD-NVFP4_K_XL-mlx](https://huggingface.co/Brooooooklyn/Gemma-4-31B-IT-UD-NVFP4_K_XL-mlx) | nvfp4 | 24 GB | 9.9 |
+| [Brooooooklyn/Gemma-4-31B-IT-UD-Q5_K_XL-mlx](https://huggingface.co/Brooooooklyn/Gemma-4-31B-IT-UD-Q5_K_XL-mlx) | 5-bit base | 28 GB | 6.8 |
+| [Brooooooklyn/Gemma-4-31B-IT-UD-Q6_K_XL-mlx](https://huggingface.co/Brooooooklyn/Gemma-4-31B-IT-UD-Q6_K_XL-mlx) | 6-bit base | 31 GB | 7.2 |
+| [Brooooooklyn/Gemma-4-31B-IT-UD-MXFP8_K_XL-mlx](https://huggingface.co/Brooooooklyn/Gemma-4-31B-IT-UD-MXFP8_K_XL-mlx) | mxfp8 | 34 GB | 7.8 |
+| [Brooooooklyn/Gemma-4-31B-IT-UD-Q8_K_XL-mlx](https://huggingface.co/Brooooooklyn/Gemma-4-31B-IT-UD-Q8_K_XL-mlx) | 8-bit base | 35 GB | 7.2 |
+
+Benchmarked on Apple M3 Max 128GB via [`examples/lm.ts`](https://github.com/mlx-node/mlx-node/blob/main/examples/lm.ts) (best decode tok/s across turns 2–4, steady-state, capitals chat with `reasoningEffort: 'low'`).
+
+## Performance
+
+Steady-state decode: **10.6 tok/s** on Apple M3 Max 128GB (best of turns 2–4, `examples/lm.ts` capitals chat with `reasoningEffort: 'low'`). Decode is memory-bandwidth bound on Apple Silicon — fewer bytes per token directly translates to higher throughput. Gemma-4-31B is fully dense (all 31B parameters active per token), so each decoded token must stream the entire quantized weight footprint from unified memory.
+
+## Per-Tensor Bit Assignments (N=4)
+
+| Weight | Mode | Bits | Group | Rationale |
+|---|---|---|---|---|
+| `embed_tokens` | 6-bit affine | 6 | 64 | Tied with lm_head (Gemma4 shares weights); affine-only loader |
+| `self_attn.q_proj` | 6-bit affine | 6 | 64 | AWQ-corrected via input_layernorm |
+| `self_attn.k_proj` | 6-bit affine | 6 | 64 | AWQ-corrected via input_layernorm |
+| `self_attn.v_proj` | 6-bit affine | 6 | 64 | AWQ-corrected via input_layernorm |
+| `mlp.gate_proj` | **mxfp4** | 4 | 32 | Dense MLP (top-level default) |
+| `mlp.up_proj` | **mxfp4** | 4 | 32 | Dense MLP (top-level default) |
+| `mlp.down_proj` | 5-bit affine | 5 | 64 | Dense MLP; "slightly more sensitive" (unsloth `base+1`) |
+| `self_attn.o_proj` | **bf16** | — | — | NOT AWQ-correctable; kept full-precision |
+
+## Quantization Strategy
+
+Built on Unsloth Dynamic 2.0 per-tensor KLD analysis. At `--q-bits 4` the unsloth recipe assigns 4-bit to MLP gate/up projections (the bulk of the parameter budget), 5-bit affine to down_proj, 6-bit affine + AWQ pre-scaling to attention q/k/v, and keeps `self_attn.o_proj` as bf16. Then `--q-mxfp` orthogonally promotes the 4-bit affine decisions to MXFP4 (`mode="mxfp4", bits=4, group_size=32`) — except for keys whose dequantizers are affine-only (`embed_tokens`, `lm_head`, `embed_vision.embedding_projection`).
+
+imatrix AWQ pre-scaling amplifies important weight channels and fuses inverse scales into preceding layer norms (zero inference overhead).
+
+## Architecture
+
+| Parameter | Value |
+|---|---|
+| Total parameters | ~31B (fully dense — all parameters active per token) |
+| Hidden size | 5,376 |
+| Layers | 60 (sliding-window attention) |
+| Attention heads | 32 (16 KV heads, GQA 2:1) |
+| Head dimension | 256 |
+| MLP intermediate size | 21,504 |
+| Vocab size | 262,144 |
+| Max context | 131,072 tokens |
+| Vision | yes (Gemma4ForConditionalGeneration) |
+
+## Usage
+
+```typescript
+import { loadSession } from '@mlx-node/lm';
+
+const session = await loadSession('./Gemma-4-31B-IT-UD-MXFP4_K_XL-mlx');
+
+for await (const event of session.sendStream('Explain the sliding-window attention design in Gemma-4.', {
+  config: { maxNewTokens: 2048, temperature: 0.6, reasoningEffort: 'low' },
+})) {
+  if (!event.done) process.stdout.write(event.text);
+}
+```
+
+## How It Was Made
+
+```bash
+mlx convert \
+  -i gemma-4-31b-it \
+  -o Gemma-4-31B-IT-UD-MXFP4_K_XL-mlx \
+  -q --q-bits 4 --q-mxfp --q-recipe unsloth \
+  --imatrix-path imatrix_unsloth.gguf
+```
+
+## Acknowledgments
+
+- **[Unsloth](https://unsloth.ai)** — Quantization strategy based on their [per-layer KLD benchmarks](https://unsloth.ai/docs/models/qwen3.5/gguf-benchmarks) and Dynamic 2.0 methodology
+- **[Google DeepMind](https://deepmind.google/)** — For the Gemma-4 model family
+- **[OCP Microscaling FP](https://www.opencompute.org/documents/ocp-microscaling-formats-mx-v1-0-spec-final-pdf)** — For the MXFP4/MXFP8 specification
+- **[Apple MLX](https://github.com/ml-explore/mlx)** — For the Metal-accelerated ML framework
+
+## License
+
+[Gemma Terms of Use](https://ai.google.dev/gemma/terms) (inherited from base model).

chat_template.jinja

@@ -0,0 +1,354 @@
+{%- macro format_parameters(properties, required, filter_keys=false) -%}
+    {%- set standard_keys = ['description', 'type', 'properties', 'required', 'nullable'] -%}
+    {%- set ns = namespace(found_first=false) -%}
+    {%- for key, value in properties | dictsort -%}
+        {%- set add_comma = false -%}
+        {%- if not filter_keys or key not in standard_keys -%}
+            {%- if ns.found_first %},{% endif -%}
+            {%- set ns.found_first = true -%}
+            {{ key }}:{
+            {%- if value['description'] -%}
+                description:<|"|>{{ value['description'] }}<|"|>
+                {%- set add_comma = true -%}
+            {%- endif -%}
+            {%- if value['type'] | upper == 'STRING' -%}
+                {%- if value['enum'] -%}
+                    {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
+                    enum:{{ format_argument(value['enum']) }}
+                {%- endif -%}
+            {%- elif value['type'] | upper == 'ARRAY' -%}
+                {%- if value['items'] is mapping and value['items'] -%}
+                    {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
+                    items:{
+                    {%- set ns_items = namespace(found_first=false) -%}
+                    {%- for item_key, item_value in value['items'] | dictsort -%}
+                        {%- if item_value is not none -%}
+                            {%- if ns_items.found_first %},{% endif -%}
+                            {%- set ns_items.found_first = true -%}
+                            {%- if item_key == 'properties' -%}
+                                properties:{
+                                {%- if item_value is mapping -%}
+                                    {{- format_parameters(item_value, value['items']['required'] | default([])) -}}
+                                {%- endif -%}
+                                }
+                            {%- elif item_key == 'required' -%}
+                                required:[
+                                {%- for req_item in item_value -%}
+                                    <|"|>{{- req_item -}}<|"|>
+                                    {%- if not loop.last %},{% endif -%}
+                                {%- endfor -%}
+                                ]
+                            {%- elif item_key == 'type' -%}
+                                {%- if item_value is string -%}
+                                    type:{{ format_argument(item_value | upper) }}
+                                {%- else -%}
+                                    type:{{ format_argument(item_value | map('upper') | list) }}
+                                {%- endif -%}
+                            {%- else -%}
+                                {{ item_key }}:{{ format_argument(item_value) }}
+                            {%- endif -%}
+                        {%- endif -%}
+                    {%- endfor -%}
+                    }
+                {%- endif -%}
+            {%- endif -%}
+            {%- if value['nullable'] %}
+                {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
+                nullable:true
+            {%- endif -%}
+            {%- if value['type'] | upper == 'OBJECT' -%}
+                {%- if value['properties'] is defined and value['properties'] is mapping -%}
+                    {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
+                    properties:{
+                    {{- format_parameters(value['properties'], value['required'] | default([])) -}}
+                    }
+                {%- elif value is mapping -%}
+                    {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
+                    properties:{
+                    {{- format_parameters(value, value['required'] | default([]), filter_keys=true) -}}
+                    }
+                {%- endif -%}
+                {%- if value['required'] -%}
+                    {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
+                    required:[
+                    {%- for item in value['required'] | default([]) -%}
+                        <|"|>{{- item -}}<|"|>
+                        {%- if not loop.last %},{% endif -%}
+                    {%- endfor -%}
+                    ]
+                {%- endif -%}
+            {%- endif -%}
+            {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
+            type:<|"|>{{ value['type'] | upper }}<|"|>}
+        {%- endif -%}
+    {%- endfor -%}
+{%- endmacro -%}
+{%- macro format_function_declaration(tool_data) -%}
+    declaration:{{- tool_data['function']['name'] -}}{description:<|"|>{{- tool_data['function']['description'] -}}<|"|>
+    {%- set params = tool_data['function']['parameters'] -%}
+    {%- if params -%}
+        ,parameters:{
+        {%- if params['properties'] -%}
+            properties:{ {{- format_parameters(params['properties'], params['required']) -}} },
+        {%- endif -%}
+        {%- if params['required'] -%}
+            required:[
+            {%- for item in params['required'] -%}
+                <|"|>{{- item -}}<|"|>
+                {{- ',' if not loop.last -}}
+            {%- endfor -%}
+            ],
+        {%- endif -%}
+        {%- if params['type'] -%}
+            type:<|"|>{{- params['type'] | upper -}}<|"|>}
+        {%- endif -%}
+    {%- endif -%}
+    {%- if 'response' in tool_data['function'] -%}
+        {%- set response_declaration = tool_data['function']['response'] -%}
+        ,response:{
+        {%- if response_declaration['description'] -%}
+            description:<|"|>{{- response_declaration['description'] -}}<|"|>,
+        {%- endif -%}
+        {%- if response_declaration['type'] | upper == 'OBJECT' -%}
+            type:<|"|>{{- response_declaration['type'] | upper -}}<|"|>}
+        {%- endif -%}
+    {%- endif -%}
+    }
+{%- endmacro -%}
+{%- macro format_argument(argument, escape_keys=True) -%}
+    {%- if argument is string -%}
+        {{- '<|"|>' + argument + '<|"|>' -}}
+    {%- elif argument is boolean -%}
+        {{- 'true' if argument else 'false' -}}
+    {%- elif argument is mapping -%}
+        {{- '{' -}}
+        {%- set ns = namespace(found_first=false) -%}
+        {%- for key, value in argument | dictsort -%}
+            {%- if ns.found_first %},{% endif -%}
+            {%- set ns.found_first = true -%}
+            {%- if escape_keys -%}
+                {{- '<|"|>' + key + '<|"|>' -}}
+            {%- else -%}
+                {{- key -}}
+            {%- endif -%}
+            :{{- format_argument(value, escape_keys=escape_keys) -}}
+        {%- endfor -%}
+        {{- '}' -}}
+    {%- elif argument is sequence -%}
+        {{- '[' -}}
+        {%- for item in argument -%}
+            {{- format_argument(item, escape_keys=escape_keys) -}}
+            {%- if not loop.last %},{% endif -%}
+        {%- endfor -%}
+        {{- ']' -}}
+    {%- else -%}
+        {{- argument -}}
+    {%- endif -%}
+{%- endmacro -%}
+{%- macro strip_thinking(text) -%}
+    {%- set ns = namespace(result='') -%}
+    {%- for part in text.split('<channel|>') -%}
+        {%- if '<|channel>' in part -%}
+            {%- set ns.result = ns.result + part.split('<|channel>')[0] -%}
+        {%- else -%}
+            {%- set ns.result = ns.result + part -%}
+        {%- endif -%}
+    {%- endfor -%}
+    {{- ns.result | trim -}}
+{%- endmacro -%}
+
+{%- macro format_tool_response_block(tool_name, response) -%}
+    {{- '<|tool_response>' -}}
+    {%- if response is mapping -%}
+        {{- 'response:' + tool_name + '{' -}}
+        {%- for key, value in response | dictsort -%}
+            {{- key -}}:{{- format_argument(value, escape_keys=False) -}}
+            {%- if not loop.last %},{% endif -%}
+        {%- endfor -%}
+        {{- '}' -}}
+    {%- else -%}
+        {{- 'response:' + tool_name + '{value:' + format_argument(response, escape_keys=False) + '}' -}}
+    {%- endif -%}
+    {{- '<tool_response|>' -}}
+{%- endmacro -%}
+
+{%- set ns = namespace(prev_message_type=None) -%}
+{%- set loop_messages = messages -%}
+{{- bos_token -}}
+{#- Handle System/Tool Definitions Block -#}
+{%- if (enable_thinking is defined and enable_thinking) or tools or messages[0]['role'] in ['system', 'developer'] -%}
+    {{- '<|turn>system\n' -}}
+    {#- Inject Thinking token at the very top of the FIRST system turn -#}
+    {%- if enable_thinking is defined and enable_thinking -%}
+        {{- '<|think|>\n' -}}
+        {%- set ns.prev_message_type = 'think' -%}
+    {%- endif -%}
+    {%- if messages[0]['role'] in ['system', 'developer'] -%}
+        {%- if messages[0]['content'] is string -%}
+            {{- messages[0]['content'] | trim -}}
+        {%- elif messages[0]['content'] is sequence -%}
+            {%- for item in messages[0]['content'] -%}
+                {{- item['text'] | trim + ' '-}}
+            {%- endfor -%}
+        {%- endif -%}
+        {%- set loop_messages = messages[1:] -%}
+    {%- endif -%}
+    {%- if tools -%}
+        {%- for tool in tools %}
+            {{- '<|tool>' -}}
+            {{- format_function_declaration(tool) | trim -}}
+            {{- '<tool|>' -}}
+        {%- endfor %}
+        {%- set ns.prev_message_type = 'tool' -%}
+    {%- endif -%}
+    {{- '<turn|>\n' -}}
+{%- endif %}
+
+{#- Pre-scan: find last user message index for reasoning guard -#}
+{%- set ns_turn = namespace(last_user_idx=-1) -%}
+{%- for i in range(loop_messages | length) -%}
+    {%- if loop_messages[i]['role'] == 'user' -%}
+        {%- set ns_turn.last_user_idx = i -%}
+    {%- endif -%}
+{%- endfor -%}
+
+{#- Loop through messages -#}
+{%- for message in loop_messages -%}
+    {%- if message['role'] != 'tool' -%}
+    {%- set ns.prev_message_type = None -%}
+    {%- set role = 'model' if message['role'] == 'assistant' else message['role'] -%}
+    {#- Detect continuation: suppress duplicate <|turn>model when previous non-tool message was also assistant -#}
+    {%- set prev_nt = namespace(role=None, found=false) -%}
+    {%- if loop.index0 > 0 -%}
+        {%- for j in range(loop.index0 - 1, -1, -1) -%}
+            {%- if not prev_nt.found -%}
+                {%- if loop_messages[j]['role'] != 'tool' -%}
+                    {%- set prev_nt.role = loop_messages[j]['role'] -%}
+                    {%- set prev_nt.found = true -%}
+                {%- endif -%}
+            {%- endif -%}
+        {%- endfor -%}
+    {%- endif -%}
+    {%- set continue_same_model_turn = (role == 'model' and prev_nt.role == 'assistant') -%}
+    {%- if not continue_same_model_turn -%}
+        {{- '<|turn>' + role + '\n' }}
+    {%- endif -%}
+
+    {#- Render reasoning/reasoning_content as thinking channel -#}
+    {%- set thinking_text = message.get('reasoning') or message.get('reasoning_content') -%}
+    {%- if thinking_text and loop.index0 > ns_turn.last_user_idx and message.get('tool_calls') -%}
+        {{- '<|channel>thought\n' + thinking_text + '\n<channel|>' -}}
+    {%- endif -%}
+
+            {%- if message['tool_calls'] -%}
+                {%- for tool_call in message['tool_calls'] -%}
+                    {%- set function = tool_call['function'] -%}
+                    {{- '<|tool_call>call:' + function['name'] + '{' -}}
+                    {%- if function['arguments'] is mapping -%}
+                        {%- set ns_args = namespace(found_first=false) -%}
+                        {%- for key, value in function['arguments'] | dictsort -%}
+                            {%- if ns_args.found_first %},{% endif -%}
+                            {%- set ns_args.found_first = true -%}
+                            {{- key -}}:{{- format_argument(value, escape_keys=False) -}}
+                        {%- endfor -%}
+                    {%- elif function['arguments'] is string -%}
+                        {{- function['arguments'] -}}
+                    {%- endif -%}
+                    {{- '}<tool_call|>' -}}
+                {%- endfor -%}
+                {%- set ns.prev_message_type = 'tool_call' -%}
+            {%- endif -%}
+
+            {%- set ns_tr_out = namespace(flag=false) -%}
+            {%- if message.get('tool_responses') -%}
+                {#- Legacy: tool_responses embedded on the assistant message (Google/Gemma native) -#}
+                {%- for tool_response in message['tool_responses'] -%}
+                    {{- format_tool_response_block(tool_response['name'] | default('unknown'), tool_response['response']) -}}
+                    {%- set ns_tr_out.flag = true -%}
+                    {%- set ns.prev_message_type = 'tool_response' -%}
+                {%- endfor -%}
+            {%- elif message.get('tool_calls') -%}
+                {#- OpenAI Chat Completions: forward-scan consecutive role:tool messages -#}
+                {%- set ns_tool_scan = namespace(stopped=false) -%}
+                {%- for k in range(loop.index0 + 1, loop_messages | length) -%}
+                    {%- if ns_tool_scan.stopped -%}
+                    {%- elif loop_messages[k]['role'] != 'tool' -%}
+                        {%- set ns_tool_scan.stopped = true -%}
+                    {%- else -%}
+                        {%- set follow = loop_messages[k] -%}
+                        {#- Resolve tool_call_id to function name -#}
+                        {%- set ns_tname = namespace(name=follow.get('name') | default('unknown')) -%}
+                        {%- for tc in message['tool_calls'] -%}
+                            {%- if tc.get('id') == follow.get('tool_call_id') -%}
+                                {%- set ns_tname.name = tc['function']['name'] -%}
+                            {%- endif -%}
+                        {%- endfor -%}
+                        {#- Handle content as string or content-parts array -#}
+                        {%- set tool_body = follow.get('content') -%}
+                        {%- if tool_body is string -%}
+                            {{- format_tool_response_block(ns_tname.name, tool_body) -}}
+                        {%- elif tool_body is sequence and tool_body is not string -%}
+                            {%- set ns_txt = namespace(s='') -%}
+                            {%- for part in tool_body -%}
+                                {%- if part.get('type') == 'text' -%}
+                                    {%- set ns_txt.s = ns_txt.s + (part.get('text') | default('')) -%}
+                                {%- endif -%}
+                            {%- endfor -%}
+                            {{- format_tool_response_block(ns_tname.name, ns_txt.s) -}}
+                        {%- else -%}
+                            {{- format_tool_response_block(ns_tname.name, tool_body) -}}
+                        {%- endif -%}
+                        {%- set ns_tr_out.flag = true -%}
+                        {%- set ns.prev_message_type = 'tool_response' -%}
+                    {%- endif -%}
+                {%- endfor -%}
+            {%- endif -%}
+
+            {%- set captured_content -%}
+            {%- if message['content'] is string -%}
+                {%- if role == 'model' -%}
+                    {{- strip_thinking(message['content']) -}}
+                {%- else -%}
+                    {{- message['content'] | trim -}}
+                {%- endif -%}
+            {%- elif message['content'] is sequence -%}
+                {%- for item in message['content'] -%}
+                    {%- if item['type'] == 'text' -%}
+                        {%- if role == 'model' -%}
+                            {{- strip_thinking(item['text']) -}}
+                        {%- else -%}
+                            {{- item['text'] | trim -}}
+                        {%- endif -%}
+                    {%- elif item['type'] == 'image' -%}
+                        {{- '<|image|>' -}}
+                        {%- set ns.prev_message_type = 'image' -%}
+                    {%- elif item['type'] == 'audio' -%}
+                        {{- '<|audio|>' -}}
+                        {%- set ns.prev_message_type = 'audio' -%}
+                    {%- elif item['type'] == 'video' -%}
+                        {{- '<|video|>' -}}
+                        {%- set ns.prev_message_type = 'video' -%}
+                    {%- endif -%}
+                {%- endfor -%}
+            {%- endif -%}
+            {%- endset -%}
+
+            {{- captured_content -}}
+            {%- set has_content = captured_content | trim | length > 0 -%}
+
+        {%- if ns.prev_message_type == 'tool_call' and not ns_tr_out.flag -%}
+            {{- '<|tool_response>' -}}
+        {%- elif not (ns_tr_out.flag and not has_content) -%}
+            {{- '<turn|>\n' -}}
+        {%- endif -%}
+    {%- endif -%}
+{%- endfor -%}
+
+{%- if add_generation_prompt -%}
+    {%- if ns.prev_message_type != 'tool_response' and ns.prev_message_type != 'tool_call' -%}
+        {{- '<|turn>model\n' -}}
+        {%- if not enable_thinking | default(false) -%}
+            {{- '<|channel>thought\n<channel|>' -}}
+        {%- endif -%}
+    {%- endif -%}
+{%- endif -%}

config.json

@@ -0,0 +1,3696 @@
+{
+  "architectures": [
+    "Gemma4ForConditionalGeneration"
+  ],
+  "audio_config": null,
+  "audio_token_id": 258881,
+  "boa_token_id": 256000,
+  "boi_token_id": 255999,
+  "dtype": "bfloat16",
+  "eoa_token_id": 258883,
+  "eoa_token_index": 258883,
+  "eoi_token_id": 258882,
+  "eos_token_id": [
+    1,
+    106
+  ],
+  "image_token_id": 258880,
+  "initializer_range": 0.02,
+  "model_type": "gemma4",
+  "quantization": {
+    "group_size": 64,
+    "bits": 4,
+    "mode": "affine",
+    "language_model.model.layers.37.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.54.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.16.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.25.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.21.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.7.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.58.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.24.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.39.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.51.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.25.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.41.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.11.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.29.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.22.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.44.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.42.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.39.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.6.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.24.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.28.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.0.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.19.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.7.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.4.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.33.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.51.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.embed_tokens": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.45.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.44.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.15.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.35.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.8.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.0.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.36.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.46.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.18.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.32.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.23.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.55.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.57.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.54.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.31.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.14.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.27.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.6.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.13.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.1.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.50.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.38.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.4.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.38.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.37.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.43.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.44.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.16.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.41.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.31.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.57.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.20.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.23.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.12.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.41.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.55.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.33.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.42.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.56.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.52.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.25.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.28.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.53.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.23.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.55.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.53.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.36.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.6.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.30.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.51.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.17.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.5.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.28.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.39.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.36.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.9.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.59.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.56.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.11.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.59.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.42.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.29.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.57.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.27.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.5.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.4.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.46.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.48.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.57.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.10.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.38.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.13.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.2.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.41.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.4.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.49.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.48.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.31.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.34.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.45.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.16.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.8.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.34.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.56.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.18.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.40.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.20.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.56.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.15.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.28.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.23.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.47.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.46.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.17.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.24.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.10.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.47.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.1.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.54.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.44.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.7.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.1.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.0.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.9.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.52.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.58.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.21.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.59.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.18.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.58.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.3.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.35.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.39.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.16.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.55.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.59.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.12.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.21.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.34.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.49.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.25.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.42.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.30.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.17.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.59.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.51.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.36.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.33.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.8.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.50.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.9.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.5.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.40.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.53.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.2.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.50.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.43.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.58.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.58.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.28.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.11.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.39.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.16.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.44.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.31.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.10.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.21.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.21.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.12.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.23.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.40.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.52.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.55.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.22.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.56.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.30.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.12.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.37.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.46.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.33.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.26.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.19.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.10.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.50.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.15.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.27.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.9.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.3.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.14.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.30.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.2.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.48.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.49.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.2.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.47.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.24.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.14.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.15.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.10.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.10.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.31.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.34.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.26.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.45.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.30.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.31.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.20.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.14.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.24.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.20.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.32.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.15.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.51.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.47.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.49.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.26.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.57.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.28.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.35.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.22.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.3.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.1.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.50.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.49.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.45.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.37.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.49.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.43.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.19.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.37.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.52.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.43.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.27.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.32.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.17.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.53.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.16.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.4.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.19.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.34.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.1.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.25.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.56.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.54.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.8.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.22.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.20.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.5.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.9.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.6.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.13.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.21.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.51.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.18.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.25.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.52.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.8.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.42.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.11.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.4.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.54.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.9.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.30.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.32.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.7.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.53.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.43.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.12.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.19.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.18.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.13.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.34.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.26.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.13.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.13.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.57.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.14.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.12.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.2.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.43.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.38.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.8.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.27.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.45.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.58.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.0.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.22.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.1.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.41.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.24.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.2.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.19.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.33.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.26.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.45.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.46.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.48.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.15.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.18.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.48.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.0.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.46.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.48.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.42.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.29.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.47.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.20.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.33.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.6.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.40.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.38.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.44.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.0.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.5.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.7.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.6.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.50.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.3.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.55.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.35.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.7.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.36.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.29.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.29.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.38.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.40.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.37.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.40.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.3.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.14.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.32.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.27.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.3.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.54.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.39.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.35.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.11.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.36.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.26.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.32.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.22.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.17.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.52.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    }
+  },
+  "quantization_config": {
+    "group_size": 64,
+    "bits": 4,
+    "mode": "affine",
+    "language_model.model.layers.37.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.54.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.16.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.25.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.21.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.7.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.58.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.24.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.39.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.51.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.25.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.41.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.11.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.29.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.22.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.44.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.42.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.39.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.6.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.24.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.28.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.0.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.19.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.7.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.4.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.33.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.51.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.embed_tokens": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.45.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.44.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.15.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.35.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.8.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.0.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.36.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.46.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.18.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.32.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.23.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.55.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.57.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.54.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.31.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.14.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.27.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.6.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.13.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.1.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.50.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.38.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.4.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.38.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.37.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.43.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.44.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.16.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.41.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.31.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.57.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.20.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.23.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.12.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.41.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.55.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.33.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.42.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.56.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.52.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.25.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.28.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.53.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.23.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.55.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.53.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.36.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.6.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.30.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.51.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.17.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.5.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.28.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.39.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.36.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.9.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.59.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.56.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.11.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.59.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.42.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.29.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.57.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.27.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.5.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.4.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.46.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.48.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.57.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.10.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.38.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.13.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.2.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.41.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.4.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.49.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.48.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.31.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.34.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.45.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.16.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.8.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.34.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.56.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.18.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.40.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.20.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.56.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.15.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.28.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.23.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.47.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.46.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.17.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.24.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.10.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.47.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.1.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.54.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.44.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.7.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.1.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.0.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.9.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.52.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.58.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.21.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.59.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.18.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.58.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.3.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.35.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.39.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.16.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.55.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.59.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.12.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.21.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.34.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.49.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.25.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.42.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.30.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.17.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.59.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.51.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.36.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.33.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.8.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.50.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.9.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.5.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.40.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.53.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.2.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.50.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.43.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.58.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.58.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.28.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.11.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.39.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.16.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.44.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.31.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.10.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.21.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.21.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.12.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.23.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.40.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.52.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.55.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.22.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.56.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.30.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.12.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.37.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.46.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.33.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.26.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.19.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.10.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.50.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.15.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.27.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.9.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.3.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.14.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.30.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.2.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.48.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.49.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.2.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.47.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.24.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.14.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.15.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.10.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.10.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.31.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.34.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.26.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.45.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.30.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.31.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.20.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.14.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.24.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.20.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.32.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.15.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.51.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.47.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.49.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.26.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.57.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.28.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.35.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.22.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.3.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.1.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.50.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.49.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.45.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.37.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.49.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.43.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.19.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.37.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.52.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.43.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.27.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.32.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.17.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.53.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.16.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.4.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.19.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.34.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.1.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.25.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.56.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.54.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.8.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.22.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.20.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.5.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.9.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.6.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.13.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.21.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.51.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.18.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.25.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.52.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.8.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.42.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.11.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.4.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.54.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.9.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.30.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.32.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.7.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.53.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.43.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.12.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.19.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.18.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.13.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.34.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.26.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.13.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.13.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.57.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.14.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.12.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.2.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.43.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.38.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.8.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.27.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.45.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.58.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.0.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.22.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.1.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.41.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.24.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.2.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.19.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.33.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.26.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.45.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.46.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.48.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.15.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.18.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.48.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.0.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.46.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.48.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.42.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.29.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.47.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.20.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.33.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.6.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.40.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.38.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.44.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.0.self_attn.v_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.5.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.7.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.6.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.50.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.3.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.55.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.35.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.7.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.36.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.29.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.29.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.38.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.40.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.37.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.40.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.3.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.14.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.32.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.27.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.3.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.54.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.39.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.35.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.11.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.36.mlp.down_proj": {
+      "bits": 5,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.26.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.32.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.22.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "mxfp4"
+    },
+    "language_model.model.layers.17.self_attn.k_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.52.self_attn.q_proj": {
+      "bits": 6,
+      "group_size": 64,
+      "mode": "affine"
+    }
+  },
+  "text_config": {
+    "attention_bias": false,
+    "attention_dropout": 0.0,
+    "attention_k_eq_v": true,
+    "bos_token_id": 2,
+    "dtype": "bfloat16",
+    "enable_moe_block": false,
+    "eos_token_id": 1,
+    "expert_intermediate_size": null,
+    "final_logit_softcapping": 30.0,
+    "global_head_dim": 512,
+    "head_dim": 256,
+    "hidden_activation": "gelu_pytorch_tanh",
+    "hidden_size": 5376,
+    "hidden_size_per_layer_input": 0,
+    "initializer_range": 0.02,
+    "intermediate_size": 21504,
+    "layer_types": [
+      "sliding_attention",
+      "sliding_attention",
+      "sliding_attention",
+      "sliding_attention",
+      "sliding_attention",
+      "full_attention",
+      "sliding_attention",
+      "sliding_attention",
+      "sliding_attention",
+      "sliding_attention",
+      "sliding_attention",
+      "full_attention",
+      "sliding_attention",
+      "sliding_attention",
+      "sliding_attention",
+      "sliding_attention",
+      "sliding_attention",
+      "full_attention",
+      "sliding_attention",
+      "sliding_attention",
+      "sliding_attention",
+      "sliding_attention",
+      "sliding_attention",
+      "full_attention",
+      "sliding_attention",
+      "sliding_attention",
+      "sliding_attention",
+      "sliding_attention",
+      "sliding_attention",
+      "full_attention",
+      "sliding_attention",
+      "sliding_attention",
+      "sliding_attention",
+      "sliding_attention",
+      "sliding_attention",
+      "full_attention",
+      "sliding_attention",
+      "sliding_attention",
+      "sliding_attention",
+      "sliding_attention",
+      "sliding_attention",
+      "full_attention",
+      "sliding_attention",
+      "sliding_attention",
+      "sliding_attention",
+      "sliding_attention",
+      "sliding_attention",
+      "full_attention",
+      "sliding_attention",
+      "sliding_attention",
+      "sliding_attention",
+      "sliding_attention",
+      "sliding_attention",
+      "full_attention",
+      "sliding_attention",
+      "sliding_attention",
+      "sliding_attention",
+      "sliding_attention",
+      "sliding_attention",
+      "full_attention"
+    ],
+    "max_position_embeddings": 262144,
+    "model_type": "gemma4_text",
+    "num_attention_heads": 32,
+    "num_experts": null,
+    "num_global_key_value_heads": 4,
+    "num_hidden_layers": 60,
+    "num_key_value_heads": 16,
+    "num_kv_shared_layers": 0,
+    "pad_token_id": 0,
+    "rms_norm_eps": 1e-6,
+    "rope_parameters": {
+      "full_attention": {
+        "partial_rotary_factor": 0.25,
+        "rope_theta": 1000000.0,
+        "rope_type": "proportional"
+      },
+      "sliding_attention": {
+        "rope_theta": 10000.0,
+        "rope_type": "default"
+      }
+    },
+    "sliding_window": 1024,
+    "tie_word_embeddings": true,
+    "top_k_experts": null,
+    "use_bidirectional_attention": "vision",
+    "use_cache": true,
+    "use_double_wide_mlp": false,
+    "vocab_size": 262144,
+    "vocab_size_per_layer_input": 262144
+  },
+  "tie_word_embeddings": true,
+  "transformers_version": "5.5.0.dev0",
+  "video_token_id": 258884,
+  "vision_config": {
+    "_name_or_path": "",
+    "architectures": null,
+    "attention_bias": false,
+    "attention_dropout": 0.0,
+    "chunk_size_feed_forward": 0,
+    "default_output_length": 280,
+    "dtype": "bfloat16",
+    "global_head_dim": 72,
+    "head_dim": 72,
+    "hidden_activation": "gelu_pytorch_tanh",
+    "hidden_size": 1152,
+    "id2label": {
+      "0": "LABEL_0",
+      "1": "LABEL_1"
+    },
+    "initializer_range": 0.02,
+    "intermediate_size": 4304,
+    "is_encoder_decoder": false,
+    "label2id": {
+      "LABEL_0": 0,
+      "LABEL_1": 1
+    },
+    "max_position_embeddings": 131072,
+    "model_type": "gemma4_vision",
+    "num_attention_heads": 16,
+    "num_hidden_layers": 27,
+    "num_key_value_heads": 16,
+    "output_attentions": false,
+    "output_hidden_states": false,
+    "patch_size": 16,
+    "pooling_kernel_size": 3,
+    "position_embedding_size": 10240,
+    "problem_type": null,
+    "return_dict": true,
+    "rms_norm_eps": 1e-6,
+    "rope_parameters": {
+      "rope_theta": 100.0,
+      "rope_type": "default"
+    },
+    "standardize": true,
+    "use_clipped_linears": false
+  },
+  "vision_soft_tokens_per_image": 280
+}

generation_config.json

@@ -0,0 +1,14 @@
+{
+  "bos_token_id": 2,
+  "do_sample": true,
+  "eos_token_id": [
+    1,
+    106,
+    50
+  ],
+  "pad_token_id": 0,
+  "temperature": 1.0,
+  "top_k": 64,
+  "top_p": 0.95,
+  "transformers_version": "5.5.0.dev0"
+}

model-00001-of-00005.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:73d0aa42597869dc0cae5460a273c0ba01553777c137964fbc4601ef5aaddf13
+size 5331846544

model-00002-of-00005.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:89ad855a563d73742a87ff3ceebcf805fc2606760d280b97aa485c68a95037f6
+size 5337954313

model-00003-of-00005.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:9c7705486b748092520affad0bfefbdd6cd99ec6ede5549e8caac98cae7754d8
+size 5323503637

model-00004-of-00005.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:db24d5ef7fd45f9b97b39fc29b118b506404c69374be9f8434aebd22e0c404dc
+size 5359114625

model-00005-of-00005.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:76e8f260c25c72c909d6f9d9cbc21a1692b21ad09c00d3fb44813f9538b6dc25
+size 3625684634

processor_config.json

@@ -0,0 +1,75 @@
+{
+  "audio_ms_per_token": 40,
+  "audio_seq_length": 750,
+  "feature_extractor": {
+    "dither": 0.0,
+    "feature_extractor_type": "Gemma4AudioFeatureExtractor",
+    "feature_size": 128,
+    "fft_length": 512,
+    "fft_overdrive": false,
+    "frame_length": 320,
+    "hop_length": 160,
+    "input_scale_factor": 1.0,
+    "max_frequency": 8000.0,
+    "mel_floor": 0.001,
+    "min_frequency": 0.0,
+    "padding_side": "right",
+    "padding_value": 0.0,
+    "per_bin_mean": null,
+    "per_bin_stddev": null,
+    "preemphasis": 0.0,
+    "preemphasis_htk_flavor": true,
+    "return_attention_mask": true,
+    "sampling_rate": 16000
+  },
+  "image_processor": {
+    "do_convert_rgb": true,
+    "do_normalize": false,
+    "do_rescale": true,
+    "do_resize": true,
+    "image_mean": [
+      0.0,
+      0.0,
+      0.0
+    ],
+    "image_processor_type": "Gemma4ImageProcessor",
+    "image_seq_length": 280,
+    "image_std": [
+      1.0,
+      1.0,
+      1.0
+    ],
+    "max_soft_tokens": 280,
+    "patch_size": 16,
+    "pooling_kernel_size": 3,
+    "resample": 3,
+    "rescale_factor": 0.00392156862745098
+  },
+  "image_seq_length": 280,
+  "processor_class": "Gemma4Processor",
+  "video_processor": {
+    "do_convert_rgb": true,
+    "do_normalize": true,
+    "do_rescale": true,
+    "do_resize": true,
+    "do_sample_frames": true,
+    "image_mean": [
+      0.0,
+      0.0,
+      0.0
+    ],
+    "image_std": [
+      1.0,
+      1.0,
+      1.0
+    ],
+    "max_soft_tokens": 70,
+    "num_frames": 32,
+    "patch_size": 16,
+    "pooling_kernel_size": 3,
+    "resample": 3,
+    "rescale_factor": 0.00392156862745098,
+    "return_metadata": false,
+    "video_processor_type": "Gemma4VideoProcessor"
+  }
+}

tokenizer.json

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cc8d3a0ce36466ccc1278bf987df5f71db1719b9ca6b4118264f45cb627bfe0f
+size 32169626

tokenizer_config.json

@@ -0,0 +1,74 @@
+{
+  "audio_token": "<|audio|>",
+  "backend": "tokenizers",
+  "boa_token": "<|audio>",
+  "boi_token": "<|image>",
+  "bos_token": "<bos>",
+  "eoa_token": "<audio|>",
+  "eoc_token": "<channel|>",
+  "eoi_token": "<image|>",
+  "eos_token": "<eos>",
+  "eot_token": "<turn|>",
+  "escape_token": "<|\"|>",
+  "etc_token": "<tool_call|>",
+  "etd_token": "<tool|>",
+  "etr_token": "<tool_response|>",
+  "extra_special_tokens": [
+    "<|video|>"
+  ],
+  "image_token": "<|image|>",
+  "mask_token": "<mask>",
+  "model_max_length": 1000000000000000019884624838656,
+  "pad_token": "<pad>",
+  "padding_side": "left",
+  "processor_class": "Gemma4Processor",
+  "response_schema": {
+    "type": "object",
+    "properties": {
+      "role": {
+        "const": "assistant"
+      },
+      "thinking": {
+        "type": "string"
+      },
+      "content": {
+        "type": "string"
+      },
+      "tool_calls": {
+        "x-regex-iterator": "<\\|tool_call>(.*?)<tool_call\\|>",
+        "type": "array",
+        "items": {
+          "type": "object",
+          "properties": {
+            "type": {
+              "const": "function"
+            },
+            "function": {
+              "type": "object",
+              "x-regex": "call\\:(?P<name>\\w+)(?P<arguments>\\{.*\\})",
+              "properties": {
+                "name": {
+                  "type": "string"
+                },
+                "arguments": {
+                  "type": "object",
+                  "x-parser": "gemma4-tool-call",
+                  "additionalProperties": {}
+                }
+              }
+            }
+          }
+        }
+      }
+    },
+    "x-regex": "(\\<\\|channel\\>thought\\n(?P<thinking>.*?)\\<channel\\|\\>)?(?P<tool_calls>\\<\\|tool_call\\>.*\\<tool_call\\|\\>)?(?P<content>(?:(?!\\<turn\\|\\>)(?!\\<\\|tool_response\\>).)+)?(?:\\<turn\\|\\>|\\<\\|tool_response\\>)?"
+  },
+  "soc_token": "<|channel>",
+  "sot_token": "<|turn>",
+  "stc_token": "<|tool_call>",
+  "std_token": "<|tool>",
+  "str_token": "<|tool_response>",
+  "think_token": "<|think|>",
+  "tokenizer_class": "GemmaTokenizer",
+  "unk_token": "<unk>"
+}