Instructions to use AutomatosX/AX-Qwen3.5-9B-MLX-AXQ-4bit-MTP with libraries, inference providers, notebooks, and local apps. Follow these links to get started.

Libraries

How to use AutomatosX/AX-Qwen3.5-9B-MLX-AXQ-4bit-MTP with MLX:

# Make sure mlx-lm is installed
# pip install --upgrade mlx-lm

# Generate text with mlx-lm
from mlx_lm import load, generate

model, tokenizer = load("AutomatosX/AX-Qwen3.5-9B-MLX-AXQ-4bit-MTP")

prompt = "Write a story about Einstein"
messages = [{"role": "user", "content": prompt}]
prompt = tokenizer.apply_chat_template(
    messages, add_generation_prompt=True
)

text = generate(model, tokenizer, prompt=prompt, verbose=True)

Notebooks
Google Colab
Kaggle
Local Apps Settings
LM Studio

How to use AutomatosX/AX-Qwen3.5-9B-MLX-AXQ-4bit-MTP with Pi:

Start the MLX server

# Install MLX LM:
uv tool install mlx-lm
# Start a local OpenAI-compatible server:
mlx_lm.server --model "AutomatosX/AX-Qwen3.5-9B-MLX-AXQ-4bit-MTP"

Configure the model in Pi

# Install Pi:
npm install -g @mariozechner/pi-coding-agent
# Add to ~/.pi/agent/models.json:
{
  "providers": {
    "mlx-lm": {
      "baseUrl": "http://localhost:8080/v1",
      "api": "openai-completions",
      "apiKey": "none",
      "models": [
        {
          "id": "AutomatosX/AX-Qwen3.5-9B-MLX-AXQ-4bit-MTP"
        }
      ]
    }
  }
}

Run Pi

# Start Pi in your project directory:
pi

Hermes Agent new

How to use AutomatosX/AX-Qwen3.5-9B-MLX-AXQ-4bit-MTP with Hermes Agent:

Start the MLX server

# Install MLX LM:
uv tool install mlx-lm
# Start a local OpenAI-compatible server:
mlx_lm.server --model "AutomatosX/AX-Qwen3.5-9B-MLX-AXQ-4bit-MTP"

Configure Hermes

# Install Hermes:
curl -fsSL https://hermes-agent.nousresearch.com/install.sh | bash
hermes setup
# Point Hermes at the local server:
hermes config set model.provider custom
hermes config set model.base_url http://127.0.0.1:8080/v1
hermes config set model.default AutomatosX/AX-Qwen3.5-9B-MLX-AXQ-4bit-MTP

Run Hermes

hermes

OpenClaw new

How to use AutomatosX/AX-Qwen3.5-9B-MLX-AXQ-4bit-MTP with OpenClaw:

Start the MLX server

# Install MLX LM:
uv tool install mlx-lm
# Start a local OpenAI-compatible server:
mlx_lm.server --model "AutomatosX/AX-Qwen3.5-9B-MLX-AXQ-4bit-MTP"

Configure OpenClaw

# Install OpenClaw:
npm install -g openclaw@latest
# Register the local server and set it as the default model:
openclaw onboard --non-interactive --mode local \
  --auth-choice custom-api-key \
  --custom-base-url http://127.0.0.1:8080/v1 \
  --custom-model-id "AutomatosX/AX-Qwen3.5-9B-MLX-AXQ-4bit-MTP" \
  --custom-provider-id mlx-lm \
  --custom-compatibility openai \
  --custom-text-input \
  --accept-risk \
  --skip-health

Run OpenClaw

openclaw agent --local --agent main --message "Hello from Hugging Face"

MLX LM

How to use AutomatosX/AX-Qwen3.5-9B-MLX-AXQ-4bit-MTP with MLX LM:

Generate or start a chat session

# Install MLX LM
uv tool install mlx-lm
# Interactive chat REPL
mlx_lm.chat --model "AutomatosX/AX-Qwen3.5-9B-MLX-AXQ-4bit-MTP"

Run an OpenAI-compatible server

# Install MLX LM
uv tool install mlx-lm
# Start the server
mlx_lm.server --model "AutomatosX/AX-Qwen3.5-9B-MLX-AXQ-4bit-MTP"
# Calling the OpenAI-compatible server with curl
curl -X POST "http://localhost:8000/v1/chat/completions" \
   -H "Content-Type: application/json" \
   --data '{
     "model": "AutomatosX/AX-Qwen3.5-9B-MLX-AXQ-4bit-MTP",
     "messages": [
       {"role": "user", "content": "Hello"}
     ]
   }'

AutomatosX commited on 2 days ago

Commit

8d999aa

verified ·

1 Parent(s): 522d2cc

Add development AXQ pack AX-Qwen3.5-9B-MLX-AXQ-4bit-MTP

Browse files

Files changed (19) hide show

.gitattributes +1 -0
README.md +38 -0
axquant_manifest.json +211 -0
axquant_mtp_sidecar_manifest.json +40 -0
axquant_plan.json +0 -0
axquant_quantizer_execution.json +2497 -0
axquant_runtime.json +58 -0
axquant_vision_sidecar_manifest.json +30 -0
chat_template.jinja +154 -0
config.json +2603 -0
model-00001-of-00002.safetensors +3 -0
model-00002-of-00002.safetensors +3 -0
model-manifest.json +0 -0
model.safetensors.index.json +933 -0
mtp.safetensors +3 -0
mtplx_runtime.json +5 -0
tokenizer.json +3 -0
tokenizer_config.json +33 -0
vision.safetensors +3 -0

.gitattributes CHANGED Viewed

@@ -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

 *.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 ADDED Viewed

	@@ -0,0 +1,38 @@

+---
+license: apache-2.0
+library_name: mlx
+tags:
+- mlx
+- axquant
+- axq
+- development
+pipeline_tag: text-generation
+---
+# AX-Qwen3.5-9B-MLX-AXQ-4bit-MTP
+**Development** AXQuant (AXQ) MLX pack from BF16 via `axquant quantize` (architecture-prior ladder). **Not** a certified release.
+| Field | Value |
+| --- | --- |
+| Source | `Qwen/Qwen3.5-9B` |
+| Measured total BPW | `6.970133429722802` |
+| MTP present | `True` |
+| Runtime smoke (mlx-lm) | `True` |
+| AX Engine model-manifest | `True` |
+| Evidence | `architecture_prior` |
+## Load (MLX-LM)
+```python
+from mlx_lm import load, generate
+model, tokenizer = load("AutomatosX/AX-Qwen3.5-9B-MLX-AXQ-4bit-MTP")
+print(generate(model, tokenizer, prompt="Hello", max_tokens=64))
+```
+## Notes
+- Product naming: `AX-<Base>-MLX-AXQ-<4bit|6bit>[-MTP]`
+- Protection floors may raise realized BPW above the requested target (especially small / heavily protected models).
+- Nemotron thin support may not generate `model-manifest.json` for AX Engine yet; MLX-LM path still applies.
+- Formal certification requires measured analyze → plan → validate → release-audit.

axquant_manifest.json ADDED Viewed

	@@ -0,0 +1,211 @@

