Add MLX quantized model with KV cache compression

.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,102 @@
+---
+base_model: google/gemma-4-26B-A4B-it
+library_name: mlx
+tags:
+  - rotorquant
+  - kv-cache-quantization
+  - gemma
+  - gemma4
+  - multimodal
+  - quantized
+  - mlx
+  - 4bit
+license: apache-2.0
+pipeline_tag: image-text-to-text
+---
+
+# Gemma 4 26B-A4B-it - RotorQuant MLX 4-bit
+
+**4-bit weight-quantized MLX version** of [google/gemma-4-26B-A4B-it](https://huggingface.co/google/gemma-4-26B-A4B-it) with RotorQuant KV-cache quantization. Optimized for Apple Silicon inference via the [MLX](https://github.com/ml-explore/mlx) framework. RotorQuant delivers 5.3x faster prefill and 28% faster decode compared to TurboQuant. A good balance between model quality and memory efficiency. Only 4B parameters are active per token despite 26B total, making this model significantly more efficient at inference time than its parameter count suggests.
+
+Approximate model size: **~14 GB**
+
+## Model Specifications
+
+| Property | Value |
+|---|---|
+| **Base Model** | [google/gemma-4-26B-A4B-it](https://huggingface.co/google/gemma-4-26B-A4B-it) |
+| **Parameters** | 26 billion total (4 billion active per token) |
+| **Architecture** | Mixture-of-Experts (MoE) (4B active per token) |
+| **Modality** | Multimodal: image + text input, text output |
+| **License** | Apache 2.0 |
+| **Weight Quantization** | 4-bit (~14 GB) |
+| **KV-Cache Quantization** | RotorQuant |
+| **Framework** | MLX (Apple Silicon) |
+
+## Quickstart
+
+```python
+import mlx.core as mx
+from mlx_lm import load, generate
+
+model, tokenizer = load("majentik/gemma-4-26B-A4B-it-RotorQuant-MLX-4bit")
+
+prompt = "Describe this image in detail."
+response = generate(model, tokenizer, prompt=prompt, max_tokens=512)
+print(response)
+```
+
+For multimodal usage with images:
+
+```python
+from mlx_vlm import load, generate
+
+model, processor = load("majentik/gemma-4-26B-A4B-it-RotorQuant-MLX-4bit")
+
+prompt = "What do you see in this image?"
+output = generate(model, processor, prompt=prompt, image="path/to/image.jpg", max_tokens=512)
+print(output)
+```
+
+## What is RotorQuant?
+
+[RotorQuant](https://github.com/scrya-com/rotorquant) is a high-performance KV-cache quantization method that achieves significantly better throughput than TurboQuant. Combined with 4-bit weight quantization in MLX, this provides a dual compression strategy with superior KV-cache performance: smaller model weights plus faster compressed KV cache for efficient long-context generation.
+
+Key advantages over TurboQuant:
+- **5.3x faster prefill**
+- **28% faster decode**
+- Equivalent memory savings
+
+## KV-Cache Quantization Comparison
+
+| Method | Prefill Speed | Decode Speed | Memory Savings | Reference |
+|---|---|---|---|---|
+| **TurboQuant** | 1x (baseline) | 1x (baseline) | High | [arXiv: 2504.19874](https://arxiv.org/abs/2504.19874) |
+| **RotorQuant** | **5.3x faster** | **28% faster** | High | [GitHub](https://github.com/scrya-com/rotorquant) |
+
+## Memory Estimates (Gemma 4 26B-A4B-it)
+
+| Precision | Approximate Size | MLX Variant |
+|---|---|---|
+| FP16 (original) | ~52 GB | -- |
+| 8-bit quantized | ~26 GB | [RotorQuant-MLX-8bit](https://huggingface.co/majentik/gemma-4-26B-A4B-it-RotorQuant-MLX-8bit) |
+| **4-bit quantized** | **~14 GB** | **This model** |
+| 2-bit quantized | ~7 GB | [RotorQuant-MLX-2bit](https://huggingface.co/majentik/gemma-4-26B-A4B-it-RotorQuant-MLX-2bit) |
+
+## Hardware Requirements
+
+This model requires approximately 14 GB of unified memory. Recommended hardware:
+- Apple M2 Pro (24 GB+)
+- Apple M3 Pro (24 GB+)
+- Apple M4 Pro (24 GB+)
+- Any Apple Silicon Mac with 24 GB+ unified memory
+
+## See Also
+
+- [google/gemma-4-26B-A4B-it](https://huggingface.co/google/gemma-4-26B-A4B-it) -- Base model
+- [majentik/gemma-4-26B-A4B-it-RotorQuant](https://huggingface.co/majentik/gemma-4-26B-A4B-it-RotorQuant) -- RotorQuant KV-cache only (transformers)
+- [majentik/gemma-4-26B-A4B-it-RotorQuant-MLX-8bit](https://huggingface.co/majentik/gemma-4-26B-A4B-it-RotorQuant-MLX-8bit) -- MLX 8-bit variant
+- [majentik/gemma-4-26B-A4B-it-RotorQuant-MLX-2bit](https://huggingface.co/majentik/gemma-4-26B-A4B-it-RotorQuant-MLX-2bit) -- MLX 2-bit variant
+- [majentik/gemma-4-26B-A4B-it-TurboQuant-MLX-4bit](https://huggingface.co/majentik/gemma-4-26B-A4B-it-TurboQuant-MLX-4bit) -- TurboQuant MLX 4-bit variant
+- [RotorQuant GitHub](https://github.com/scrya-com/rotorquant)
+- [MLX Framework](https://github.com/ml-explore/mlx)

chat_template.jinja

@@ -0,0 +1,266 @@
+{%- macro format_parameters(properties, required) -%}
+    {%- 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 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['nullable'] %}
+                {%- if add_comma %},{%- else -%} {%- set add_comma = true -%} {% endif -%}
+                nullable: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 == 'OBJECT' -%}
+                ,properties:{
+                {%- if value['properties'] is defined and value['properties'] is mapping -%}
+                    {{- format_parameters(value['properties'], value['required'] | default([])) -}}
+                {%- elif value is mapping -%}
+                    {{- format_parameters(value, value['required'] | default([])) -}}
+                {%- endif -%}
+                }
+                {%- if value['required'] -%}
+                    ,required:[
+                    {%- for item in value['required'] | default([]) -%}
+                        <|"|>{{- item -}}<|"|>
+                        {%- if not loop.last %},{% endif -%}
+                    {%- endfor -%}
+                    ]
+                {%- endif -%}
+            {%- elif value['type'] | upper == 'ARRAY' -%}
+                {%- if value['items'] is mapping and value['items'] -%}
+                    ,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 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 -%}
+
+{%- 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|>' -}}
+        {%- set ns.prev_message_type = 'think' -%}
+    {%- endif -%}
+
+    {%- if messages[0]['role'] in ['system', 'developer'] -%}
+        {{- messages[0]['content'] | trim -}}
+        {%- 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 %}
+
+{#- Loop through messages -#}
+{%- for message in loop_messages -%}
+    {%- set ns.prev_message_type = None -%}
+    {%- set role = 'model' if message['role'] == 'assistant' else message['role'] -%}
+        {{- '<|turn>' + role + '\n' }}
+
+            {%- 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 -%}
+
+            {%- if message['tool_responses'] -%}
+                {#- Tool Response handling -#}
+                {%- for tool_response in message['tool_responses'] -%}
+                    {{- '<|tool_response>' -}}
+                    {%- if tool_response['response'] is mapping -%}
+                        {{- 'response:' + tool_response['name'] | default('unknown') + '{' -}}
+                        {%- for key, value in tool_response['response'] | dictsort -%}
+                            {{- key -}}:{{- format_argument(value, escape_keys=False) -}}
+                            {%- if not loop.last %},{% endif -%}
+                        {%- endfor -%}
+                        {{- '}' -}}
+                    {%- else -%}
+                        {{- 'response:' + tool_response['name'] | default('unknown') + '{value:' + format_argument(tool_response['response'], escape_keys=False) + '}' -}}
+                    {%- endif -%}
+                    {{- '<tool_response|>' -}}
+                {%- endfor -%}
+                {%- set ns.prev_message_type = 'tool_response' -%}
+            {%- endif -%}
+
+            {%- 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' -%}
+                        {{- '\n\n<|image|>\n\n' -}}
+                        {%- set ns.prev_message_type = 'image' -%}
+                    {%- elif item['type'] == 'audio' -%}
+                        {{- '<|audio|>' -}}
+                        {%- set ns.prev_message_type = 'audio' -%}
+                    {%- elif item['type'] == 'video' -%}
+                        {{- '\n\n<|video|>\n\n' -}}
+                        {%- set ns.prev_message_type = 'video' -%}
+                    {%- endif -%}
+                {%- endfor -%}
+            {%- endif -%}
+
+        {%- if not (message['tool_responses'] and not message['content']) -%}
+            {{- '<turn|>\n' -}}
+        {%- endif -%}
+{%- endfor -%}
+
+{%- if add_generation_prompt -%}
+    {%- if ns.prev_message_type != 'tool_response' -%}
+        {{- '<|turn>model\n' -}}
+    {%- endif -%}
+    {%- if not enable_thinking | default(false) -%}
+        {{- '<|channel>thought\n<channel|>' -}}
+    {%- endif -%}
+{%- endif -%}

config.json

@@ -0,0 +1,1117 @@
+{
+    "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,
+        50
+    ],
+    "image_token_id": 258880,
+    "initializer_range": 0.02,
+    "model_type": "gemma4",
+    "quantization": {
+        "group_size": 64,
+        "bits": 4,
+        "mode": "affine",
+        "language_model.model.layers.0.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.0.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.0.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.0.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.1.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.1.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.1.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.1.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.2.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.2.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.2.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.2.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.3.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.3.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.3.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.3.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.4.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.4.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.4.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.4.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.5.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.5.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.5.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.5.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.6.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.6.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.6.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.6.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.7.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.7.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.7.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.7.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.8.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.8.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.8.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.8.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.9.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.9.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.9.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.9.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.10.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.10.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.10.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.10.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.11.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.11.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.11.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.11.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.12.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.12.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.12.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.12.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.13.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.13.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.13.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.13.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.14.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.14.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.14.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.14.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.15.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.15.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.15.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.15.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.16.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.16.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.16.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.16.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.17.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.17.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.17.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.17.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.18.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.18.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.18.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.18.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.19.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.19.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.19.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.19.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.20.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.20.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.20.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.20.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.21.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.21.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.21.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.21.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.22.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.22.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.22.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.22.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.23.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.23.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.23.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.23.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.24.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.24.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.24.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.24.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.25.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.25.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.25.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.25.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.26.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.26.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.26.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.26.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.27.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.27.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.27.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.27.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.28.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.28.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.28.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.28.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.29.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.29.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.29.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.29.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        }
+    },
+    "quantization_config": {
+        "group_size": 64,
+        "bits": 4,
+        "mode": "affine",
+        "language_model.model.layers.0.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.0.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.0.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.0.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.1.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.1.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.1.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.1.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.2.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.2.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.2.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.2.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.3.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.3.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.3.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.3.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.4.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.4.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.4.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.4.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.5.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.5.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.5.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.5.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.6.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.6.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.6.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.6.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.7.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.7.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.7.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.7.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.8.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.8.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.8.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.8.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.9.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.9.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.9.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.9.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.10.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.10.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.10.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.10.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.11.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.11.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.11.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.11.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.12.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.12.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.12.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.12.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.13.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.13.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.13.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.13.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.14.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.14.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.14.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.14.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.15.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.15.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.15.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.15.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.16.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.16.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.16.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.16.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.17.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.17.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.17.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.17.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.18.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.18.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.18.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.18.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.19.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.19.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.19.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.19.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.20.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.20.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.20.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.20.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.21.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.21.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.21.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.21.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.22.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.22.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.22.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.22.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.23.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.23.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.23.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.23.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.24.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.24.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.24.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.24.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.25.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.25.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.25.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.25.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.26.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.26.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.26.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.26.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.27.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.27.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.27.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.27.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.28.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.28.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.28.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.28.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.29.mlp.gate_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.29.mlp.down_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.29.mlp.up_proj": {
+            "group_size": 64,
+            "bits": 8
+        },
+        "language_model.model.layers.29.router.proj": {
+            "group_size": 64,
+            "bits": 8
+        }
+    },
+    "text_config": {
+        "attention_bias": false,
+        "attention_dropout": 0.0,
+        "attention_k_eq_v": true,
+        "bos_token_id": 2,
+        "dtype": "bfloat16",
+        "enable_moe_block": true,
+        "eos_token_id": 1,
+        "final_logit_softcapping": 30.0,
+        "global_head_dim": 512,
+        "head_dim": 256,
+        "hidden_activation": "gelu_pytorch_tanh",
+        "hidden_size": 2816,
+        "hidden_size_per_layer_input": 0,
+        "initializer_range": 0.02,
+        "intermediate_size": 2112,
+        "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"
+        ],
+        "max_position_embeddings": 262144,
+        "model_type": "gemma4_text",
+        "moe_intermediate_size": 704,
+        "num_attention_heads": 16,
+        "num_experts": 128,
+        "num_global_key_value_heads": 2,
+        "num_hidden_layers": 30,
+        "num_key_value_heads": 8,
+        "num_kv_shared_layers": 0,
+        "pad_token_id": 0,
+        "rms_norm_eps": 1e-06,
+        "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": 8,
+        "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-06,
+        "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-00003.safetensors

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

model-00002-of-00003.safetensors

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

model-00003-of-00003.safetensors

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

processor_config.json

@@ -0,0 +1,32 @@
+{
+  "audio_seq_length": 750,
+  "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,
+    "size": {
+      "height": 224,
+      "width": 224
+    }
+  },
+  "image_seq_length": 280,
+  "processor_class": "Gemma4Processor"
+}

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,95 @@
+{
+  "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|>",
+  "is_local": true,
+  "mask_token": "<mask>",
+  "model_max_length": 1000000000000000019884624838656,
+  "model_specific_special_tokens": {
+    "audio_token": "<|audio|>",
+    "boa_token": "<|audio>",
+    "boi_token": "<|image>",
+    "eoa_token": "<audio|>",
+    "eoc_token": "<channel|>",
+    "eoi_token": "<image|>",
+    "eot_token": "<turn|>",
+    "escape_token": "<|\"|>",
+    "etc_token": "<tool_call|>",
+    "etd_token": "<tool|>",
+    "etr_token": "<tool_response|>",
+    "image_token": "<|image|>",
+    "soc_token": "<|channel>",
+    "sot_token": "<|turn>",
+    "stc_token": "<|tool_call>",
+    "std_token": "<|tool>",
+    "str_token": "<|tool_response>",
+    "think_token": "<|think|>"
+  },
+  "pad_token": "<pad>",
+  "padding_side": "left",
+  "processor_class": "Gemma4Processor",
+  "response_schema": {
+    "properties": {
+      "content": {
+        "type": "string"
+      },
+      "role": {
+        "const": "assistant"
+      },
+      "thinking": {
+        "type": "string"
+      },
+      "tool_calls": {
+        "items": {
+          "properties": {
+            "function": {
+              "properties": {
+                "arguments": {
+                  "additionalProperties": {},
+                  "type": "object",
+                  "x-parser": "gemma4-tool-call"
+                },
+                "name": {
+                  "type": "string"
+                }
+              },
+              "type": "object",
+              "x-regex": "call\\:(?P<name>\\w+)(?P<arguments>\\{.*\\})"
+            },
+            "type": {
+              "const": "function"
+            }
+          },
+          "type": "object"
+        },
+        "type": "array",
+        "x-regex-iterator": "<\\|tool_call>(.*?)<tool_call\\|>"
+      }
+    },
+    "type": "object",
+    "x-regex": "(\\<\\|channel\\>thought\\n(?P<thinking>.*?)\\<channel\\|\\>)?(?P<content>(?:(?!\\<\\|tool_call\\>)(?!\\<turn\\|\\>).)+)?(?P<tool_calls>\\<\\|tool_call\\>.*\\<tool_call\\|\\>)?(?:\\<turn\\|\\>)?"
+  },
+  "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>"
+}