Text Generation
MLX
Safetensors
k2_horizon
quantized
dense
k2-horizon
conversational
4-bit precision
Instructions to use mlx-community/K2-Horizon-3.7B-4bit with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- MLX
How to use mlx-community/K2-Horizon-3.7B-4bit 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("mlx-community/K2-Horizon-3.7B-4bit") 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
- Pi
How to use mlx-community/K2-Horizon-3.7B-4bit with Pi:
Start the MLX server
# Install MLX LM: uv tool install mlx-lm # Start a local OpenAI-compatible server: mlx_lm.server --model "mlx-community/K2-Horizon-3.7B-4bit"
Configure the model in Pi
# Install Pi: npm install -g @earendil-works/pi-coding-agent # Add to ~/.pi/agent/models.json: { "providers": { "mlx-lm": { "baseUrl": "http://localhost:8080/v1", "api": "openai-completions", "apiKey": "none", "models": [ { "id": "mlx-community/K2-Horizon-3.7B-4bit" } ] } } }Run Pi
# Start Pi in your project directory: pi
- MLX LM
How to use mlx-community/K2-Horizon-3.7B-4bit with MLX LM:
Generate or start a chat session
# Install MLX LM uv tool install mlx-lm # Interactive chat REPL mlx_lm.chat --model "mlx-community/K2-Horizon-3.7B-4bit"
Run an OpenAI-compatible server
# Install MLX LM uv tool install mlx-lm # Start the server mlx_lm.server --model "mlx-community/K2-Horizon-3.7B-4bit" # Calling the OpenAI-compatible server with curl curl -X POST "http://localhost:8000/v1/chat/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "mlx-community/K2-Horizon-3.7B-4bit", "messages": [ {"role": "user", "content": "Hello"} ] }' - Hermes Agent
How to use mlx-community/K2-Horizon-3.7B-4bit 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 "mlx-community/K2-Horizon-3.7B-4bit"
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 mlx-community/K2-Horizon-3.7B-4bit
Run Hermes
hermes
- Atomic Chat
- OpenClaw
How to use mlx-community/K2-Horizon-3.7B-4bit with OpenClaw:
Start the MLX server
# Install MLX LM: uv tool install mlx-lm # Start a local OpenAI-compatible server: mlx_lm.server --model "mlx-community/K2-Horizon-3.7B-4bit"
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 "mlx-community/K2-Horizon-3.7B-4bit" \ --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"
Download config.json from mlx-community/K2-Horizon-3.7B-4bit: direct link, hf CLI and curl.
- Browser
- Download file 71.7 kB
-
https://huggingface.co/mlx-community/K2-Horizon-3.7B-4bit/resolve/main/config.json
- Command line
-
hf download hf://mlx-community/K2-Horizon-3.7B-4bit/config.json
-
curl -L -o config.json https://huggingface.co/mlx-community/K2-Horizon-3.7B-4bit/resolve/main/config.json
71.7 kB
| { | |
| "architectures": [ | |
| "K2HorizonForCausalLM" | |
| ], | |
| "attention_bias": false, | |
| "attention_dropout": 0.0, | |
| "attention_gate_func": null, | |
| "bos_token_id": 0, | |
| "decoder_sparse_step": 1, | |
| "dtype": "bfloat16", | |
| "eos_token_id": [ | |
| 1, | |
| 250019 | |
| ], | |
| "head_dim": 128, | |
| "hidden_act": "silu", | |
| "hidden_size": 2560, | |
| "initializer_range": 0.02, | |
| "intermediate_size": 10240, | |
| "layernorm_num_groups": 2, | |
| "max_position_embeddings": 524288, | |
| "mlp_only_layers": [ | |
| 0, | |
| 1, | |
| 2, | |
| 3, | |
| 4, | |
| 5, | |
| 6, | |
| 7, | |
| 8, | |
| 9, | |
| 10, | |
| 11, | |
| 12, | |
| 13, | |
| 14, | |
| 15, | |
| 16, | |
| 17, | |
| 18, | |
| 19, | |
| 20, | |
| 21, | |
| 22, | |
| 23, | |
| 24, | |
| 25, | |
| 26, | |
| 27, | |
| 28, | |
| 29, | |
| 30, | |
| 31, | |
| 32, | |
| 33, | |
| 34, | |
| 35 | |
| ], | |
| "model_type": "k2_horizon", | |
| "moe_gate_bias": false, | |
| "moe_intermediate_size": 0, | |
| "mova_num_experts": 0, | |
| "mova_num_experts_per_tok": 0, | |
| "norm_topk_prob": true, | |
| "num_attention_heads": 32, | |
| "num_experts": 0, | |
| "num_experts_per_tok": 0, | |
| "num_hidden_layers": 36, | |
| "num_key_value_heads": 8, | |
| "num_shared_experts": 0, | |
| "output_router_logits": false, | |
| "pad_token_id": null, | |
| "quantization": { | |
| "group_size": 64, | |
| "bits": 4, | |
| "mode": "affine", | |
| "model.embed_tokens": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.0.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.0.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.0.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.0.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.0.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.0.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.0.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.1.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.1.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.1.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.1.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.1.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.1.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.1.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.2.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.2.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.2.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.2.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.2.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.2.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.2.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.3.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.3.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.3.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.3.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.3.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.3.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.3.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.4.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.4.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.4.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.4.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.4.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.4.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.4.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.5.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.5.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.5.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.5.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.5.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.5.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.5.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.6.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.6.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.6.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.6.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.6.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.6.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.6.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.7.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.7.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.7.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.7.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.7.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.7.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.7.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.8.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.8.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.8.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.8.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.8.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.8.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.8.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.9.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.9.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.9.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.9.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.9.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.9.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.9.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.10.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.10.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.10.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.10.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.10.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.10.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.10.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.11.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.11.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.11.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.11.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.11.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.11.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.11.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.12.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.12.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.12.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.12.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.12.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.12.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.12.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.13.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.13.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.13.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.13.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.13.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.13.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.13.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.14.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.14.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.14.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.14.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.14.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.14.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.14.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.15.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.15.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.15.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.15.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.15.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.15.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.15.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.16.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.16.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.16.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.16.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.16.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.16.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.16.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.17.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.17.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.17.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.17.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.17.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.17.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.17.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.18.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.18.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.18.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.18.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.18.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.18.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.18.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.19.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.19.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.19.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.19.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.19.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.19.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.19.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.20.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.20.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.20.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.20.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.20.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.20.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.20.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.21.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.21.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.21.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.21.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.21.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.21.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.21.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.22.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.22.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.22.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.22.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.22.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.22.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.22.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.23.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.23.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.23.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.23.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.23.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.23.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.23.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.24.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.24.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.24.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.24.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.24.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.24.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.24.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.25.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.25.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.25.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.25.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.25.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.25.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.25.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.26.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.26.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.26.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.26.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.26.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.26.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.26.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.27.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.27.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.27.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.27.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.27.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.27.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.27.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.28.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.28.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.28.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.28.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.28.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.28.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.28.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.29.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.29.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.29.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.29.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.29.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.29.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.29.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.30.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.30.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.30.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.30.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.30.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.30.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.30.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.31.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.31.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.31.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.31.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.31.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.31.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.31.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.32.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.32.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.32.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.32.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.32.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.32.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.32.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.33.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.33.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.33.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.33.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.33.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.33.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.33.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.34.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.34.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.34.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.34.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.34.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.34.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.34.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.35.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.35.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.35.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.35.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.35.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.35.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.35.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "lm_head": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| } | |
| }, | |
| "quantization_config": { | |
| "group_size": 64, | |
| "bits": 4, | |
| "mode": "affine", | |
| "model.embed_tokens": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.0.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.0.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.0.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.0.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.0.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.0.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.0.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.1.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.1.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.1.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.1.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.1.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.1.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.1.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.2.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.2.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.2.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.2.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.2.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.2.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.2.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.3.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.3.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.3.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.3.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.3.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.3.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.3.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.4.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.4.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.4.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.4.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.4.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.4.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.4.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.5.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.5.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.5.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.5.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.5.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.5.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.5.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.6.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.6.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.6.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.6.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.6.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.6.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.6.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.7.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.7.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.7.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.7.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.7.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.7.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.7.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.8.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.8.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.8.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.8.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.8.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.8.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.8.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.9.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.9.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.9.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.9.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.9.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.9.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.9.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.10.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.10.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.10.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.10.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.10.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.10.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.10.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.11.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.11.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.11.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.11.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.11.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.11.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.11.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.12.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.12.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.12.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.12.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.12.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.12.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.12.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.13.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.13.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.13.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.13.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.13.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.13.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.13.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.14.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.14.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.14.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.14.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.14.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.14.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.14.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.15.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.15.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.15.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.15.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.15.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.15.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.15.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.16.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.16.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.16.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.16.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.16.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.16.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.16.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.17.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.17.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.17.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.17.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.17.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.17.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.17.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.18.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.18.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.18.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.18.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.18.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.18.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.18.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.19.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.19.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.19.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.19.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.19.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.19.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.19.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.20.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.20.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.20.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.20.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.20.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.20.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.20.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.21.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.21.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.21.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.21.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.21.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.21.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.21.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.22.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.22.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.22.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.22.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.22.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.22.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.22.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.23.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.23.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.23.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.23.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.23.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.23.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.23.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.24.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.24.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.24.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.24.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.24.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.24.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.24.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.25.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.25.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.25.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.25.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.25.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.25.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.25.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.26.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.26.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.26.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.26.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.26.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.26.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.26.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.27.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.27.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.27.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.27.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.27.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.27.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.27.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.28.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.28.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.28.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.28.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.28.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.28.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.28.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.29.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.29.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.29.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.29.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.29.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.29.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.29.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.30.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.30.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.30.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.30.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.30.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.30.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.30.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.31.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.31.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.31.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.31.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.31.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.31.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.31.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.32.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.32.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.32.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.32.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.32.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.32.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.32.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.33.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.33.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.33.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.33.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.33.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.33.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.33.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.34.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.34.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.34.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.34.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.34.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.34.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.34.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.35.self_attn.q_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.35.self_attn.k_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.35.self_attn.o_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.35.self_attn.v_proj": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| }, | |
| "model.layers.35.mlp.gate_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.35.mlp.down_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "model.layers.35.mlp.up_proj": { | |
| "group_size": 32, | |
| "bits": 4, | |
| "mode": "affine" | |
| }, | |
| "lm_head": { | |
| "group_size": 64, | |
| "bits": 8, | |
| "mode": "affine" | |
| } | |
| }, | |
| "query_key_norm": false, | |
| "rms_norm_eps": 1e-06, | |
| "rope_head_dim": 128, | |
| "rope_parameters": { | |
| "rope_theta": 10000000.0, | |
| "rope_type": "default" | |
| }, | |
| "router_aux_loss_coef": 0.001, | |
| "router_scaling_factor": 1.0, | |
| "router_score_func": "sigmoid", | |
| "sliding_window": null, | |
| "tie_word_embeddings": false, | |
| "transformers_version": "5.17.0", | |
| "use_cache": true, | |
| "use_sliding_window": false, | |
| "vocab_size": 250624, | |
| "model_file": "k2_horizon.py" | |
| } |