+{
+  "axquant_version": "1.0.0",
+  "calibration": null,
+  "created_at": "2026-08-02T18:28:44.200633Z",
+  "effective_bpw": 6.969999936501257,
+  "files": [
+    {
+      "path": "README.md",
+      "sha256": "748c964f9b7e5f2c3770ce013bbc0153be7c54311d9692c343b6188eefe77ac6",
+      "size_bytes": 81
+    },
+    {
+      "path": "axquant_mtp_sidecar_manifest.json",
+      "sha256": "7a6fc0579e76a524c01ac3909aa8108407742279a99efcfe426288dd0367c01c",
+      "size_bytes": 1279
+    },
+    {
+      "path": "axquant_plan.json",
+      "sha256": "fabfd1d0610d8f88315d3c9135660151fd8d858b78117558d4be4bde922122b7",
+      "size_bytes": 704568
+    },
+    {
+      "path": "axquant_quantizer_execution.json",
+      "sha256": "7073c44f0c00cad3b61090eb30d3fee34908684cb41bf7a54035e88b3b7179d3",
+      "size_bytes": 65128
+    },
+    {
+      "path": "axquant_runtime.json",
+      "sha256": "2aa322fc1c43804edfb206ed1c4c421c22d36110b3b4cdc014de30ed77dfd7b5",
+      "size_bytes": 1676
+    },
+    {
+      "path": "axquant_vision_sidecar_manifest.json",
+      "sha256": "0f1cbd95e06db2b9690526b12bbf517910a4695a69335efcecd227c1516e9ac4",
+      "size_bytes": 906
+    },
+    {
+      "path": "chat_template.jinja",
+      "sha256": "a4aee8afcf2e0711942cf848899be66016f8d14a889ff9ede07bca099c28f715",
+      "size_bytes": 7756
+    },
+    {
+      "path": "config.json",
+      "sha256": "dbd13f7d669f728dd45c664130da64f2b5a8e038b05710def8aa24d2aaf4b02f",
+      "size_bytes": 66933
+    },
+    {
+      "path": "model-00001-of-00002.safetensors",
+      "sha256": "c7b13cac43465c53c5a8133519e9b819a2021b58ab36cbbd5d344a655598b8a8",
+      "size_bytes": 4977549600
+    },
+    {
+      "path": "model-00002-of-00002.safetensors",
+      "sha256": "2da4d602a9e856abed386491e3eeba9abe41aec5fde0719e66e40d1ea61008d2",
+      "size_bytes": 2034237582
+    },
+    {
+      "path": "model.safetensors.index.json",
+      "sha256": "f2d488b1f31408d4c34c930027bc68c5ff5279abdbaa4cccb6a82ef311569e71",
+      "size_bytes": 94945
+    },
+    {
+      "path": "mtp.safetensors",
+      "sha256": "105d9d7ae9ff7c7c538df915495c198bb784feea30521e848b3791223c703f06",
+      "size_bytes": 486582944
+    },
+    {
+      "path": "mtplx_runtime.json",
+      "sha256": "119b930ed4b470a08008aa2a51b4a83162907a85b8455f559cffa03e33236ed6",
+      "size_bytes": 108
+    },
+    {
+      "path": "tokenizer.json",
+      "sha256": "06b9509352d2af50381ab2247e083b80d32d5c0aba91c272ca9ff729b6a0e523",
+      "size_bytes": 19989325
+    },
+    {
+      "path": "tokenizer_config.json",
+      "sha256": "95c557768e6b88a7128befc7bfd3c7de50e5d51af9b8b33a9f4dee0e04f99679",
+      "size_bytes": 1161
+    },
+    {
+      "path": "vision.safetensors",
+      "sha256": "95424a2beb8065e493985eb0d50aef3cf20ff17694172ad9f44047c9e0f6107d",
+      "size_bytes": 912058056
+    }
+  ],
+  "format": "mlx",
+  "logical_parameters": 9653104368,
+  "main_logical_parameters": 9409813744,
+  "main_weight_file_size_bytes": 7923845238,
+  "measured_main_bpw": 6.736664893544783,
+  "measured_total_bpw": 6.970133429722802,
+  "mtp_acceptance_retention": null,
+  "mtp_distribution": {
+    "bf16": {
+      "fraction": 1.0,
+      "parameters": 243290624
+    }
+  },
+  "mtp_measured_speedup": null,
+  "mtp_policy": {
+    "candidate_bits": [
+      8,
+      16
+    ],
+    "min_bits": 8,
+    "mode": "protected",
+    "optimize_for_acceptance": true,
+    "preserve_external_sidecar": true,
+    "protect_norms": true,
+    "protect_output_head": true
+  },
+  "mtp_present": true,
+  "mtp_weight_file_size_bytes": 486582944,
+  "plan_sha256": "18a2368d8737f0a12ef7eb9d9432dcf9caa4d9ce09570e704c7bd8f87d3821d6",
+  "profile": "general",
+  "protected_weight_file_size_bytes": 912058056,
+  "quantizer": "axquant",
+  "runtime": {
+    "ax_engine": {
+      "decode_kernel": null,
+      "fused_mtp": null,
+      "kernel_evidence": "unmeasured",
+      "model_manifest": "model-manifest.json",
+      "preferred_group_size": 32
+    },
+    "compatible_runtimes": [
+      {
+        "compatibility_level": "B",
+        "manifest": "config.json",
+        "mtp_support": "runtime-dependent",
+        "name": "mlx-lm",
+        "notes": [
+          "Standard backbone inference is the compatibility target.",
+          "AXQuant MTP metadata may be ignored by MLX-LM."
+        ],
+        "standard_inference": true,
+        "standard_mlx_weights": true,
+        "support_level": "standard-inference"
+      }
+    ],
+    "created_at": "2026-08-02T18:28:40.264399Z",
+    "kv_cache": null,
+    "memory_policy": {
+      "kv_cache_precision": "runtime-default",
+      "mtp_buffers": "preallocate-when-enabled",
+      "prefix_cache": "runtime-managed",
+      "unified_memory_safety_margin": "benchmark-required"
+    },
+    "mtp": {
+      "acceptance_retention": null,
+      "detected": true,
+      "draft_tokens": 1,
+      "enabled_by_default": true,
+      "head_precision": null,
+      "measured_speedup": null,
+      "optimized": false,
+      "recommended_temperature_max": null,
+      "sidecar_file": "mtp.safetensors",
+      "verification_mode": "runtime-default"
+    },
+    "optimization_scope": "text-path",
+    "primary_runtime": {
+      "compatibility_level": "A",
+      "manifest": "model-manifest.json",
+      "mtp_support": "native",
+      "name": "ax-engine",
+      "notes": [
+        "Runtime claims require a passing AX Engine doctor and benchmark report."
+      ],
+      "standard_inference": true,
+      "standard_mlx_weights": true,
+      "support_level": "optimized"
+    },
+    "schema_version": "axquant.runtime.v1"
+  },
+  "schema_version": "axquant.artifact.v2",
+  "software_versions": {
+    "ax_engine": null,
+    "axquant": "1.0.0",
+    "mlx": "0.32.0",
+    "mlx_lm": "0.31.3",
+    "pydantic": "2.13.4",
+    "python": "3.13.14",
+    "safetensors": "0.8.0"
+  },
+  "source_model": {
+    "architecture": "Qwen3_5ForConditionalGeneration",
+    "format": "mlx",
+    "local_path": "/Volumes/Ext4T-02/axquant/models/Qwen3.5-9B-bf16",
+    "model_id": "Qwen/Qwen3.5-9B",
+    "revision": null
+  },
+  "target_class": "4bit",
+  "weight_distribution": {
+    "4bit": {
+      "fraction": 0.7167129023213312,
+      "parameters": 6918504448
+    },
+    "8bit": {
+      "fraction": 0.10536700746463948,
+      "parameters": 1017118720
+    },
+    "bf16": {
+      "fraction": 0.17792009021402927,
+      "parameters": 1717481200
+    }
+  },
+  "weight_file_size_bytes": 8410428182
+}

axquant_mtp_sidecar_manifest.json ADDED Viewed

	@@ -0,0 +1,40 @@

+{
+  "created_at": "2026-08-02T18:28:40.256583Z",
+  "dtypes": [
+    "BF16"
+  ],
+  "output": {
+    "path": "mtp.safetensors",
+    "sha256": "105d9d7ae9ff7c7c538df915495c198bb784feea30521e848b3791223c703f06",
+    "size_bytes": 486582944
+  },
+  "parameters": 243290624,
+  "role": "mtp",
+  "schema_version": "axquant.protected-tensor-sidecar.v1",
+  "source_files": [
+    {
+      "path": "model.safetensors-00002-of-00004.safetensors",
+      "sha256": "31c7d7e2dd5d207840b31cc59083c8f4c4718959149e0358c0364052bb9a0330",
+      "size_bytes": 5335161512
+    },
+    {
+      "path": "model.safetensors-00003-of-00004.safetensors",
+      "sha256": "7ec36ba3a4176a44c3c0876ad80c56a2f70c84bf008d82e9501df642f17dadec",
+      "size_bytes": 5368717440
+    },
+    {
+      "path": "model.safetensors-00004-of-00004.safetensors",
+      "sha256": "b62b0c4cd7e44edee103ee8f4fe225f246d5e768e07bfd5f25b63a8aa1fdd0c6",
+      "size_bytes": 3325995712
+    }
+  ],
+  "source_model": {
+    "architecture": "Qwen3_5ForConditionalGeneration",
+    "format": "mlx",
+    "local_path": "/Volumes/Ext4T-02/axquant/models/Qwen3.5-9B-bf16",
+    "model_id": "Qwen/Qwen3.5-9B",
+    "revision": null
+  },
+  "tensor_count": 15,
+  "tensor_names_sha256": "8cadbfe029ec03d26d19a94845830662c4220caa1b9293ab908b76099355217e"
+}

axquant_plan.json ADDED Viewed

The diff for this file is too large to render. See raw diff

axquant_quantizer_execution.json ADDED Viewed

	@@ -0,0 +1,2497 @@

+{
+  "created_at": "2026-08-02T18:28:30.872671Z",
+  "plan_sha256": "18a2368d8737f0a12ef7eb9d9432dcf9caa4d9ce09570e704c7bd8f87d3821d6",
+  "records": [
+    {
+      "bits": 8,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.embed_tokens",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.14.mlp.down_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.14.mlp.gate_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.14.mlp.up_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.15.mlp.down_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.15.mlp.gate_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.15.mlp.up_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.25.mlp.down_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.26.mlp.down_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.26.mlp.gate_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.26.mlp.up_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.27.mlp.down_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.27.mlp.gate_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.0.mlp.down_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.0.mlp.gate_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.0.mlp.up_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.1.mlp.down_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.1.mlp.gate_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.1.mlp.up_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.10.mlp.down_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.10.mlp.gate_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.10.mlp.up_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.11.mlp.down_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.11.mlp.gate_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.11.mlp.up_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.16.mlp.down_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.16.mlp.gate_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.16.mlp.up_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.17.mlp.down_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.17.mlp.gate_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.17.mlp.up_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.18.mlp.down_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.18.mlp.gate_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.18.mlp.up_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.19.mlp.down_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.21.mlp.down_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.21.mlp.gate_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.21.mlp.up_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.22.mlp.down_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.22.mlp.gate_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.22.mlp.up_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.23.mlp.down_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.23.mlp.gate_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.23.mlp.up_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.27.mlp.up_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.3.mlp.down_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.3.mlp.gate_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.3.mlp.up_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.30.mlp.down_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.30.mlp.gate_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.30.mlp.up_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.31.mlp.down_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.31.mlp.gate_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.31.mlp.up_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.4.mlp.down_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.4.mlp.gate_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.4.mlp.up_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.8.mlp.down_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.8.mlp.gate_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.8.mlp.up_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.9.mlp.down_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.9.mlp.gate_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.9.mlp.up_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.0.linear_attn.in_proj_qkv",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.1.linear_attn.in_proj_qkv",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.10.linear_attn.in_proj_qkv",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.11.self_attn.q_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.12.mlp.down_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.12.mlp.gate_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.12.mlp.up_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.13.mlp.down_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.13.mlp.gate_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.13.mlp.up_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.14.linear_attn.in_proj_qkv",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.15.self_attn.q_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.16.linear_attn.in_proj_qkv",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.17.linear_attn.in_proj_qkv",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.18.linear_attn.in_proj_qkv",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.19.mlp.gate_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.19.mlp.up_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.19.self_attn.q_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.2.linear_attn.in_proj_qkv",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.2.mlp.down_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.2.mlp.gate_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.2.mlp.up_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.20.linear_attn.in_proj_qkv",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.20.mlp.down_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.20.mlp.gate_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.20.mlp.up_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.22.linear_attn.in_proj_qkv",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.23.self_attn.q_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.24.linear_attn.in_proj_qkv",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.24.mlp.down_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.24.mlp.gate_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.24.mlp.up_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.25.linear_attn.in_proj_qkv",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.25.mlp.gate_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.25.mlp.up_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.26.linear_attn.in_proj_qkv",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.27.self_attn.q_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.28.mlp.down_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.28.mlp.gate_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.28.mlp.up_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.29.mlp.down_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.29.mlp.gate_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.29.mlp.up_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.3.self_attn.q_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.30.linear_attn.in_proj_qkv",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.31.self_attn.q_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.4.linear_attn.in_proj_qkv",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.5.linear_attn.in_proj_qkv",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.5.mlp.down_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.5.mlp.gate_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.5.mlp.up_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.6.linear_attn.in_proj_qkv",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.6.mlp.down_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.6.mlp.gate_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.6.mlp.up_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.7.mlp.down_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.7.mlp.gate_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.7.mlp.up_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.7.self_attn.q_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.8.linear_attn.in_proj_qkv",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.9.linear_attn.in_proj_qkv",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.0.linear_attn.in_proj_a",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.0.linear_attn.in_proj_b",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.0.linear_attn.in_proj_z",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.0.linear_attn.out_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.1.linear_attn.in_proj_a",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.1.linear_attn.in_proj_b",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.1.linear_attn.in_proj_z",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.1.linear_attn.out_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.10.linear_attn.in_proj_a",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.10.linear_attn.in_proj_b",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.10.linear_attn.in_proj_z",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.10.linear_attn.out_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.11.self_attn.k_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.11.self_attn.o_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.11.self_attn.v_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.12.linear_attn.in_proj_a",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.12.linear_attn.in_proj_b",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.12.linear_attn.in_proj_qkv",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.12.linear_attn.in_proj_z",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.12.linear_attn.out_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.13.linear_attn.in_proj_a",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.13.linear_attn.in_proj_b",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.13.linear_attn.in_proj_qkv",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.13.linear_attn.in_proj_z",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.13.linear_attn.out_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.14.linear_attn.in_proj_a",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.14.linear_attn.in_proj_b",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.14.linear_attn.in_proj_z",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.14.linear_attn.out_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.15.self_attn.k_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.15.self_attn.o_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.15.self_attn.v_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.16.linear_attn.in_proj_a",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.16.linear_attn.in_proj_b",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.16.linear_attn.in_proj_z",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.16.linear_attn.out_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.17.linear_attn.in_proj_a",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.17.linear_attn.in_proj_b",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.17.linear_attn.in_proj_z",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.17.linear_attn.out_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.18.linear_attn.in_proj_a",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.18.linear_attn.in_proj_b",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.18.linear_attn.in_proj_z",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.18.linear_attn.out_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.19.self_attn.k_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.19.self_attn.o_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.19.self_attn.v_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.2.linear_attn.in_proj_a",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.2.linear_attn.in_proj_b",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.2.linear_attn.in_proj_z",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.2.linear_attn.out_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.20.linear_attn.in_proj_a",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.20.linear_attn.in_proj_b",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.20.linear_attn.in_proj_z",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.20.linear_attn.out_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.21.linear_attn.in_proj_a",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.21.linear_attn.in_proj_b",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.21.linear_attn.in_proj_qkv",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.21.linear_attn.in_proj_z",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.21.linear_attn.out_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.22.linear_attn.in_proj_a",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.22.linear_attn.in_proj_b",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.22.linear_attn.in_proj_z",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.22.linear_attn.out_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.23.self_attn.k_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.23.self_attn.o_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.23.self_attn.v_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.24.linear_attn.in_proj_a",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.24.linear_attn.in_proj_b",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.24.linear_attn.in_proj_z",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.24.linear_attn.out_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.25.linear_attn.in_proj_a",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.25.linear_attn.in_proj_b",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.25.linear_attn.in_proj_z",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.25.linear_attn.out_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.26.linear_attn.in_proj_a",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.26.linear_attn.in_proj_b",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.26.linear_attn.in_proj_z",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.26.linear_attn.out_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.27.self_attn.k_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.27.self_attn.o_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.27.self_attn.v_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.28.linear_attn.in_proj_a",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.28.linear_attn.in_proj_b",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.28.linear_attn.in_proj_qkv",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.28.linear_attn.in_proj_z",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.28.linear_attn.out_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.29.linear_attn.in_proj_a",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.29.linear_attn.in_proj_b",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.29.linear_attn.in_proj_qkv",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.29.linear_attn.in_proj_z",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.29.linear_attn.out_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.3.self_attn.k_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.3.self_attn.o_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.3.self_attn.v_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.30.linear_attn.in_proj_a",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.30.linear_attn.in_proj_b",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.30.linear_attn.in_proj_z",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.30.linear_attn.out_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.31.self_attn.k_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.31.self_attn.o_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.31.self_attn.v_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.4.linear_attn.in_proj_a",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.4.linear_attn.in_proj_b",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.4.linear_attn.in_proj_z",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.4.linear_attn.out_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.5.linear_attn.in_proj_a",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.5.linear_attn.in_proj_b",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.5.linear_attn.in_proj_z",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.5.linear_attn.out_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.6.linear_attn.in_proj_a",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.6.linear_attn.in_proj_b",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.6.linear_attn.in_proj_z",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.6.linear_attn.out_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.7.self_attn.k_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.7.self_attn.o_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.7.self_attn.v_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.8.linear_attn.in_proj_a",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.8.linear_attn.in_proj_b",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.8.linear_attn.in_proj_z",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 64,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.8.linear_attn.out_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.9.linear_attn.in_proj_a",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.9.linear_attn.in_proj_b",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.9.linear_attn.in_proj_z",
+      "note": "MLX-LM affine packing",
+      "success": true
+    },
+    {
+      "bits": 4,
+      "fallback": false,
+      "group_size": 32,
+      "metadata": {},
+      "method": "affine",
+      "module_path": "model.language_model.layers.9.linear_attn.out_proj",
+      "note": "MLX-LM affine packing",
+      "success": true
+    }
+  ],
+  "schema_version": "axquant.quantizer-execution.v1"
+}

axquant_runtime.json ADDED Viewed

	@@ -0,0 +1,58 @@

+{
+  "ax_engine": {
+    "decode_kernel": null,
+    "fused_mtp": null,
+    "kernel_evidence": "unmeasured",
+    "model_manifest": "model-manifest.json",
+    "preferred_group_size": 32
+  },
+  "compatible_runtimes": [
+    {
+      "compatibility_level": "B",
+      "manifest": "config.json",
+      "mtp_support": "runtime-dependent",
+      "name": "mlx-lm",
+      "notes": [
+        "Standard backbone inference is the compatibility target.",
+        "AXQuant MTP metadata may be ignored by MLX-LM."
+      ],
+      "standard_inference": true,
+      "standard_mlx_weights": true,
+      "support_level": "standard-inference"
+    }
+  ],
+  "created_at": "2026-08-02T18:28:40.264399Z",
+  "kv_cache": null,
+  "memory_policy": {
+    "kv_cache_precision": "runtime-default",
+    "mtp_buffers": "preallocate-when-enabled",
+    "prefix_cache": "runtime-managed",
+    "unified_memory_safety_margin": "benchmark-required"
+  },
+  "mtp": {
+    "acceptance_retention": null,
+    "detected": true,
+    "draft_tokens": 1,
+    "enabled_by_default": true,
+    "head_precision": null,
+    "measured_speedup": null,
+    "optimized": false,
+    "recommended_temperature_max": null,
+    "sidecar_file": "mtp.safetensors",
+    "verification_mode": "runtime-default"
+  },
+  "optimization_scope": "text-path",
+  "primary_runtime": {
+    "compatibility_level": "A",
+    "manifest": "model-manifest.json",
+    "mtp_support": "native",
+    "name": "ax-engine",
+    "notes": [
+      "Runtime claims require a passing AX Engine doctor and benchmark report."
+    ],
+    "standard_inference": true,
+    "standard_mlx_weights": true,
+    "support_level": "optimized"
+  },
+  "schema_version": "axquant.runtime.v1"
+}

axquant_vision_sidecar_manifest.json ADDED Viewed

	@@ -0,0 +1,30 @@

+{
+  "created_at": "2026-08-02T18:28:33.417925Z",
+  "dtypes": [
+    "BF16"
+  ],
+  "output": {
+    "path": "vision.safetensors",
+    "sha256": "95424a2beb8065e493985eb0d50aef3cf20ff17694172ad9f44047c9e0f6107d",
+    "size_bytes": 912058056
+  },
+  "parameters": 456010480,
+  "role": "vision",
+  "schema_version": "axquant.protected-tensor-sidecar.v1",
+  "source_files": [
+    {
+      "path": "model.safetensors-00004-of-00004.safetensors",
+      "sha256": "b62b0c4cd7e44edee103ee8f4fe225f246d5e768e07bfd5f25b63a8aa1fdd0c6",
+      "size_bytes": 3325995712
+    }
+  ],
+  "source_model": {
+    "architecture": "Qwen3_5ForConditionalGeneration",
+    "format": "mlx",
+    "local_path": "/Volumes/Ext4T-02/axquant/models/Qwen3.5-9B-bf16",
+    "model_id": "Qwen/Qwen3.5-9B",
+    "revision": null
+  },
+  "tensor_count": 333,
+  "tensor_names_sha256": "74db978fa718039692971f2ece846f2cc7a62a960047b5cebe2d758cc82ac2e9"
+}

chat_template.jinja ADDED Viewed

	@@ -0,0 +1,154 @@

+{%- set image_count = namespace(value=0) %}
+{%- set video_count = namespace(value=0) %}
+{%- macro render_content(content, do_vision_count, is_system_content=false) %}
+    {%- if content is string %}
+        {{- content }}
+    {%- elif content is iterable and content is not mapping %}
+        {%- for item in content %}
+            {%- if 'image' in item or 'image_url' in item or item.type == 'image' %}
+                {%- if is_system_content %}
+                    {{- raise_exception('System message cannot contain images.') }}
+                {%- endif %}
+                {%- if do_vision_count %}
+                    {%- set image_count.value = image_count.value + 1 %}
+                {%- endif %}
+                {%- if add_vision_id %}
+                    {{- 'Picture ' ~ image_count.value ~ ': ' }}
+                {%- endif %}
+                {{- '<|vision_start|><|image_pad|><|vision_end|>' }}
+            {%- elif 'video' in item or item.type == 'video' %}
+                {%- if is_system_content %}
+                    {{- raise_exception('System message cannot contain videos.') }}
+                {%- endif %}
+                {%- if do_vision_count %}
+                    {%- set video_count.value = video_count.value + 1 %}
+                {%- endif %}
+                {%- if add_vision_id %}
+                    {{- 'Video ' ~ video_count.value ~ ': ' }}
+                {%- endif %}
+                {{- '<|vision_start|><|video_pad|><|vision_end|>' }}
+            {%- elif 'text' in item %}
+                {{- item.text }}
+            {%- else %}
+                {{- raise_exception('Unexpected item type in content.') }}
+            {%- endif %}
+        {%- endfor %}
+    {%- elif content is none or content is undefined %}
+        {{- '' }}
+    {%- else %}
+        {{- raise_exception('Unexpected content type.') }}
+    {%- endif %}
+{%- endmacro %}
+{%- if not messages %}
+    {{- raise_exception('No messages provided.') }}
+{%- endif %}
+{%- if tools and tools is iterable and tools is not mapping %}
+    {{- '<|im_start|>system\n' }}
+    {{- "# Tools\n\nYou have access to the following functions:\n\n<tools>" }}
+    {%- for tool in tools %}
+        {{- "\n" }}
+        {{- tool | tojson }}
+    {%- endfor %}
+    {{- "\n</tools>" }}
+    {{- '\n\nIf you choose to call a function ONLY reply in the following format with NO suffix:\n\n<tool_call>\n<function=example_function_name>\n<parameter=example_parameter_1>\nvalue_1\n</parameter>\n<parameter=example_parameter_2>\nThis is the value for the second parameter\nthat can span\nmultiple lines\n</parameter>\n</function>\n</tool_call>\n\n<IMPORTANT>\nReminder:\n- Function calls MUST follow the specified format: an inner <function=...></function> block must be nested within <tool_call></tool_call> XML tags\n- Required parameters MUST be specified\n- You may provide optional reasoning for your function call in natural language BEFORE the function call, but NOT after\n- If there is no function call available, answer the question like normal with your current knowledge and do not tell the user about function calls\n</IMPORTANT>' }}
+    {%- if messages[0].role == 'system' %}
+        {%- set content = render_content(messages[0].content, false, true)|trim %}
+        {%- if content %}
+            {{- '\n\n' + content }}
+        {%- endif %}
+    {%- endif %}
+    {{- '<|im_end|>\n' }}
+{%- else %}
+    {%- if messages[0].role == 'system' %}
+        {%- set content = render_content(messages[0].content, false, true)|trim %}
+        {{- '<|im_start|>system\n' + content + '<|im_end|>\n' }}
+    {%- endif %}
+{%- endif %}
+{%- set ns = namespace(multi_step_tool=true, last_query_index=messages|length - 1) %}
+{%- for message in messages[::-1] %}
+    {%- set index = (messages|length - 1) - loop.index0 %}
+    {%- if ns.multi_step_tool and message.role == "user" %}
+        {%- set content = render_content(message.content, false)|trim %}
+        {%- if not(content.startswith('<tool_response>') and content.endswith('</tool_response>')) %}
+            {%- set ns.multi_step_tool = false %}
+            {%- set ns.last_query_index = index %}
+        {%- endif %}
+    {%- endif %}
+{%- endfor %}
+{%- if ns.multi_step_tool %}
+    {{- raise_exception('No user query found in messages.') }}
+{%- endif %}
+{%- for message in messages %}
+    {%- set content = render_content(message.content, true)|trim %}
+    {%- if message.role == "system" %}
+        {%- if not loop.first %}
+            {{- raise_exception('System message must be at the beginning.') }}
+        {%- endif %}
+    {%- elif message.role == "user" %}
+        {{- '<|im_start|>' + message.role + '\n' + content + '<|im_end|>' + '\n' }}
+    {%- elif message.role == "assistant" %}
+        {%- set reasoning_content = '' %}
+        {%- if message.reasoning_content is string %}
+            {%- set reasoning_content = message.reasoning_content %}
+        {%- else %}
+            {%- if '</think>' in content %}
+                {%- set reasoning_content = content.split('</think>')[0].rstrip('\n').split('<think>')[-1].lstrip('\n') %}
+                {%- set content = content.split('</think>')[-1].lstrip('\n') %}
+            {%- endif %}
+        {%- endif %}
+        {%- set reasoning_content = reasoning_content|trim %}
+        {%- if loop.index0 > ns.last_query_index %}
+            {{- '<|im_start|>' + message.role + '\n<think>\n' + reasoning_content + '\n</think>\n\n' + content }}
+        {%- else %}
+            {{- '<|im_start|>' + message.role + '\n' + content }}
+        {%- endif %}
+        {%- if message.tool_calls and message.tool_calls is iterable and message.tool_calls is not mapping %}
+            {%- for tool_call in message.tool_calls %}
+                {%- if tool_call.function is defined %}
+                    {%- set tool_call = tool_call.function %}
+                {%- endif %}
+                {%- if loop.first %}
+                    {%- if content|trim %}
+                        {{- '\n\n<tool_call>\n<function=' + tool_call.name + '>\n' }}
+                    {%- else %}
+                        {{- '<tool_call>\n<function=' + tool_call.name + '>\n' }}
+                    {%- endif %}
+                {%- else %}
+                    {{- '\n<tool_call>\n<function=' + tool_call.name + '>\n' }}
+                {%- endif %}
+                {%- if tool_call.arguments is defined %}
+                    {%- for args_name, args_value in tool_call.arguments|items %}
+                        {{- '<parameter=' + args_name + '>\n' }}
+                        {%- set args_value = args_value | tojson | safe if args_value is mapping or (args_value is sequence and args_value is not string) else args_value | string %}
+                        {{- args_value }}
+                        {{- '\n</parameter>\n' }}
+                    {%- endfor %}
+                {%- endif %}
+                {{- '</function>\n</tool_call>' }}
+            {%- endfor %}
+        {%- endif %}
+        {{- '<|im_end|>\n' }}
+    {%- elif message.role == "tool" %}
+        {%- if loop.previtem and loop.previtem.role != "tool" %}
+            {{- '<|im_start|>user' }}
+        {%- endif %}
+        {{- '\n<tool_response>\n' }}
+        {{- content }}
+        {{- '\n</tool_response>' }}
+        {%- if not loop.last and loop.nextitem.role != "tool" %}
+            {{- '<|im_end|>\n' }}
+        {%- elif loop.last %}
+            {{- '<|im_end|>\n' }}
+        {%- endif %}
+    {%- else %}
+        {{- raise_exception('Unexpected message role.') }}
+    {%- endif %}
+{%- endfor %}
+{%- if add_generation_prompt %}
+    {{- '<|im_start|>assistant\n' }}
+    {%- if enable_thinking is defined and enable_thinking is false %}
+        {{- '<think>\n\n</think>\n\n' }}
+    {%- else %}
+        {{- '<think>\n' }}
+    {%- endif %}
+{%- endif %}

config.json ADDED Viewed

	@@ -0,0 +1,2603 @@

+{
+  "architectures": [
+    "Qwen3_5ForConditionalGeneration"
+  ],
+  "image_token_id": 248056,
+  "model_type": "qwen3_5",
+  "quantization": {
+    "bits": 4,
+    "group_size": 32,
+    "language_model.model.embed_tokens": {
+      "bits": 8,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.0.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.0.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.0.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.0.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.0.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.0.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.0.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.0.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.1.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.1.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.1.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.1.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.1.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.1.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.1.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.1.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.10.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.10.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.10.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.10.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.10.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.10.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.10.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.10.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.11.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.11.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.11.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.11.self_attn.k_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.11.self_attn.o_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.11.self_attn.q_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.11.self_attn.v_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.12.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.12.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.12.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.12.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.12.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.12.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.12.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.12.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.13.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.13.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.13.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.13.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.13.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.13.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.13.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.13.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.14.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.14.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.14.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.14.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.14.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.14.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.14.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.14.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.15.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.15.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.15.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.15.self_attn.k_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.15.self_attn.o_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.15.self_attn.q_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.15.self_attn.v_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.16.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.16.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.16.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.16.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.16.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.16.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.16.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.16.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.17.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.17.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.17.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.17.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.17.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.17.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.17.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.17.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.18.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.18.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.18.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.18.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.18.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.18.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.18.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.18.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.19.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.19.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.19.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.19.self_attn.k_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.19.self_attn.o_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.19.self_attn.q_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.19.self_attn.v_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.2.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.2.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.2.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.2.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.2.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.2.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.2.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.2.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.20.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.20.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.20.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.20.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.20.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.20.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.20.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.20.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.21.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.21.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.21.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.21.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.21.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.21.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.21.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.21.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.22.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.22.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.22.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.22.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.22.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.22.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.22.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.22.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.23.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.23.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.23.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.23.self_attn.k_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.23.self_attn.o_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.23.self_attn.q_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.23.self_attn.v_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.24.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.24.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.24.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.24.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.24.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.24.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.24.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.24.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.25.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.25.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.25.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.25.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.25.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.25.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.25.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.25.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.26.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.26.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.26.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.26.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.26.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.26.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.26.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.26.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.27.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.27.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.27.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.27.self_attn.k_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.27.self_attn.o_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.27.self_attn.q_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.27.self_attn.v_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.28.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.28.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.28.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.28.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.28.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.28.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.28.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.28.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.29.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.29.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.29.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.29.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.29.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.29.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.29.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.29.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.3.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.3.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.3.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.3.self_attn.k_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.3.self_attn.o_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.3.self_attn.q_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.3.self_attn.v_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.30.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.30.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.30.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.30.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.30.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.30.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.30.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.30.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.31.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.31.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.31.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.31.self_attn.k_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.31.self_attn.o_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.31.self_attn.q_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.31.self_attn.v_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.4.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.4.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.4.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.4.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.4.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.4.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.4.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.4.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.5.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.5.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.5.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.5.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.5.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.5.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.5.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.5.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.6.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.6.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.6.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.6.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.6.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.6.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.6.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.6.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.7.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.7.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.7.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.7.self_attn.k_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.7.self_attn.o_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.7.self_attn.q_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.7.self_attn.v_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.8.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.8.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.8.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.8.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.8.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.8.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.8.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.8.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.9.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.9.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.9.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.9.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.9.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.9.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.9.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.9.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "mode": "affine"
+  },
+  "quantization_config": {
+    "bits": 4,
+    "group_size": 32,
+    "language_model.model.embed_tokens": {
+      "bits": 8,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.0.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.0.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.0.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.0.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.0.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.0.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.0.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.0.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.1.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.1.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.1.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.1.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.1.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.1.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.1.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.1.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.10.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.10.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.10.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.10.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.10.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.10.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.10.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.10.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.11.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.11.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.11.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.11.self_attn.k_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.11.self_attn.o_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.11.self_attn.q_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.11.self_attn.v_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.12.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.12.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.12.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.12.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.12.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.12.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.12.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.12.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.13.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.13.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.13.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.13.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.13.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.13.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.13.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.13.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.14.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.14.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.14.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.14.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.14.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.14.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.14.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.14.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.15.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.15.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.15.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.15.self_attn.k_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.15.self_attn.o_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.15.self_attn.q_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.15.self_attn.v_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.16.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.16.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.16.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.16.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.16.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.16.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.16.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.16.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.17.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.17.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.17.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.17.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.17.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.17.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.17.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.17.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.18.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.18.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.18.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.18.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.18.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.18.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.18.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.18.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.19.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.19.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.19.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.19.self_attn.k_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.19.self_attn.o_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.19.self_attn.q_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.19.self_attn.v_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.2.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.2.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.2.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.2.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.2.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.2.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.2.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.2.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.20.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.20.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.20.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.20.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.20.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.20.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.20.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.20.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.21.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.21.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.21.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.21.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.21.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.21.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.21.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.21.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.22.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.22.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.22.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.22.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.22.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.22.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.22.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.22.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.23.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.23.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.23.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.23.self_attn.k_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.23.self_attn.o_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.23.self_attn.q_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.23.self_attn.v_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.24.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.24.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.24.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.24.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.24.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.24.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.24.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.24.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.25.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.25.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.25.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.25.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.25.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.25.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.25.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.25.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.26.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.26.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.26.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.26.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.26.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.26.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.26.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.26.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.27.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.27.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.27.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.27.self_attn.k_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.27.self_attn.o_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.27.self_attn.q_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.27.self_attn.v_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.28.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.28.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.28.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.28.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.28.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.28.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.28.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.28.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.29.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.29.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.29.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.29.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.29.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.29.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.29.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.29.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.3.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.3.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.3.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.3.self_attn.k_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.3.self_attn.o_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.3.self_attn.q_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.3.self_attn.v_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.30.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.30.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.30.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.30.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.30.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.30.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.30.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.30.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.31.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.31.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.31.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.31.self_attn.k_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.31.self_attn.o_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.31.self_attn.q_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.31.self_attn.v_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.4.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.4.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.4.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.4.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.4.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.4.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.4.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.4.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.5.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.5.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.5.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.5.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.5.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.5.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.5.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.5.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.6.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.6.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.6.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.6.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.6.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.6.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.6.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.6.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.7.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.7.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.7.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.7.self_attn.k_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.7.self_attn.o_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.7.self_attn.q_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.7.self_attn.v_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.8.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.8.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.8.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.8.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.8.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.8.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.8.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.8.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.9.linear_attn.in_proj_a": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.9.linear_attn.in_proj_b": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.9.linear_attn.in_proj_qkv": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.9.linear_attn.in_proj_z": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.9.linear_attn.out_proj": {
+      "bits": 4,
+      "group_size": 32,
+      "mode": "affine"
+    },
+    "language_model.model.layers.9.mlp.down_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.9.mlp.gate_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "language_model.model.layers.9.mlp.up_proj": {
+      "bits": 4,
+      "group_size": 64,
+      "mode": "affine"
+    },
+    "mode": "affine"
+  },
+  "text_config": {
+    "attention_bias": false,
+    "attention_dropout": 0.0,
+    "attn_output_gate": true,
+    "dtype": "bfloat16",
+    "eos_token_id": 248044,
+    "full_attention_interval": 4,
+    "head_dim": 256,
+    "hidden_act": "silu",
+    "hidden_size": 4096,
+    "initializer_range": 0.02,
+    "intermediate_size": 12288,
+    "layer_types": [
+      "linear_attention",
+      "linear_attention",
+      "linear_attention",
+      "full_attention",
+      "linear_attention",
+      "linear_attention",
+      "linear_attention",
+      "full_attention",
+      "linear_attention",
+      "linear_attention",
+      "linear_attention",
+      "full_attention",
+      "linear_attention",
+      "linear_attention",
+      "linear_attention",
+      "full_attention",
+      "linear_attention",
+      "linear_attention",
+      "linear_attention",
+      "full_attention",
+      "linear_attention",
+      "linear_attention",
+      "linear_attention",
+      "full_attention",
+      "linear_attention",
+      "linear_attention",
+      "linear_attention",
+      "full_attention",
+      "linear_attention",
+      "linear_attention",
+      "linear_attention",
+      "full_attention"
+    ],
+    "linear_conv_kernel_dim": 4,
+    "linear_key_head_dim": 128,
+    "linear_num_key_heads": 16,
+    "linear_num_value_heads": 32,
+    "linear_value_head_dim": 128,
+    "mamba_ssm_dtype": "float32",
+    "max_position_embeddings": 262144,
+    "mlp_only_layers": [],
+    "model_type": "qwen3_5_text",
+    "mtp_num_hidden_layers": 1,
+    "mtp_use_dedicated_embeddings": false,
+    "num_attention_heads": 16,
+    "num_hidden_layers": 32,
+    "num_key_value_heads": 4,
+    "rms_norm_eps": 1e-06,
+    "rope_parameters": {
+      "mrope_interleaved": true,
+      "mrope_section": [
+        11,
+        11,
+        10
+      ],
+      "partial_rotary_factor": 0.25,
+      "rope_theta": 10000000,
+      "type": "default"
+    },
+    "use_cache": true,
+    "vocab_size": 248320
+  },
+  "tie_word_embeddings": false,
+  "transformers_version": "4.57.0.dev0",
+  "video_token_id": 248057,
+  "vision_config": {
+    "deepstack_visual_indexes": [],
+    "depth": 27,
+    "hidden_act": "gelu_pytorch_tanh",
+    "hidden_size": 1152,
+    "in_channels": 3,
+    "initializer_range": 0.02,
+    "intermediate_size": 4304,
+    "model_type": "qwen3_5",
+    "num_heads": 16,
+    "num_position_embeddings": 2304,
+    "out_hidden_size": 4096,
+    "patch_size": 16,
+    "spatial_merge_size": 2,
+    "temporal_patch_size": 2
+  },
+  "vision_end_token_id": 248054,
+  "vision_start_token_id": 248053
+}

model-00001-of-00002.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:c7b13cac43465c53c5a8133519e9b819a2021b58ab36cbbd5d344a655598b8a8
+size 4977549600

model-00002-of-00002.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:2da4d602a9e856abed386491e3eeba9abe41aec5fde0719e66e40d1ea61008d2
+size 2034237582

model-manifest.json ADDED Viewed

The diff for this file is too large to render. See raw diff

model.safetensors.index.json ADDED Viewed

	@@ -0,0 +1,933 @@

+{
+    "metadata": {
+        "total_size": 7011666432,
+        "total_parameters": 8953801728
+    },
+    "weight_map": {
+        "language_model.lm_head.weight": "model-00002-of-00002.safetensors",
+        "language_model.model.embed_tokens.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.embed_tokens.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.embed_tokens.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.0.input_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.0.linear_attn.A_log": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.0.linear_attn.conv1d.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.0.linear_attn.dt_bias": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.0.linear_attn.in_proj_a.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.0.linear_attn.in_proj_a.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.0.linear_attn.in_proj_a.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.0.linear_attn.in_proj_b.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.0.linear_attn.in_proj_b.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.0.linear_attn.in_proj_b.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.0.linear_attn.in_proj_qkv.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.0.linear_attn.in_proj_qkv.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.0.linear_attn.in_proj_qkv.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.0.linear_attn.in_proj_z.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.0.linear_attn.in_proj_z.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.0.linear_attn.in_proj_z.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.0.linear_attn.norm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.0.linear_attn.out_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.0.linear_attn.out_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.0.linear_attn.out_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.0.mlp.down_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.0.mlp.down_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.0.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.0.mlp.gate_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.0.mlp.gate_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.0.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.0.mlp.up_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.0.mlp.up_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.0.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.0.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.1.input_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.1.linear_attn.A_log": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.1.linear_attn.conv1d.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.1.linear_attn.dt_bias": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.1.linear_attn.in_proj_a.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.1.linear_attn.in_proj_a.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.1.linear_attn.in_proj_a.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.1.linear_attn.in_proj_b.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.1.linear_attn.in_proj_b.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.1.linear_attn.in_proj_b.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.1.linear_attn.in_proj_qkv.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.1.linear_attn.in_proj_qkv.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.1.linear_attn.in_proj_qkv.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.1.linear_attn.in_proj_z.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.1.linear_attn.in_proj_z.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.1.linear_attn.in_proj_z.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.1.linear_attn.norm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.1.linear_attn.out_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.1.linear_attn.out_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.1.linear_attn.out_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.1.mlp.down_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.1.mlp.down_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.1.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.1.mlp.gate_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.1.mlp.gate_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.1.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.1.mlp.up_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.1.mlp.up_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.1.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.1.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.10.input_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.10.linear_attn.A_log": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.10.linear_attn.conv1d.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.10.linear_attn.dt_bias": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.10.linear_attn.in_proj_a.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.10.linear_attn.in_proj_a.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.10.linear_attn.in_proj_a.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.10.linear_attn.in_proj_b.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.10.linear_attn.in_proj_b.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.10.linear_attn.in_proj_b.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.10.linear_attn.in_proj_qkv.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.10.linear_attn.in_proj_qkv.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.10.linear_attn.in_proj_qkv.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.10.linear_attn.in_proj_z.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.10.linear_attn.in_proj_z.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.10.linear_attn.in_proj_z.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.10.linear_attn.norm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.10.linear_attn.out_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.10.linear_attn.out_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.10.linear_attn.out_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.10.mlp.down_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.10.mlp.down_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.10.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.10.mlp.gate_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.10.mlp.gate_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.10.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.10.mlp.up_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.10.mlp.up_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.10.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.10.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.11.input_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.11.mlp.down_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.11.mlp.down_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.11.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.11.mlp.gate_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.11.mlp.gate_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.11.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.11.mlp.up_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.11.mlp.up_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.11.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.11.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.11.self_attn.k_norm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.11.self_attn.k_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.11.self_attn.k_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.11.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.11.self_attn.o_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.11.self_attn.o_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.11.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.11.self_attn.q_norm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.11.self_attn.q_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.11.self_attn.q_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.11.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.11.self_attn.v_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.11.self_attn.v_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.11.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.12.input_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.12.linear_attn.A_log": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.12.linear_attn.conv1d.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.12.linear_attn.dt_bias": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.12.linear_attn.in_proj_a.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.12.linear_attn.in_proj_a.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.12.linear_attn.in_proj_a.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.12.linear_attn.in_proj_b.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.12.linear_attn.in_proj_b.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.12.linear_attn.in_proj_b.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.12.linear_attn.in_proj_qkv.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.12.linear_attn.in_proj_qkv.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.12.linear_attn.in_proj_qkv.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.12.linear_attn.in_proj_z.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.12.linear_attn.in_proj_z.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.12.linear_attn.in_proj_z.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.12.linear_attn.norm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.12.linear_attn.out_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.12.linear_attn.out_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.12.linear_attn.out_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.12.mlp.down_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.12.mlp.down_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.12.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.12.mlp.gate_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.12.mlp.gate_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.12.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.12.mlp.up_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.12.mlp.up_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.12.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.12.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.13.input_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.13.linear_attn.A_log": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.13.linear_attn.conv1d.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.13.linear_attn.dt_bias": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.13.linear_attn.in_proj_a.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.13.linear_attn.in_proj_a.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.13.linear_attn.in_proj_a.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.13.linear_attn.in_proj_b.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.13.linear_attn.in_proj_b.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.13.linear_attn.in_proj_b.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.13.linear_attn.in_proj_qkv.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.13.linear_attn.in_proj_qkv.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.13.linear_attn.in_proj_qkv.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.13.linear_attn.in_proj_z.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.13.linear_attn.in_proj_z.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.13.linear_attn.in_proj_z.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.13.linear_attn.norm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.13.linear_attn.out_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.13.linear_attn.out_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.13.linear_attn.out_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.13.mlp.down_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.13.mlp.down_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.13.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.13.mlp.gate_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.13.mlp.gate_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.13.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.13.mlp.up_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.13.mlp.up_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.13.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.13.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.14.input_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.14.linear_attn.A_log": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.14.linear_attn.conv1d.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.14.linear_attn.dt_bias": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.14.linear_attn.in_proj_a.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.14.linear_attn.in_proj_a.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.14.linear_attn.in_proj_a.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.14.linear_attn.in_proj_b.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.14.linear_attn.in_proj_b.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.14.linear_attn.in_proj_b.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.14.linear_attn.in_proj_qkv.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.14.linear_attn.in_proj_qkv.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.14.linear_attn.in_proj_qkv.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.14.linear_attn.in_proj_z.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.14.linear_attn.in_proj_z.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.14.linear_attn.in_proj_z.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.14.linear_attn.norm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.14.linear_attn.out_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.14.linear_attn.out_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.14.linear_attn.out_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.14.mlp.down_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.14.mlp.down_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.14.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.14.mlp.gate_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.14.mlp.gate_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.14.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.14.mlp.up_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.14.mlp.up_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.14.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.14.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.15.input_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.15.mlp.down_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.15.mlp.down_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.15.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.15.mlp.gate_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.15.mlp.gate_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.15.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.15.mlp.up_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.15.mlp.up_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.15.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.15.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.15.self_attn.k_norm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.15.self_attn.k_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.15.self_attn.k_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.15.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.15.self_attn.o_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.15.self_attn.o_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.15.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.15.self_attn.q_norm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.15.self_attn.q_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.15.self_attn.q_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.15.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.15.self_attn.v_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.15.self_attn.v_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.15.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.16.input_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.16.linear_attn.A_log": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.16.linear_attn.conv1d.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.16.linear_attn.dt_bias": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.16.linear_attn.in_proj_a.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.16.linear_attn.in_proj_a.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.16.linear_attn.in_proj_a.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.16.linear_attn.in_proj_b.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.16.linear_attn.in_proj_b.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.16.linear_attn.in_proj_b.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.16.linear_attn.in_proj_qkv.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.16.linear_attn.in_proj_qkv.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.16.linear_attn.in_proj_qkv.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.16.linear_attn.in_proj_z.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.16.linear_attn.in_proj_z.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.16.linear_attn.in_proj_z.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.16.linear_attn.norm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.16.linear_attn.out_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.16.linear_attn.out_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.16.linear_attn.out_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.16.mlp.down_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.16.mlp.down_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.16.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.16.mlp.gate_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.16.mlp.gate_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.16.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.16.mlp.up_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.16.mlp.up_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.16.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.16.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.17.input_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.17.linear_attn.A_log": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.17.linear_attn.conv1d.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.17.linear_attn.dt_bias": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.17.linear_attn.in_proj_a.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.17.linear_attn.in_proj_a.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.17.linear_attn.in_proj_a.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.17.linear_attn.in_proj_b.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.17.linear_attn.in_proj_b.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.17.linear_attn.in_proj_b.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.17.linear_attn.in_proj_qkv.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.17.linear_attn.in_proj_qkv.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.17.linear_attn.in_proj_qkv.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.17.linear_attn.in_proj_z.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.17.linear_attn.in_proj_z.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.17.linear_attn.in_proj_z.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.17.linear_attn.norm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.17.linear_attn.out_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.17.linear_attn.out_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.17.linear_attn.out_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.17.mlp.down_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.17.mlp.down_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.17.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.17.mlp.gate_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.17.mlp.gate_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.17.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.17.mlp.up_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.17.mlp.up_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.17.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.17.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.18.input_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.18.linear_attn.A_log": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.18.linear_attn.conv1d.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.18.linear_attn.dt_bias": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.18.linear_attn.in_proj_a.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.18.linear_attn.in_proj_a.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.18.linear_attn.in_proj_a.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.18.linear_attn.in_proj_b.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.18.linear_attn.in_proj_b.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.18.linear_attn.in_proj_b.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.18.linear_attn.in_proj_qkv.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.18.linear_attn.in_proj_qkv.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.18.linear_attn.in_proj_qkv.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.18.linear_attn.in_proj_z.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.18.linear_attn.in_proj_z.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.18.linear_attn.in_proj_z.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.18.linear_attn.norm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.18.linear_attn.out_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.18.linear_attn.out_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.18.linear_attn.out_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.18.mlp.down_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.18.mlp.down_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.18.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.18.mlp.gate_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.18.mlp.gate_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.18.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.18.mlp.up_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.18.mlp.up_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.18.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.18.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.19.input_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.19.mlp.down_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.19.mlp.down_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.19.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.19.mlp.gate_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.19.mlp.gate_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.19.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.19.mlp.up_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.19.mlp.up_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.19.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.19.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.19.self_attn.k_norm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.19.self_attn.k_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.19.self_attn.k_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.19.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.19.self_attn.o_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.19.self_attn.o_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.19.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.19.self_attn.q_norm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.19.self_attn.q_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.19.self_attn.q_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.19.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.19.self_attn.v_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.19.self_attn.v_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.19.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.2.input_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.2.linear_attn.A_log": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.2.linear_attn.conv1d.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.2.linear_attn.dt_bias": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.2.linear_attn.in_proj_a.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.2.linear_attn.in_proj_a.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.2.linear_attn.in_proj_a.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.2.linear_attn.in_proj_b.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.2.linear_attn.in_proj_b.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.2.linear_attn.in_proj_b.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.2.linear_attn.in_proj_qkv.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.2.linear_attn.in_proj_qkv.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.2.linear_attn.in_proj_qkv.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.2.linear_attn.in_proj_z.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.2.linear_attn.in_proj_z.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.2.linear_attn.in_proj_z.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.2.linear_attn.norm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.2.linear_attn.out_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.2.linear_attn.out_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.2.linear_attn.out_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.2.mlp.down_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.2.mlp.down_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.2.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.2.mlp.gate_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.2.mlp.gate_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.2.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.2.mlp.up_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.2.mlp.up_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.2.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.2.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.20.input_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.20.linear_attn.A_log": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.20.linear_attn.conv1d.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.20.linear_attn.dt_bias": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.20.linear_attn.in_proj_a.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.20.linear_attn.in_proj_a.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.20.linear_attn.in_proj_a.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.20.linear_attn.in_proj_b.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.20.linear_attn.in_proj_b.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.20.linear_attn.in_proj_b.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.20.linear_attn.in_proj_qkv.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.20.linear_attn.in_proj_qkv.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.20.linear_attn.in_proj_qkv.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.20.linear_attn.in_proj_z.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.20.linear_attn.in_proj_z.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.20.linear_attn.in_proj_z.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.20.linear_attn.norm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.20.linear_attn.out_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.20.linear_attn.out_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.20.linear_attn.out_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.20.mlp.down_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.20.mlp.down_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.20.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.20.mlp.gate_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.20.mlp.gate_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.20.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.20.mlp.up_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.20.mlp.up_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.20.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.20.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.21.input_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.21.linear_attn.A_log": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.21.linear_attn.conv1d.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.21.linear_attn.dt_bias": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.21.linear_attn.in_proj_a.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.21.linear_attn.in_proj_a.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.21.linear_attn.in_proj_a.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.21.linear_attn.in_proj_b.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.21.linear_attn.in_proj_b.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.21.linear_attn.in_proj_b.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.21.linear_attn.in_proj_qkv.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.21.linear_attn.in_proj_qkv.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.21.linear_attn.in_proj_qkv.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.21.linear_attn.in_proj_z.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.21.linear_attn.in_proj_z.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.21.linear_attn.in_proj_z.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.21.linear_attn.norm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.21.linear_attn.out_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.21.linear_attn.out_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.21.linear_attn.out_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.21.mlp.down_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.21.mlp.down_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.21.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.21.mlp.gate_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.21.mlp.gate_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.21.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.21.mlp.up_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.21.mlp.up_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.21.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.21.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.22.input_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.22.linear_attn.A_log": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.22.linear_attn.conv1d.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.22.linear_attn.dt_bias": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.22.linear_attn.in_proj_a.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.22.linear_attn.in_proj_a.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.22.linear_attn.in_proj_a.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.22.linear_attn.in_proj_b.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.22.linear_attn.in_proj_b.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.22.linear_attn.in_proj_b.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.22.linear_attn.in_proj_qkv.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.22.linear_attn.in_proj_qkv.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.22.linear_attn.in_proj_qkv.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.22.linear_attn.in_proj_z.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.22.linear_attn.in_proj_z.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.22.linear_attn.in_proj_z.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.22.linear_attn.norm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.22.linear_attn.out_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.22.linear_attn.out_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.22.linear_attn.out_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.22.mlp.down_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.22.mlp.down_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.22.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.22.mlp.gate_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.22.mlp.gate_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.22.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.22.mlp.up_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.22.mlp.up_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.22.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.22.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.23.input_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.23.mlp.down_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.23.mlp.down_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.23.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.23.mlp.gate_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.23.mlp.gate_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.23.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.23.mlp.up_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.23.mlp.up_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.23.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.23.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.23.self_attn.k_norm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.23.self_attn.k_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.23.self_attn.k_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.23.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.23.self_attn.o_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.23.self_attn.o_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.23.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.23.self_attn.q_norm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.23.self_attn.q_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.23.self_attn.q_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.23.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.23.self_attn.v_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.23.self_attn.v_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.23.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.24.input_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.24.linear_attn.A_log": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.24.linear_attn.conv1d.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.24.linear_attn.dt_bias": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.24.linear_attn.in_proj_a.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.24.linear_attn.in_proj_a.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.24.linear_attn.in_proj_a.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.24.linear_attn.in_proj_b.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.24.linear_attn.in_proj_b.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.24.linear_attn.in_proj_b.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.24.linear_attn.in_proj_qkv.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.24.linear_attn.in_proj_qkv.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.24.linear_attn.in_proj_qkv.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.24.linear_attn.in_proj_z.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.24.linear_attn.in_proj_z.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.24.linear_attn.in_proj_z.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.24.linear_attn.norm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.24.linear_attn.out_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.24.linear_attn.out_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.24.linear_attn.out_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.24.mlp.down_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.24.mlp.down_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.24.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.24.mlp.gate_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.24.mlp.gate_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.24.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.24.mlp.up_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.24.mlp.up_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.24.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.24.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.25.input_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.25.linear_attn.A_log": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.25.linear_attn.conv1d.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.25.linear_attn.dt_bias": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.25.linear_attn.in_proj_a.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.25.linear_attn.in_proj_a.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.25.linear_attn.in_proj_a.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.25.linear_attn.in_proj_b.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.25.linear_attn.in_proj_b.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.25.linear_attn.in_proj_b.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.25.linear_attn.in_proj_qkv.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.25.linear_attn.in_proj_qkv.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.25.linear_attn.in_proj_qkv.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.25.linear_attn.in_proj_z.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.25.linear_attn.in_proj_z.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.25.linear_attn.in_proj_z.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.25.linear_attn.norm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.25.linear_attn.out_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.25.linear_attn.out_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.25.linear_attn.out_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.25.mlp.down_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.25.mlp.down_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.25.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.25.mlp.gate_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.25.mlp.gate_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.25.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.25.mlp.up_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.25.mlp.up_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.25.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.25.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.26.input_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.26.linear_attn.A_log": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.26.linear_attn.conv1d.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.26.linear_attn.dt_bias": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.26.linear_attn.in_proj_a.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.26.linear_attn.in_proj_a.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.26.linear_attn.in_proj_a.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.26.linear_attn.in_proj_b.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.26.linear_attn.in_proj_b.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.26.linear_attn.in_proj_b.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.26.linear_attn.in_proj_qkv.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.26.linear_attn.in_proj_qkv.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.26.linear_attn.in_proj_qkv.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.26.linear_attn.in_proj_z.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.26.linear_attn.in_proj_z.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.26.linear_attn.in_proj_z.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.26.linear_attn.norm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.26.linear_attn.out_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.26.linear_attn.out_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.26.linear_attn.out_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.26.mlp.down_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.26.mlp.down_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.26.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.26.mlp.gate_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.26.mlp.gate_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.26.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.26.mlp.up_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.26.mlp.up_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.26.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.26.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.27.input_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.27.mlp.down_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.27.mlp.down_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.27.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.27.mlp.gate_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.27.mlp.gate_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.27.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.27.mlp.up_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.27.mlp.up_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.27.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.27.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.27.self_attn.k_norm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.27.self_attn.k_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.27.self_attn.k_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.27.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.27.self_attn.o_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.27.self_attn.o_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.27.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.27.self_attn.q_norm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.27.self_attn.q_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.27.self_attn.q_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.27.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.27.self_attn.v_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.27.self_attn.v_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.27.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.28.input_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.28.linear_attn.A_log": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.28.linear_attn.conv1d.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.28.linear_attn.dt_bias": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.28.linear_attn.in_proj_a.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.28.linear_attn.in_proj_a.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.28.linear_attn.in_proj_a.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.28.linear_attn.in_proj_b.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.28.linear_attn.in_proj_b.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.28.linear_attn.in_proj_b.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.28.linear_attn.in_proj_qkv.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.28.linear_attn.in_proj_qkv.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.28.linear_attn.in_proj_qkv.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.28.linear_attn.in_proj_z.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.28.linear_attn.in_proj_z.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.28.linear_attn.in_proj_z.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.28.linear_attn.norm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.28.linear_attn.out_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.28.linear_attn.out_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.28.linear_attn.out_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.28.mlp.down_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.28.mlp.down_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.28.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.28.mlp.gate_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.28.mlp.gate_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.28.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.28.mlp.up_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.28.mlp.up_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.28.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.28.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.29.input_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.29.linear_attn.A_log": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.29.linear_attn.conv1d.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.29.linear_attn.dt_bias": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.29.linear_attn.in_proj_a.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.29.linear_attn.in_proj_a.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.29.linear_attn.in_proj_a.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.29.linear_attn.in_proj_b.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.29.linear_attn.in_proj_b.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.29.linear_attn.in_proj_b.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.29.linear_attn.in_proj_qkv.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.29.linear_attn.in_proj_qkv.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.29.linear_attn.in_proj_qkv.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.29.linear_attn.in_proj_z.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.29.linear_attn.in_proj_z.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.29.linear_attn.in_proj_z.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.29.linear_attn.norm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.29.linear_attn.out_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.29.linear_attn.out_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.29.linear_attn.out_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.29.mlp.down_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.29.mlp.down_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.29.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.29.mlp.gate_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.29.mlp.gate_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.29.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.29.mlp.up_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.29.mlp.up_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.29.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.29.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.3.input_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.3.mlp.down_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.3.mlp.down_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.3.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.3.mlp.gate_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.3.mlp.gate_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.3.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.3.mlp.up_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.3.mlp.up_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.3.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.3.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.3.self_attn.k_norm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.3.self_attn.k_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.3.self_attn.k_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.3.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.3.self_attn.o_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.3.self_attn.o_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.3.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.3.self_attn.q_norm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.3.self_attn.q_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.3.self_attn.q_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.3.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.3.self_attn.v_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.3.self_attn.v_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.3.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.30.input_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.30.linear_attn.A_log": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.30.linear_attn.conv1d.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.30.linear_attn.dt_bias": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.30.linear_attn.in_proj_a.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.30.linear_attn.in_proj_a.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.30.linear_attn.in_proj_a.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.30.linear_attn.in_proj_b.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.30.linear_attn.in_proj_b.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.30.linear_attn.in_proj_b.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.30.linear_attn.in_proj_qkv.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.30.linear_attn.in_proj_qkv.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.30.linear_attn.in_proj_qkv.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.30.linear_attn.in_proj_z.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.30.linear_attn.in_proj_z.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.30.linear_attn.in_proj_z.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.30.linear_attn.norm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.30.linear_attn.out_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.30.linear_attn.out_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.30.linear_attn.out_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.30.mlp.down_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.30.mlp.down_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.30.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.30.mlp.gate_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.30.mlp.gate_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.30.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.30.mlp.up_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.30.mlp.up_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.30.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.30.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.31.input_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.31.mlp.down_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.31.mlp.down_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.31.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.31.mlp.gate_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.31.mlp.gate_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.31.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.31.mlp.up_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.31.mlp.up_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.31.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.31.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.31.self_attn.k_norm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.31.self_attn.k_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.31.self_attn.k_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.31.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.31.self_attn.o_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.31.self_attn.o_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.31.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.31.self_attn.q_norm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.31.self_attn.q_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.31.self_attn.q_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.31.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.31.self_attn.v_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.31.self_attn.v_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.31.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.4.input_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.4.linear_attn.A_log": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.4.linear_attn.conv1d.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.4.linear_attn.dt_bias": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.4.linear_attn.in_proj_a.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.4.linear_attn.in_proj_a.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.4.linear_attn.in_proj_a.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.4.linear_attn.in_proj_b.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.4.linear_attn.in_proj_b.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.4.linear_attn.in_proj_b.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.4.linear_attn.in_proj_qkv.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.4.linear_attn.in_proj_qkv.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.4.linear_attn.in_proj_qkv.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.4.linear_attn.in_proj_z.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.4.linear_attn.in_proj_z.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.4.linear_attn.in_proj_z.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.4.linear_attn.norm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.4.linear_attn.out_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.4.linear_attn.out_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.4.linear_attn.out_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.4.mlp.down_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.4.mlp.down_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.4.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.4.mlp.gate_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.4.mlp.gate_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.4.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.4.mlp.up_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.4.mlp.up_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.4.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.4.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.5.input_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.5.linear_attn.A_log": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.5.linear_attn.conv1d.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.5.linear_attn.dt_bias": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.5.linear_attn.in_proj_a.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.5.linear_attn.in_proj_a.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.5.linear_attn.in_proj_a.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.5.linear_attn.in_proj_b.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.5.linear_attn.in_proj_b.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.5.linear_attn.in_proj_b.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.5.linear_attn.in_proj_qkv.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.5.linear_attn.in_proj_qkv.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.5.linear_attn.in_proj_qkv.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.5.linear_attn.in_proj_z.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.5.linear_attn.in_proj_z.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.5.linear_attn.in_proj_z.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.5.linear_attn.norm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.5.linear_attn.out_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.5.linear_attn.out_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.5.linear_attn.out_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.5.mlp.down_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.5.mlp.down_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.5.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.5.mlp.gate_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.5.mlp.gate_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.5.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.5.mlp.up_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.5.mlp.up_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.5.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.5.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.6.input_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.6.linear_attn.A_log": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.6.linear_attn.conv1d.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.6.linear_attn.dt_bias": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.6.linear_attn.in_proj_a.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.6.linear_attn.in_proj_a.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.6.linear_attn.in_proj_a.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.6.linear_attn.in_proj_b.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.6.linear_attn.in_proj_b.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.6.linear_attn.in_proj_b.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.6.linear_attn.in_proj_qkv.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.6.linear_attn.in_proj_qkv.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.6.linear_attn.in_proj_qkv.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.6.linear_attn.in_proj_z.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.6.linear_attn.in_proj_z.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.6.linear_attn.in_proj_z.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.6.linear_attn.norm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.6.linear_attn.out_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.6.linear_attn.out_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.6.linear_attn.out_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.6.mlp.down_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.6.mlp.down_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.6.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.6.mlp.gate_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.6.mlp.gate_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.6.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.6.mlp.up_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.6.mlp.up_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.6.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.6.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.7.input_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.7.mlp.down_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.7.mlp.down_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.7.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.7.mlp.gate_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.7.mlp.gate_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.7.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.7.mlp.up_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.7.mlp.up_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.7.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.7.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.7.self_attn.k_norm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.7.self_attn.k_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.7.self_attn.k_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.7.self_attn.k_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.7.self_attn.o_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.7.self_attn.o_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.7.self_attn.o_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.7.self_attn.q_norm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.7.self_attn.q_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.7.self_attn.q_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.7.self_attn.q_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.7.self_attn.v_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.7.self_attn.v_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.7.self_attn.v_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.8.input_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.8.linear_attn.A_log": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.8.linear_attn.conv1d.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.8.linear_attn.dt_bias": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.8.linear_attn.in_proj_a.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.8.linear_attn.in_proj_a.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.8.linear_attn.in_proj_a.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.8.linear_attn.in_proj_b.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.8.linear_attn.in_proj_b.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.8.linear_attn.in_proj_b.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.8.linear_attn.in_proj_qkv.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.8.linear_attn.in_proj_qkv.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.8.linear_attn.in_proj_qkv.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.8.linear_attn.in_proj_z.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.8.linear_attn.in_proj_z.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.8.linear_attn.in_proj_z.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.8.linear_attn.norm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.8.linear_attn.out_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.8.linear_attn.out_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.8.linear_attn.out_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.8.mlp.down_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.8.mlp.down_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.8.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.8.mlp.gate_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.8.mlp.gate_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.8.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.8.mlp.up_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.8.mlp.up_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.8.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.8.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.9.input_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.9.linear_attn.A_log": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.9.linear_attn.conv1d.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.9.linear_attn.dt_bias": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.9.linear_attn.in_proj_a.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.9.linear_attn.in_proj_a.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.9.linear_attn.in_proj_a.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.9.linear_attn.in_proj_b.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.9.linear_attn.in_proj_b.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.9.linear_attn.in_proj_b.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.9.linear_attn.in_proj_qkv.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.9.linear_attn.in_proj_qkv.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.9.linear_attn.in_proj_qkv.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.9.linear_attn.in_proj_z.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.9.linear_attn.in_proj_z.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.9.linear_attn.in_proj_z.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.9.linear_attn.norm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.9.linear_attn.out_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.9.linear_attn.out_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.9.linear_attn.out_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.9.mlp.down_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.9.mlp.down_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.9.mlp.down_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.9.mlp.gate_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.9.mlp.gate_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.9.mlp.gate_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.9.mlp.up_proj.biases": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.9.mlp.up_proj.scales": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.9.mlp.up_proj.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.layers.9.post_attention_layernorm.weight": "model-00001-of-00002.safetensors",
+        "language_model.model.norm.weight": "model-00001-of-00002.safetensors"
+    }
+}

mtp.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:105d9d7ae9ff7c7c538df915495c198bb784feea30521e848b3791223c703f06
+size 486582944

mtplx_runtime.json ADDED Viewed

	@@ -0,0 +1,5 @@

+{
+  "mtp_depth_max": 1,
+  "mtp_norm_layout": "raw_hf_delta",
+  "schema_version": "axquant.mtp-runtime.v1"
+}

tokenizer.json ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:06b9509352d2af50381ab2247e083b80d32d5c0aba91c272ca9ff729b6a0e523
+size 19989325

tokenizer_config.json ADDED Viewed

	@@ -0,0 +1,33 @@

+{
+  "add_prefix_space": false,
+  "audio_bos_token": "<|audio_start|>",
+  "audio_eos_token": "<|audio_end|>",
+  "audio_token": "<|audio_pad|>",
+  "backend": "tokenizers",
+  "bos_token": null,
+  "clean_up_tokenization_spaces": false,
+  "eos_token": "<|im_end|>",
+  "errors": "replace",
+  "image_token": "<|image_pad|>",
+  "is_local": true,
+  "local_files_only": false,
+  "model_max_length": 262144,
+  "model_specific_special_tokens": {
+    "audio_bos_token": "<|audio_start|>",
+    "audio_eos_token": "<|audio_end|>",
+    "audio_token": "<|audio_pad|>",
+    "image_token": "<|image_pad|>",
+    "video_token": "<|video_pad|>",
+    "vision_bos_token": "<|vision_start|>",
+    "vision_eos_token": "<|vision_end|>"
+  },
+  "pad_token": "<|endoftext|>",
+  "pretokenize_regex": "(?i:'s|'t|'re|'ve|'m|'ll|'d)|[^\\r\\n\\p{L}\\p{N}]?[\\p{L}\\p{M}]+|\\p{N}| ?[^\\s\\p{L}\\p{M}\\p{N}]+[\\r\\n]*|\\s*[\\r\\n]+|\\s+(?!\\S)|\\s+",
+  "split_special_tokens": false,
+  "tokenizer_class": "Qwen2Tokenizer",
+  "tool_parser_type": "qwen3_coder",
+  "unk_token": null,
+  "video_token": "<|video_pad|>",
+  "vision_bos_token": "<|vision_start|>",
+  "vision_eos_token": "<|vision_end|>"
+}

vision.safetensors ADDED Viewed

	@@ -0,0 +1,3 @@

+version https://git-lfs.github.com/spec/v1
+oid sha256:95424a2beb8065e493985eb0d50aef3cf20ff17694172ad9f44047c9e0f6107d
+size 912058056