{ "architectures": [ "Glm5NextForConditionalGeneration" ], "text_config": { "attention_bias": false, "attention_dropout": 0.0, "dtype": "bfloat16", "eos_token_id": [ 154820, 154827, 154829 ], "first_k_dense_replace": 3, "hc_eps": 1e-06, "hc_mult": 4, "hc_sinkhorn_iters": 20, "head_dim": 0, "hidden_act": "silu", "hidden_size": 4096, "index_head_dim": 128, "index_kpool": 4, "index_kpool_always_select_tail": true, "index_kpool_compress": true, "index_n_heads": 32, "index_topk": 2048, "index_share_for_mtp_iteration": true, "indexer_rope_interleave": true, "indexer_types": [ "full", "full", "full", "full", "full", "full", "full", "full", "full", "full", "full", "full", "full", "full", "full", "full", "full", "full", "full", "full", "full", "full", "full", "full", "full", "full", "full", "full", "full", "full", "full", "full", "full", "full", "full", "full", "full", "full", "full", "full", "full", "full", "full", "full", "full" ], "initializer_range": 0.02, "intermediate_size": 12288, "kv_lora_rank": 512, "layer_types": [ "linear_attention", "linear_attention", "linear_attention", "deepseek_sparse_attention", "linear_attention", "linear_attention", "linear_attention", "deepseek_sparse_attention", "linear_attention", "linear_attention", "linear_attention", "deepseek_sparse_attention", "linear_attention", "linear_attention", "linear_attention", "deepseek_sparse_attention", "linear_attention", "linear_attention", "linear_attention", "deepseek_sparse_attention", "linear_attention", "linear_attention", "linear_attention", "deepseek_sparse_attention", "linear_attention", "linear_attention", "linear_attention", "deepseek_sparse_attention", "linear_attention", "linear_attention", "linear_attention", "deepseek_sparse_attention", "linear_attention", "linear_attention", "linear_attention", "deepseek_sparse_attention", "linear_attention", "linear_attention", "linear_attention", "deepseek_sparse_attention", "linear_attention", "linear_attention", "linear_attention", "deepseek_sparse_attention", "linear_attention" ], "linear_attn_config": { "num_heads": 64, "gate_lower_bound": -5.0, "head_dim": 128, "short_conv_kernel_size": 4, "kda_layers": [ 0, 1, 2, 4, 5, 6, 8, 9, 10, 12, 13, 14, 16, 17, 18, 20, 21, 22, 24, 25, 26, 28, 29, 30, 32, 33, 34, 36, 37, 38, 40, 41, 42, 44 ], "full_attn_layers": [ 3, 7, 11, 15, 19, 23, 27, 31, 35, 39, 43 ] }, "max_position_embeddings": 1048576, "mhc": true, "mla_use_nope": true, "mlp_layer_types": [ "dense", "dense", "dense", "sparse", "sparse", "sparse", "sparse", "sparse", "sparse", "sparse", "sparse", "sparse", "sparse", "sparse", "sparse", "sparse", "sparse", "sparse", "sparse", "sparse", "sparse", "sparse", "sparse", "sparse", "sparse", "sparse", "sparse", "sparse", "sparse", "sparse", "sparse", "sparse", "sparse", "sparse", "sparse", "sparse", "sparse", "sparse", "sparse", "sparse", "sparse", "sparse", "sparse", "sparse", "sparse" ], "model_type": "glm5_next_text", "moe_intermediate_size": 2048, "moe_router_dtype": "float32", "n_group": 1, "n_routed_experts": 288, "n_shared_experts": 1, "norm_topk_prob": true, "num_attention_heads": 64, "num_experts_per_tok": 8, "num_hidden_layers": 45, "num_key_value_heads": 64, "num_nextn_predict_layers": 1, "output_router_logits": false, "pad_token_id": 154820, "q_lora_rank": 1536, "qk_head_dim": 256, "qk_nope_head_dim": 256, "qk_rope_head_dim": 0, "rms_norm_eps": 1e-05, "routed_scaling_factor": 2.5, "router_aux_loss_coef": 0.001, "scoring_func": "sigmoid", "swiglu_limit": 10.0, "tie_word_embeddings": false, "topk_group": 1, "topk_method": "noaux_tc", "use_cache": true, "v_head_dim": 256, "vocab_size": 154880 }, "vision_config": { "attention_bias": true, "attention_dropout": 0.0, "depth": 24, "hidden_act": "silu", "hidden_size": 1024, "image_size": 448, "in_channels": 3, "initializer_range": 0.02, "intermediate_size": 4096, "model_type": "glm5_next_vision", "num_heads": 16, "out_hidden_size": 4096, "patch_size": 14, "projection_intermediate_size": 10240, "rms_norm_eps": 1e-05, "spatial_merge_size": 2, "swiglu_limit": 10.0, "temporal_patch_size": 2 }, "image_token_id": 154854, "video_token_id": 154855, "image_start_token_id": 154830, "image_end_token_id": 154831, "video_start_token_id": 154832, "video_end_token_id": 154833, "tie_word_embeddings": false, "model_type": "glm5_next", "transformers_version": "5.16.0", "quantization_config": { "group_size": 64, "bits": 4, "mode": "affine", "language_model.lm_head": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.embed_tokens": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.0.attn_hc.base": false, "language_model.model.layers.0.attn_hc.fn": false, "language_model.model.layers.0.attn_hc.scale": false, "language_model.model.layers.0.ffn_hc.base": false, "language_model.model.layers.0.ffn_hc.fn": false, "language_model.model.layers.0.ffn_hc.scale": false, "language_model.model.layers.0.input_layernorm": false, "language_model.model.layers.0.mlp.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.0.mlp.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.0.mlp.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.0.post_attention_layernorm": false, "language_model.model.layers.0.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.0.self_attn.forget_gate.A_log": false, "language_model.model.layers.0.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.0.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.0.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.0.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.0.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.0.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.0.self_attn.o_norm": false, "language_model.model.layers.0.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.0.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.0.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.1.attn_hc.base": false, "language_model.model.layers.1.attn_hc.fn": false, "language_model.model.layers.1.attn_hc.scale": false, "language_model.model.layers.1.ffn_hc.base": false, "language_model.model.layers.1.ffn_hc.fn": false, "language_model.model.layers.1.ffn_hc.scale": false, "language_model.model.layers.1.input_layernorm": false, "language_model.model.layers.1.mlp.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.1.mlp.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.1.mlp.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.1.post_attention_layernorm": false, "language_model.model.layers.1.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.1.self_attn.forget_gate.A_log": false, "language_model.model.layers.1.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.1.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.1.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.1.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.1.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.1.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.1.self_attn.o_norm": false, "language_model.model.layers.1.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.1.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.1.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.10.attn_hc.base": false, "language_model.model.layers.10.attn_hc.fn": false, "language_model.model.layers.10.attn_hc.scale": false, "language_model.model.layers.10.ffn_hc.base": false, "language_model.model.layers.10.ffn_hc.fn": false, "language_model.model.layers.10.ffn_hc.scale": false, "language_model.model.layers.10.input_layernorm": false, "language_model.model.layers.10.mlp.gate": false, "language_model.model.layers.10.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.10.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.10.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.10.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.10.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.10.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.10.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.10.post_attention_layernorm": false, "language_model.model.layers.10.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.10.self_attn.forget_gate.A_log": false, "language_model.model.layers.10.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.10.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.10.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.10.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.10.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.10.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.10.self_attn.o_norm": false, "language_model.model.layers.10.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.10.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.10.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.11.attn_hc.base": false, "language_model.model.layers.11.attn_hc.fn": false, "language_model.model.layers.11.attn_hc.scale": false, "language_model.model.layers.11.ffn_hc.base": false, "language_model.model.layers.11.ffn_hc.fn": false, "language_model.model.layers.11.ffn_hc.scale": false, "language_model.model.layers.11.input_layernorm": false, "language_model.model.layers.11.mlp.gate": false, "language_model.model.layers.11.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.11.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.11.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.11.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.11.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.11.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.11.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.11.post_attention_layernorm": false, "language_model.model.layers.11.self_attn.embed_q": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.11.self_attn.indexer.index_kpool_compress_ape": false, "language_model.model.layers.11.self_attn.indexer.index_kpool_compress_gate": false, "language_model.model.layers.11.self_attn.indexer.k_norm": false, "language_model.model.layers.11.self_attn.indexer.k_norm.bias": false, "language_model.model.layers.11.self_attn.indexer.weights_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.11.self_attn.indexer.wk": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.11.self_attn.indexer.wq_b": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.11.self_attn.kv_a_layernorm": false, "language_model.model.layers.11.self_attn.kv_a_proj_with_mqa": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.11.self_attn.o_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.11.self_attn.q_a_layernorm": false, "language_model.model.layers.11.self_attn.q_a_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.11.self_attn.q_b_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.11.self_attn.unembed_out": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.12.attn_hc.base": false, "language_model.model.layers.12.attn_hc.fn": false, "language_model.model.layers.12.attn_hc.scale": false, "language_model.model.layers.12.ffn_hc.base": false, "language_model.model.layers.12.ffn_hc.fn": false, "language_model.model.layers.12.ffn_hc.scale": false, "language_model.model.layers.12.input_layernorm": false, "language_model.model.layers.12.mlp.gate": false, "language_model.model.layers.12.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.12.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.12.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.12.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.12.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.12.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.12.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.12.post_attention_layernorm": false, "language_model.model.layers.12.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.12.self_attn.forget_gate.A_log": false, "language_model.model.layers.12.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.12.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.12.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.12.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.12.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.12.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.12.self_attn.o_norm": false, "language_model.model.layers.12.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.12.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.12.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.13.attn_hc.base": false, "language_model.model.layers.13.attn_hc.fn": false, "language_model.model.layers.13.attn_hc.scale": false, "language_model.model.layers.13.ffn_hc.base": false, "language_model.model.layers.13.ffn_hc.fn": false, "language_model.model.layers.13.ffn_hc.scale": false, "language_model.model.layers.13.input_layernorm": false, "language_model.model.layers.13.mlp.gate": false, "language_model.model.layers.13.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.13.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.13.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.13.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.13.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.13.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.13.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.13.post_attention_layernorm": false, "language_model.model.layers.13.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.13.self_attn.forget_gate.A_log": false, "language_model.model.layers.13.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.13.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.13.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.13.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.13.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.13.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.13.self_attn.o_norm": false, "language_model.model.layers.13.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.13.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.13.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.14.attn_hc.base": false, "language_model.model.layers.14.attn_hc.fn": false, "language_model.model.layers.14.attn_hc.scale": false, "language_model.model.layers.14.ffn_hc.base": false, "language_model.model.layers.14.ffn_hc.fn": false, "language_model.model.layers.14.ffn_hc.scale": false, "language_model.model.layers.14.input_layernorm": false, "language_model.model.layers.14.mlp.gate": false, "language_model.model.layers.14.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.14.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.14.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.14.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.14.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.14.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.14.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.14.post_attention_layernorm": false, "language_model.model.layers.14.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.14.self_attn.forget_gate.A_log": false, "language_model.model.layers.14.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.14.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.14.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.14.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.14.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.14.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.14.self_attn.o_norm": false, "language_model.model.layers.14.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.14.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.14.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.15.attn_hc.base": false, "language_model.model.layers.15.attn_hc.fn": false, "language_model.model.layers.15.attn_hc.scale": false, "language_model.model.layers.15.ffn_hc.base": false, "language_model.model.layers.15.ffn_hc.fn": false, "language_model.model.layers.15.ffn_hc.scale": false, "language_model.model.layers.15.input_layernorm": false, "language_model.model.layers.15.mlp.gate": false, "language_model.model.layers.15.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.15.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.15.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.15.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.15.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.15.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.15.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.15.post_attention_layernorm": false, "language_model.model.layers.15.self_attn.embed_q": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.15.self_attn.indexer.index_kpool_compress_ape": false, "language_model.model.layers.15.self_attn.indexer.index_kpool_compress_gate": false, "language_model.model.layers.15.self_attn.indexer.k_norm": false, "language_model.model.layers.15.self_attn.indexer.k_norm.bias": false, "language_model.model.layers.15.self_attn.indexer.weights_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.15.self_attn.indexer.wk": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.15.self_attn.indexer.wq_b": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.15.self_attn.kv_a_layernorm": false, "language_model.model.layers.15.self_attn.kv_a_proj_with_mqa": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.15.self_attn.o_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.15.self_attn.q_a_layernorm": false, "language_model.model.layers.15.self_attn.q_a_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.15.self_attn.q_b_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.15.self_attn.unembed_out": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.16.attn_hc.base": false, "language_model.model.layers.16.attn_hc.fn": false, "language_model.model.layers.16.attn_hc.scale": false, "language_model.model.layers.16.ffn_hc.base": false, "language_model.model.layers.16.ffn_hc.fn": false, "language_model.model.layers.16.ffn_hc.scale": false, "language_model.model.layers.16.input_layernorm": false, "language_model.model.layers.16.mlp.gate": false, "language_model.model.layers.16.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.16.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.16.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.16.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.16.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.16.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.16.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.16.post_attention_layernorm": false, "language_model.model.layers.16.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.16.self_attn.forget_gate.A_log": false, "language_model.model.layers.16.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.16.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.16.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.16.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.16.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.16.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.16.self_attn.o_norm": false, "language_model.model.layers.16.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.16.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.16.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.17.attn_hc.base": false, "language_model.model.layers.17.attn_hc.fn": false, "language_model.model.layers.17.attn_hc.scale": false, "language_model.model.layers.17.ffn_hc.base": false, "language_model.model.layers.17.ffn_hc.fn": false, "language_model.model.layers.17.ffn_hc.scale": false, "language_model.model.layers.17.input_layernorm": false, "language_model.model.layers.17.mlp.gate": false, "language_model.model.layers.17.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.17.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.17.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.17.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.17.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.17.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.17.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.17.post_attention_layernorm": false, "language_model.model.layers.17.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.17.self_attn.forget_gate.A_log": false, "language_model.model.layers.17.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.17.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.17.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.17.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.17.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.17.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.17.self_attn.o_norm": false, "language_model.model.layers.17.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.17.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.17.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.18.attn_hc.base": false, "language_model.model.layers.18.attn_hc.fn": false, "language_model.model.layers.18.attn_hc.scale": false, "language_model.model.layers.18.ffn_hc.base": false, "language_model.model.layers.18.ffn_hc.fn": false, "language_model.model.layers.18.ffn_hc.scale": false, "language_model.model.layers.18.input_layernorm": false, "language_model.model.layers.18.mlp.gate": false, "language_model.model.layers.18.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.18.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.18.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.18.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.18.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.18.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.18.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.18.post_attention_layernorm": false, "language_model.model.layers.18.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.18.self_attn.forget_gate.A_log": false, "language_model.model.layers.18.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.18.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.18.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.18.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.18.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.18.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.18.self_attn.o_norm": false, "language_model.model.layers.18.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.18.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.18.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.19.attn_hc.base": false, "language_model.model.layers.19.attn_hc.fn": false, "language_model.model.layers.19.attn_hc.scale": false, "language_model.model.layers.19.ffn_hc.base": false, "language_model.model.layers.19.ffn_hc.fn": false, "language_model.model.layers.19.ffn_hc.scale": false, "language_model.model.layers.19.input_layernorm": false, "language_model.model.layers.19.mlp.gate": false, "language_model.model.layers.19.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.19.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.19.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.19.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.19.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.19.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.19.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.19.post_attention_layernorm": false, "language_model.model.layers.19.self_attn.embed_q": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.19.self_attn.indexer.index_kpool_compress_ape": false, "language_model.model.layers.19.self_attn.indexer.index_kpool_compress_gate": false, "language_model.model.layers.19.self_attn.indexer.k_norm": false, "language_model.model.layers.19.self_attn.indexer.k_norm.bias": false, "language_model.model.layers.19.self_attn.indexer.weights_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.19.self_attn.indexer.wk": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.19.self_attn.indexer.wq_b": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.19.self_attn.kv_a_layernorm": false, "language_model.model.layers.19.self_attn.kv_a_proj_with_mqa": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.19.self_attn.o_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.19.self_attn.q_a_layernorm": false, "language_model.model.layers.19.self_attn.q_a_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.19.self_attn.q_b_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.19.self_attn.unembed_out": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.2.attn_hc.base": false, "language_model.model.layers.2.attn_hc.fn": false, "language_model.model.layers.2.attn_hc.scale": false, "language_model.model.layers.2.ffn_hc.base": false, "language_model.model.layers.2.ffn_hc.fn": false, "language_model.model.layers.2.ffn_hc.scale": false, "language_model.model.layers.2.input_layernorm": false, "language_model.model.layers.2.mlp.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.2.mlp.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.2.mlp.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.2.post_attention_layernorm": false, "language_model.model.layers.2.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.2.self_attn.forget_gate.A_log": false, "language_model.model.layers.2.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.2.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.2.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.2.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.2.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.2.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.2.self_attn.o_norm": false, "language_model.model.layers.2.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.2.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.2.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.20.attn_hc.base": false, "language_model.model.layers.20.attn_hc.fn": false, "language_model.model.layers.20.attn_hc.scale": false, "language_model.model.layers.20.ffn_hc.base": false, "language_model.model.layers.20.ffn_hc.fn": false, "language_model.model.layers.20.ffn_hc.scale": false, "language_model.model.layers.20.input_layernorm": false, "language_model.model.layers.20.mlp.gate": false, "language_model.model.layers.20.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.20.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.20.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.20.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.20.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.20.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.20.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.20.post_attention_layernorm": false, "language_model.model.layers.20.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.20.self_attn.forget_gate.A_log": false, "language_model.model.layers.20.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.20.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.20.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.20.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.20.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.20.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.20.self_attn.o_norm": false, "language_model.model.layers.20.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.20.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.20.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.21.attn_hc.base": false, "language_model.model.layers.21.attn_hc.fn": false, "language_model.model.layers.21.attn_hc.scale": false, "language_model.model.layers.21.ffn_hc.base": false, "language_model.model.layers.21.ffn_hc.fn": false, "language_model.model.layers.21.ffn_hc.scale": false, "language_model.model.layers.21.input_layernorm": false, "language_model.model.layers.21.mlp.gate": false, "language_model.model.layers.21.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.21.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.21.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.21.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.21.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.21.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.21.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.21.post_attention_layernorm": false, "language_model.model.layers.21.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.21.self_attn.forget_gate.A_log": false, "language_model.model.layers.21.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.21.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.21.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.21.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.21.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.21.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.21.self_attn.o_norm": false, "language_model.model.layers.21.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.21.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.21.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.22.attn_hc.base": false, "language_model.model.layers.22.attn_hc.fn": false, "language_model.model.layers.22.attn_hc.scale": false, "language_model.model.layers.22.ffn_hc.base": false, "language_model.model.layers.22.ffn_hc.fn": false, "language_model.model.layers.22.ffn_hc.scale": false, "language_model.model.layers.22.input_layernorm": false, "language_model.model.layers.22.mlp.gate": false, "language_model.model.layers.22.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.22.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.22.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.22.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.22.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.22.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.22.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.22.post_attention_layernorm": false, "language_model.model.layers.22.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.22.self_attn.forget_gate.A_log": false, "language_model.model.layers.22.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.22.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.22.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.22.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.22.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.22.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.22.self_attn.o_norm": false, "language_model.model.layers.22.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.22.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.22.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.23.attn_hc.base": false, "language_model.model.layers.23.attn_hc.fn": false, "language_model.model.layers.23.attn_hc.scale": false, "language_model.model.layers.23.ffn_hc.base": false, "language_model.model.layers.23.ffn_hc.fn": false, "language_model.model.layers.23.ffn_hc.scale": false, "language_model.model.layers.23.input_layernorm": false, "language_model.model.layers.23.mlp.gate": false, "language_model.model.layers.23.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.23.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.23.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.23.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.23.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.23.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.23.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.23.post_attention_layernorm": false, "language_model.model.layers.23.self_attn.embed_q": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.23.self_attn.indexer.index_kpool_compress_ape": false, "language_model.model.layers.23.self_attn.indexer.index_kpool_compress_gate": false, "language_model.model.layers.23.self_attn.indexer.k_norm": false, "language_model.model.layers.23.self_attn.indexer.k_norm.bias": false, "language_model.model.layers.23.self_attn.indexer.weights_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.23.self_attn.indexer.wk": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.23.self_attn.indexer.wq_b": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.23.self_attn.kv_a_layernorm": false, "language_model.model.layers.23.self_attn.kv_a_proj_with_mqa": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.23.self_attn.o_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.23.self_attn.q_a_layernorm": false, "language_model.model.layers.23.self_attn.q_a_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.23.self_attn.q_b_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.23.self_attn.unembed_out": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.24.attn_hc.base": false, "language_model.model.layers.24.attn_hc.fn": false, "language_model.model.layers.24.attn_hc.scale": false, "language_model.model.layers.24.ffn_hc.base": false, "language_model.model.layers.24.ffn_hc.fn": false, "language_model.model.layers.24.ffn_hc.scale": false, "language_model.model.layers.24.input_layernorm": false, "language_model.model.layers.24.mlp.gate": false, "language_model.model.layers.24.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.24.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.24.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.24.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.24.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.24.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.24.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.24.post_attention_layernorm": false, "language_model.model.layers.24.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.24.self_attn.forget_gate.A_log": false, "language_model.model.layers.24.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.24.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.24.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.24.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.24.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.24.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.24.self_attn.o_norm": false, "language_model.model.layers.24.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.24.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.24.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.25.attn_hc.base": false, "language_model.model.layers.25.attn_hc.fn": false, "language_model.model.layers.25.attn_hc.scale": false, "language_model.model.layers.25.ffn_hc.base": false, "language_model.model.layers.25.ffn_hc.fn": false, "language_model.model.layers.25.ffn_hc.scale": false, "language_model.model.layers.25.input_layernorm": false, "language_model.model.layers.25.mlp.gate": false, "language_model.model.layers.25.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.25.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.25.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.25.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.25.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.25.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.25.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.25.post_attention_layernorm": false, "language_model.model.layers.25.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.25.self_attn.forget_gate.A_log": false, "language_model.model.layers.25.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.25.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.25.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.25.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.25.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.25.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.25.self_attn.o_norm": false, "language_model.model.layers.25.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.25.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.25.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.26.attn_hc.base": false, "language_model.model.layers.26.attn_hc.fn": false, "language_model.model.layers.26.attn_hc.scale": false, "language_model.model.layers.26.ffn_hc.base": false, "language_model.model.layers.26.ffn_hc.fn": false, "language_model.model.layers.26.ffn_hc.scale": false, "language_model.model.layers.26.input_layernorm": false, "language_model.model.layers.26.mlp.gate": false, "language_model.model.layers.26.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.26.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.26.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.26.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.26.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.26.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.26.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.26.post_attention_layernorm": false, "language_model.model.layers.26.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.26.self_attn.forget_gate.A_log": false, "language_model.model.layers.26.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.26.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.26.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.26.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.26.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.26.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.26.self_attn.o_norm": false, "language_model.model.layers.26.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.26.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.26.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.27.attn_hc.base": false, "language_model.model.layers.27.attn_hc.fn": false, "language_model.model.layers.27.attn_hc.scale": false, "language_model.model.layers.27.ffn_hc.base": false, "language_model.model.layers.27.ffn_hc.fn": false, "language_model.model.layers.27.ffn_hc.scale": false, "language_model.model.layers.27.input_layernorm": false, "language_model.model.layers.27.mlp.gate": false, "language_model.model.layers.27.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.27.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.27.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.27.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.27.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.27.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.27.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.27.post_attention_layernorm": false, "language_model.model.layers.27.self_attn.embed_q": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.27.self_attn.indexer.index_kpool_compress_ape": false, "language_model.model.layers.27.self_attn.indexer.index_kpool_compress_gate": false, "language_model.model.layers.27.self_attn.indexer.k_norm": false, "language_model.model.layers.27.self_attn.indexer.k_norm.bias": false, "language_model.model.layers.27.self_attn.indexer.weights_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.27.self_attn.indexer.wk": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.27.self_attn.indexer.wq_b": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.27.self_attn.kv_a_layernorm": false, "language_model.model.layers.27.self_attn.kv_a_proj_with_mqa": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.27.self_attn.o_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.27.self_attn.q_a_layernorm": false, "language_model.model.layers.27.self_attn.q_a_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.27.self_attn.q_b_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.27.self_attn.unembed_out": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.28.attn_hc.base": false, "language_model.model.layers.28.attn_hc.fn": false, "language_model.model.layers.28.attn_hc.scale": false, "language_model.model.layers.28.ffn_hc.base": false, "language_model.model.layers.28.ffn_hc.fn": false, "language_model.model.layers.28.ffn_hc.scale": false, "language_model.model.layers.28.input_layernorm": false, "language_model.model.layers.28.mlp.gate": false, "language_model.model.layers.28.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.28.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.28.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.28.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.28.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.28.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.28.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.28.post_attention_layernorm": false, "language_model.model.layers.28.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.28.self_attn.forget_gate.A_log": false, "language_model.model.layers.28.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.28.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.28.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.28.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.28.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.28.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.28.self_attn.o_norm": false, "language_model.model.layers.28.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.28.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.28.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.29.attn_hc.base": false, "language_model.model.layers.29.attn_hc.fn": false, "language_model.model.layers.29.attn_hc.scale": false, "language_model.model.layers.29.ffn_hc.base": false, "language_model.model.layers.29.ffn_hc.fn": false, "language_model.model.layers.29.ffn_hc.scale": false, "language_model.model.layers.29.input_layernorm": false, "language_model.model.layers.29.mlp.gate": false, "language_model.model.layers.29.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.29.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.29.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.29.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.29.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.29.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.29.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.29.post_attention_layernorm": false, "language_model.model.layers.29.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.29.self_attn.forget_gate.A_log": false, "language_model.model.layers.29.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.29.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.29.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.29.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.29.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.29.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.29.self_attn.o_norm": false, "language_model.model.layers.29.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.29.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.29.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.3.attn_hc.base": false, "language_model.model.layers.3.attn_hc.fn": false, "language_model.model.layers.3.attn_hc.scale": false, "language_model.model.layers.3.ffn_hc.base": false, "language_model.model.layers.3.ffn_hc.fn": false, "language_model.model.layers.3.ffn_hc.scale": false, "language_model.model.layers.3.input_layernorm": false, "language_model.model.layers.3.mlp.gate": false, "language_model.model.layers.3.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.3.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.3.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.3.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.3.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.3.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.3.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.3.post_attention_layernorm": false, "language_model.model.layers.3.self_attn.embed_q": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.3.self_attn.indexer.index_kpool_compress_ape": false, "language_model.model.layers.3.self_attn.indexer.index_kpool_compress_gate": false, "language_model.model.layers.3.self_attn.indexer.k_norm": false, "language_model.model.layers.3.self_attn.indexer.k_norm.bias": false, "language_model.model.layers.3.self_attn.indexer.weights_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.3.self_attn.indexer.wk": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.3.self_attn.indexer.wq_b": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.3.self_attn.kv_a_layernorm": false, "language_model.model.layers.3.self_attn.kv_a_proj_with_mqa": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.3.self_attn.o_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.3.self_attn.q_a_layernorm": false, "language_model.model.layers.3.self_attn.q_a_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.3.self_attn.q_b_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.3.self_attn.unembed_out": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.30.attn_hc.base": false, "language_model.model.layers.30.attn_hc.fn": false, "language_model.model.layers.30.attn_hc.scale": false, "language_model.model.layers.30.ffn_hc.base": false, "language_model.model.layers.30.ffn_hc.fn": false, "language_model.model.layers.30.ffn_hc.scale": false, "language_model.model.layers.30.input_layernorm": false, "language_model.model.layers.30.mlp.gate": false, "language_model.model.layers.30.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.30.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.30.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.30.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.30.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.30.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.30.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.30.post_attention_layernorm": false, "language_model.model.layers.30.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.30.self_attn.forget_gate.A_log": false, "language_model.model.layers.30.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.30.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.30.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.30.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.30.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.30.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.30.self_attn.o_norm": false, "language_model.model.layers.30.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.30.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.30.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.31.attn_hc.base": false, "language_model.model.layers.31.attn_hc.fn": false, "language_model.model.layers.31.attn_hc.scale": false, "language_model.model.layers.31.ffn_hc.base": false, "language_model.model.layers.31.ffn_hc.fn": false, "language_model.model.layers.31.ffn_hc.scale": false, "language_model.model.layers.31.input_layernorm": false, "language_model.model.layers.31.mlp.gate": false, "language_model.model.layers.31.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.31.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.31.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.31.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.31.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.31.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.31.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.31.post_attention_layernorm": false, "language_model.model.layers.31.self_attn.embed_q": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.31.self_attn.indexer.index_kpool_compress_ape": false, "language_model.model.layers.31.self_attn.indexer.index_kpool_compress_gate": false, "language_model.model.layers.31.self_attn.indexer.k_norm": false, "language_model.model.layers.31.self_attn.indexer.k_norm.bias": false, "language_model.model.layers.31.self_attn.indexer.weights_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.31.self_attn.indexer.wk": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.31.self_attn.indexer.wq_b": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.31.self_attn.kv_a_layernorm": false, "language_model.model.layers.31.self_attn.kv_a_proj_with_mqa": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.31.self_attn.o_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.31.self_attn.q_a_layernorm": false, "language_model.model.layers.31.self_attn.q_a_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.31.self_attn.q_b_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.31.self_attn.unembed_out": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.32.attn_hc.base": false, "language_model.model.layers.32.attn_hc.fn": false, "language_model.model.layers.32.attn_hc.scale": false, "language_model.model.layers.32.ffn_hc.base": false, "language_model.model.layers.32.ffn_hc.fn": false, "language_model.model.layers.32.ffn_hc.scale": false, "language_model.model.layers.32.input_layernorm": false, "language_model.model.layers.32.mlp.gate": false, "language_model.model.layers.32.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.32.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.32.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.32.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.32.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.32.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.32.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.32.post_attention_layernorm": false, "language_model.model.layers.32.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.32.self_attn.forget_gate.A_log": false, "language_model.model.layers.32.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.32.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.32.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.32.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.32.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.32.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.32.self_attn.o_norm": false, "language_model.model.layers.32.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.32.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.32.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.33.attn_hc.base": false, "language_model.model.layers.33.attn_hc.fn": false, "language_model.model.layers.33.attn_hc.scale": false, "language_model.model.layers.33.ffn_hc.base": false, "language_model.model.layers.33.ffn_hc.fn": false, "language_model.model.layers.33.ffn_hc.scale": false, "language_model.model.layers.33.input_layernorm": false, "language_model.model.layers.33.mlp.gate": false, "language_model.model.layers.33.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.33.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.33.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.33.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.33.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.33.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.33.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.33.post_attention_layernorm": false, "language_model.model.layers.33.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.33.self_attn.forget_gate.A_log": false, "language_model.model.layers.33.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.33.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.33.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.33.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.33.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.33.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.33.self_attn.o_norm": false, "language_model.model.layers.33.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.33.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.33.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.34.attn_hc.base": false, "language_model.model.layers.34.attn_hc.fn": false, "language_model.model.layers.34.attn_hc.scale": false, "language_model.model.layers.34.ffn_hc.base": false, "language_model.model.layers.34.ffn_hc.fn": false, "language_model.model.layers.34.ffn_hc.scale": false, "language_model.model.layers.34.input_layernorm": false, "language_model.model.layers.34.mlp.gate": false, "language_model.model.layers.34.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.34.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.34.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.34.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.34.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.34.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.34.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.34.post_attention_layernorm": false, "language_model.model.layers.34.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.34.self_attn.forget_gate.A_log": false, "language_model.model.layers.34.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.34.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.34.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.34.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.34.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.34.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.34.self_attn.o_norm": false, "language_model.model.layers.34.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.34.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.34.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.35.attn_hc.base": false, "language_model.model.layers.35.attn_hc.fn": false, "language_model.model.layers.35.attn_hc.scale": false, "language_model.model.layers.35.ffn_hc.base": false, "language_model.model.layers.35.ffn_hc.fn": false, "language_model.model.layers.35.ffn_hc.scale": false, "language_model.model.layers.35.input_layernorm": false, "language_model.model.layers.35.mlp.gate": false, "language_model.model.layers.35.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.35.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.35.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.35.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.35.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.35.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.35.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.35.post_attention_layernorm": false, "language_model.model.layers.35.self_attn.embed_q": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.35.self_attn.indexer.index_kpool_compress_ape": false, "language_model.model.layers.35.self_attn.indexer.index_kpool_compress_gate": false, "language_model.model.layers.35.self_attn.indexer.k_norm": false, "language_model.model.layers.35.self_attn.indexer.k_norm.bias": false, "language_model.model.layers.35.self_attn.indexer.weights_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.35.self_attn.indexer.wk": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.35.self_attn.indexer.wq_b": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.35.self_attn.kv_a_layernorm": false, "language_model.model.layers.35.self_attn.kv_a_proj_with_mqa": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.35.self_attn.o_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.35.self_attn.q_a_layernorm": false, "language_model.model.layers.35.self_attn.q_a_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.35.self_attn.q_b_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.35.self_attn.unembed_out": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.36.attn_hc.base": false, "language_model.model.layers.36.attn_hc.fn": false, "language_model.model.layers.36.attn_hc.scale": false, "language_model.model.layers.36.ffn_hc.base": false, "language_model.model.layers.36.ffn_hc.fn": false, "language_model.model.layers.36.ffn_hc.scale": false, "language_model.model.layers.36.input_layernorm": false, "language_model.model.layers.36.mlp.gate": false, "language_model.model.layers.36.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.36.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.36.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.36.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.36.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.36.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.36.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.36.post_attention_layernorm": false, "language_model.model.layers.36.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.36.self_attn.forget_gate.A_log": false, "language_model.model.layers.36.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.36.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.36.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.36.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.36.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.36.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.36.self_attn.o_norm": false, "language_model.model.layers.36.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.36.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.36.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.37.attn_hc.base": false, "language_model.model.layers.37.attn_hc.fn": false, "language_model.model.layers.37.attn_hc.scale": false, "language_model.model.layers.37.ffn_hc.base": false, "language_model.model.layers.37.ffn_hc.fn": false, "language_model.model.layers.37.ffn_hc.scale": false, "language_model.model.layers.37.input_layernorm": false, "language_model.model.layers.37.mlp.gate": false, "language_model.model.layers.37.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.37.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.37.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.37.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.37.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.37.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.37.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.37.post_attention_layernorm": false, "language_model.model.layers.37.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.37.self_attn.forget_gate.A_log": false, "language_model.model.layers.37.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.37.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.37.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.37.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.37.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.37.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.37.self_attn.o_norm": false, "language_model.model.layers.37.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.37.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.37.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.38.attn_hc.base": false, "language_model.model.layers.38.attn_hc.fn": false, "language_model.model.layers.38.attn_hc.scale": false, "language_model.model.layers.38.ffn_hc.base": false, "language_model.model.layers.38.ffn_hc.fn": false, "language_model.model.layers.38.ffn_hc.scale": false, "language_model.model.layers.38.input_layernorm": false, "language_model.model.layers.38.mlp.gate": false, "language_model.model.layers.38.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.38.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.38.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.38.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.38.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.38.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.38.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.38.post_attention_layernorm": false, "language_model.model.layers.38.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.38.self_attn.forget_gate.A_log": false, "language_model.model.layers.38.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.38.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.38.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.38.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.38.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.38.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.38.self_attn.o_norm": false, "language_model.model.layers.38.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.38.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.38.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.39.attn_hc.base": false, "language_model.model.layers.39.attn_hc.fn": false, "language_model.model.layers.39.attn_hc.scale": false, "language_model.model.layers.39.ffn_hc.base": false, "language_model.model.layers.39.ffn_hc.fn": false, "language_model.model.layers.39.ffn_hc.scale": false, "language_model.model.layers.39.input_layernorm": false, "language_model.model.layers.39.mlp.gate": false, "language_model.model.layers.39.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.39.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.39.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.39.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.39.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.39.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.39.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.39.post_attention_layernorm": false, "language_model.model.layers.39.self_attn.embed_q": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.39.self_attn.indexer.index_kpool_compress_ape": false, "language_model.model.layers.39.self_attn.indexer.index_kpool_compress_gate": false, "language_model.model.layers.39.self_attn.indexer.k_norm": false, "language_model.model.layers.39.self_attn.indexer.k_norm.bias": false, "language_model.model.layers.39.self_attn.indexer.weights_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.39.self_attn.indexer.wk": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.39.self_attn.indexer.wq_b": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.39.self_attn.kv_a_layernorm": false, "language_model.model.layers.39.self_attn.kv_a_proj_with_mqa": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.39.self_attn.o_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.39.self_attn.q_a_layernorm": false, "language_model.model.layers.39.self_attn.q_a_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.39.self_attn.q_b_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.39.self_attn.unembed_out": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.4.attn_hc.base": false, "language_model.model.layers.4.attn_hc.fn": false, "language_model.model.layers.4.attn_hc.scale": false, "language_model.model.layers.4.ffn_hc.base": false, "language_model.model.layers.4.ffn_hc.fn": false, "language_model.model.layers.4.ffn_hc.scale": false, "language_model.model.layers.4.input_layernorm": false, "language_model.model.layers.4.mlp.gate": false, "language_model.model.layers.4.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.4.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.4.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.4.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.4.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.4.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.4.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.4.post_attention_layernorm": false, "language_model.model.layers.4.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.4.self_attn.forget_gate.A_log": false, "language_model.model.layers.4.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.4.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.4.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.4.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.4.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.4.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.4.self_attn.o_norm": false, "language_model.model.layers.4.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.4.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.4.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.40.attn_hc.base": false, "language_model.model.layers.40.attn_hc.fn": false, "language_model.model.layers.40.attn_hc.scale": false, "language_model.model.layers.40.ffn_hc.base": false, "language_model.model.layers.40.ffn_hc.fn": false, "language_model.model.layers.40.ffn_hc.scale": false, "language_model.model.layers.40.input_layernorm": false, "language_model.model.layers.40.mlp.gate": false, "language_model.model.layers.40.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.40.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.40.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.40.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.40.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.40.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.40.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.40.post_attention_layernorm": false, "language_model.model.layers.40.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.40.self_attn.forget_gate.A_log": false, "language_model.model.layers.40.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.40.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.40.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.40.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.40.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.40.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.40.self_attn.o_norm": false, "language_model.model.layers.40.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.40.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.40.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.41.attn_hc.base": false, "language_model.model.layers.41.attn_hc.fn": false, "language_model.model.layers.41.attn_hc.scale": false, "language_model.model.layers.41.ffn_hc.base": false, "language_model.model.layers.41.ffn_hc.fn": false, "language_model.model.layers.41.ffn_hc.scale": false, "language_model.model.layers.41.input_layernorm": false, "language_model.model.layers.41.mlp.gate": false, "language_model.model.layers.41.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.41.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.41.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.41.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.41.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.41.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.41.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.41.post_attention_layernorm": false, "language_model.model.layers.41.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.41.self_attn.forget_gate.A_log": false, "language_model.model.layers.41.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.41.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.41.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.41.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.41.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.41.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.41.self_attn.o_norm": false, "language_model.model.layers.41.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.41.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.41.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.42.attn_hc.base": false, "language_model.model.layers.42.attn_hc.fn": false, "language_model.model.layers.42.attn_hc.scale": false, "language_model.model.layers.42.ffn_hc.base": false, "language_model.model.layers.42.ffn_hc.fn": false, "language_model.model.layers.42.ffn_hc.scale": false, "language_model.model.layers.42.input_layernorm": false, "language_model.model.layers.42.mlp.gate": false, "language_model.model.layers.42.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.42.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.42.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.42.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.42.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.42.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.42.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.42.post_attention_layernorm": false, "language_model.model.layers.42.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.42.self_attn.forget_gate.A_log": false, "language_model.model.layers.42.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.42.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.42.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.42.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.42.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.42.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.42.self_attn.o_norm": false, "language_model.model.layers.42.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.42.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.42.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.43.attn_hc.base": false, "language_model.model.layers.43.attn_hc.fn": false, "language_model.model.layers.43.attn_hc.scale": false, "language_model.model.layers.43.ffn_hc.base": false, "language_model.model.layers.43.ffn_hc.fn": false, "language_model.model.layers.43.ffn_hc.scale": false, "language_model.model.layers.43.input_layernorm": false, "language_model.model.layers.43.mlp.gate": false, "language_model.model.layers.43.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.43.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.43.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.43.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.43.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.43.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.43.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.43.post_attention_layernorm": false, "language_model.model.layers.43.self_attn.embed_q": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.43.self_attn.indexer.index_kpool_compress_ape": false, "language_model.model.layers.43.self_attn.indexer.index_kpool_compress_gate": false, "language_model.model.layers.43.self_attn.indexer.k_norm": false, "language_model.model.layers.43.self_attn.indexer.k_norm.bias": false, "language_model.model.layers.43.self_attn.indexer.weights_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.43.self_attn.indexer.wk": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.43.self_attn.indexer.wq_b": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.43.self_attn.kv_a_layernorm": false, "language_model.model.layers.43.self_attn.kv_a_proj_with_mqa": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.43.self_attn.o_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.43.self_attn.q_a_layernorm": false, "language_model.model.layers.43.self_attn.q_a_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.43.self_attn.q_b_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.43.self_attn.unembed_out": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.44.attn_hc.base": false, "language_model.model.layers.44.attn_hc.fn": false, "language_model.model.layers.44.attn_hc.scale": false, "language_model.model.layers.44.ffn_hc.base": false, "language_model.model.layers.44.ffn_hc.fn": false, "language_model.model.layers.44.ffn_hc.scale": false, "language_model.model.layers.44.input_layernorm": false, "language_model.model.layers.44.mlp.gate": false, "language_model.model.layers.44.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.44.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.44.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.44.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.44.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.44.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.44.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.44.post_attention_layernorm": false, "language_model.model.layers.44.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.44.self_attn.forget_gate.A_log": false, "language_model.model.layers.44.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.44.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.44.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.44.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.44.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.44.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.44.self_attn.o_norm": false, "language_model.model.layers.44.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.44.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.44.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.5.attn_hc.base": false, "language_model.model.layers.5.attn_hc.fn": false, "language_model.model.layers.5.attn_hc.scale": false, "language_model.model.layers.5.ffn_hc.base": false, "language_model.model.layers.5.ffn_hc.fn": false, "language_model.model.layers.5.ffn_hc.scale": false, "language_model.model.layers.5.input_layernorm": false, "language_model.model.layers.5.mlp.gate": false, "language_model.model.layers.5.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.5.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.5.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.5.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.5.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.5.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.5.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.5.post_attention_layernorm": false, "language_model.model.layers.5.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.5.self_attn.forget_gate.A_log": false, "language_model.model.layers.5.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.5.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.5.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.5.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.5.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.5.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.5.self_attn.o_norm": false, "language_model.model.layers.5.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.5.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.5.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.6.attn_hc.base": false, "language_model.model.layers.6.attn_hc.fn": false, "language_model.model.layers.6.attn_hc.scale": false, "language_model.model.layers.6.ffn_hc.base": false, "language_model.model.layers.6.ffn_hc.fn": false, "language_model.model.layers.6.ffn_hc.scale": false, "language_model.model.layers.6.input_layernorm": false, "language_model.model.layers.6.mlp.gate": false, "language_model.model.layers.6.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.6.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.6.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.6.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.6.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.6.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.6.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.6.post_attention_layernorm": false, "language_model.model.layers.6.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.6.self_attn.forget_gate.A_log": false, "language_model.model.layers.6.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.6.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.6.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.6.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.6.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.6.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.6.self_attn.o_norm": false, "language_model.model.layers.6.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.6.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.6.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.7.attn_hc.base": false, "language_model.model.layers.7.attn_hc.fn": false, "language_model.model.layers.7.attn_hc.scale": false, "language_model.model.layers.7.ffn_hc.base": false, "language_model.model.layers.7.ffn_hc.fn": false, "language_model.model.layers.7.ffn_hc.scale": false, "language_model.model.layers.7.input_layernorm": false, "language_model.model.layers.7.mlp.gate": false, "language_model.model.layers.7.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.7.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.7.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.7.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.7.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.7.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.7.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.7.post_attention_layernorm": false, "language_model.model.layers.7.self_attn.embed_q": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.7.self_attn.indexer.index_kpool_compress_ape": false, "language_model.model.layers.7.self_attn.indexer.index_kpool_compress_gate": false, "language_model.model.layers.7.self_attn.indexer.k_norm": false, "language_model.model.layers.7.self_attn.indexer.k_norm.bias": false, "language_model.model.layers.7.self_attn.indexer.weights_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.7.self_attn.indexer.wk": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.7.self_attn.indexer.wq_b": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.7.self_attn.kv_a_layernorm": false, "language_model.model.layers.7.self_attn.kv_a_proj_with_mqa": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.7.self_attn.o_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.7.self_attn.q_a_layernorm": false, "language_model.model.layers.7.self_attn.q_a_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.7.self_attn.q_b_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.7.self_attn.unembed_out": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.8.attn_hc.base": false, "language_model.model.layers.8.attn_hc.fn": false, "language_model.model.layers.8.attn_hc.scale": false, "language_model.model.layers.8.ffn_hc.base": false, "language_model.model.layers.8.ffn_hc.fn": false, "language_model.model.layers.8.ffn_hc.scale": false, "language_model.model.layers.8.input_layernorm": false, "language_model.model.layers.8.mlp.gate": false, "language_model.model.layers.8.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.8.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.8.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.8.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.8.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.8.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.8.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.8.post_attention_layernorm": false, "language_model.model.layers.8.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.8.self_attn.forget_gate.A_log": false, "language_model.model.layers.8.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.8.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.8.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.8.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.8.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.8.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.8.self_attn.o_norm": false, "language_model.model.layers.8.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.8.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.8.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.9.attn_hc.base": false, "language_model.model.layers.9.attn_hc.fn": false, "language_model.model.layers.9.attn_hc.scale": false, "language_model.model.layers.9.ffn_hc.base": false, "language_model.model.layers.9.ffn_hc.fn": false, "language_model.model.layers.9.ffn_hc.scale": false, "language_model.model.layers.9.input_layernorm": false, "language_model.model.layers.9.mlp.gate": false, "language_model.model.layers.9.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.9.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.9.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.9.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.9.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.9.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.9.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.9.post_attention_layernorm": false, "language_model.model.layers.9.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.9.self_attn.forget_gate.A_log": false, "language_model.model.layers.9.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.9.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.9.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.9.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.9.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.9.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.9.self_attn.o_norm": false, "language_model.model.layers.9.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.9.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.9.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.norm": false }, "quantization": { "group_size": 64, "bits": 4, "mode": "affine", "language_model.lm_head": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.embed_tokens": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.0.attn_hc.base": false, "language_model.model.layers.0.attn_hc.fn": false, "language_model.model.layers.0.attn_hc.scale": false, "language_model.model.layers.0.ffn_hc.base": false, "language_model.model.layers.0.ffn_hc.fn": false, "language_model.model.layers.0.ffn_hc.scale": false, "language_model.model.layers.0.input_layernorm": false, "language_model.model.layers.0.mlp.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.0.mlp.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.0.mlp.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.0.post_attention_layernorm": false, "language_model.model.layers.0.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.0.self_attn.forget_gate.A_log": false, "language_model.model.layers.0.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.0.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.0.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.0.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.0.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.0.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.0.self_attn.o_norm": false, "language_model.model.layers.0.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.0.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.0.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.1.attn_hc.base": false, "language_model.model.layers.1.attn_hc.fn": false, "language_model.model.layers.1.attn_hc.scale": false, "language_model.model.layers.1.ffn_hc.base": false, "language_model.model.layers.1.ffn_hc.fn": false, "language_model.model.layers.1.ffn_hc.scale": false, "language_model.model.layers.1.input_layernorm": false, "language_model.model.layers.1.mlp.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.1.mlp.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.1.mlp.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.1.post_attention_layernorm": false, "language_model.model.layers.1.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.1.self_attn.forget_gate.A_log": false, "language_model.model.layers.1.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.1.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.1.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.1.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.1.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.1.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.1.self_attn.o_norm": false, "language_model.model.layers.1.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.1.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.1.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.10.attn_hc.base": false, "language_model.model.layers.10.attn_hc.fn": false, "language_model.model.layers.10.attn_hc.scale": false, "language_model.model.layers.10.ffn_hc.base": false, "language_model.model.layers.10.ffn_hc.fn": false, "language_model.model.layers.10.ffn_hc.scale": false, "language_model.model.layers.10.input_layernorm": false, "language_model.model.layers.10.mlp.gate": false, "language_model.model.layers.10.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.10.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.10.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.10.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.10.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.10.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.10.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.10.post_attention_layernorm": false, "language_model.model.layers.10.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.10.self_attn.forget_gate.A_log": false, "language_model.model.layers.10.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.10.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.10.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.10.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.10.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.10.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.10.self_attn.o_norm": false, "language_model.model.layers.10.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.10.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.10.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.11.attn_hc.base": false, "language_model.model.layers.11.attn_hc.fn": false, "language_model.model.layers.11.attn_hc.scale": false, "language_model.model.layers.11.ffn_hc.base": false, "language_model.model.layers.11.ffn_hc.fn": false, "language_model.model.layers.11.ffn_hc.scale": false, "language_model.model.layers.11.input_layernorm": false, "language_model.model.layers.11.mlp.gate": false, "language_model.model.layers.11.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.11.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.11.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.11.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.11.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.11.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.11.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.11.post_attention_layernorm": false, "language_model.model.layers.11.self_attn.embed_q": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.11.self_attn.indexer.index_kpool_compress_ape": false, "language_model.model.layers.11.self_attn.indexer.index_kpool_compress_gate": false, "language_model.model.layers.11.self_attn.indexer.k_norm": false, "language_model.model.layers.11.self_attn.indexer.k_norm.bias": false, "language_model.model.layers.11.self_attn.indexer.weights_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.11.self_attn.indexer.wk": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.11.self_attn.indexer.wq_b": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.11.self_attn.kv_a_layernorm": false, "language_model.model.layers.11.self_attn.kv_a_proj_with_mqa": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.11.self_attn.o_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.11.self_attn.q_a_layernorm": false, "language_model.model.layers.11.self_attn.q_a_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.11.self_attn.q_b_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.11.self_attn.unembed_out": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.12.attn_hc.base": false, "language_model.model.layers.12.attn_hc.fn": false, "language_model.model.layers.12.attn_hc.scale": false, "language_model.model.layers.12.ffn_hc.base": false, "language_model.model.layers.12.ffn_hc.fn": false, "language_model.model.layers.12.ffn_hc.scale": false, "language_model.model.layers.12.input_layernorm": false, "language_model.model.layers.12.mlp.gate": false, "language_model.model.layers.12.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.12.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.12.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.12.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.12.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.12.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.12.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.12.post_attention_layernorm": false, "language_model.model.layers.12.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.12.self_attn.forget_gate.A_log": false, "language_model.model.layers.12.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.12.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.12.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.12.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.12.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.12.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.12.self_attn.o_norm": false, "language_model.model.layers.12.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.12.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.12.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.13.attn_hc.base": false, "language_model.model.layers.13.attn_hc.fn": false, "language_model.model.layers.13.attn_hc.scale": false, "language_model.model.layers.13.ffn_hc.base": false, "language_model.model.layers.13.ffn_hc.fn": false, "language_model.model.layers.13.ffn_hc.scale": false, "language_model.model.layers.13.input_layernorm": false, "language_model.model.layers.13.mlp.gate": false, "language_model.model.layers.13.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.13.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.13.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.13.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.13.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.13.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.13.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.13.post_attention_layernorm": false, "language_model.model.layers.13.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.13.self_attn.forget_gate.A_log": false, "language_model.model.layers.13.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.13.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.13.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.13.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.13.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.13.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.13.self_attn.o_norm": false, "language_model.model.layers.13.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.13.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.13.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.14.attn_hc.base": false, "language_model.model.layers.14.attn_hc.fn": false, "language_model.model.layers.14.attn_hc.scale": false, "language_model.model.layers.14.ffn_hc.base": false, "language_model.model.layers.14.ffn_hc.fn": false, "language_model.model.layers.14.ffn_hc.scale": false, "language_model.model.layers.14.input_layernorm": false, "language_model.model.layers.14.mlp.gate": false, "language_model.model.layers.14.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.14.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.14.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.14.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.14.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.14.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.14.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.14.post_attention_layernorm": false, "language_model.model.layers.14.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.14.self_attn.forget_gate.A_log": false, "language_model.model.layers.14.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.14.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.14.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.14.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.14.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.14.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.14.self_attn.o_norm": false, "language_model.model.layers.14.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.14.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.14.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.15.attn_hc.base": false, "language_model.model.layers.15.attn_hc.fn": false, "language_model.model.layers.15.attn_hc.scale": false, "language_model.model.layers.15.ffn_hc.base": false, "language_model.model.layers.15.ffn_hc.fn": false, "language_model.model.layers.15.ffn_hc.scale": false, "language_model.model.layers.15.input_layernorm": false, "language_model.model.layers.15.mlp.gate": false, "language_model.model.layers.15.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.15.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.15.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.15.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.15.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.15.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.15.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.15.post_attention_layernorm": false, "language_model.model.layers.15.self_attn.embed_q": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.15.self_attn.indexer.index_kpool_compress_ape": false, "language_model.model.layers.15.self_attn.indexer.index_kpool_compress_gate": false, "language_model.model.layers.15.self_attn.indexer.k_norm": false, "language_model.model.layers.15.self_attn.indexer.k_norm.bias": false, "language_model.model.layers.15.self_attn.indexer.weights_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.15.self_attn.indexer.wk": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.15.self_attn.indexer.wq_b": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.15.self_attn.kv_a_layernorm": false, "language_model.model.layers.15.self_attn.kv_a_proj_with_mqa": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.15.self_attn.o_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.15.self_attn.q_a_layernorm": false, "language_model.model.layers.15.self_attn.q_a_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.15.self_attn.q_b_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.15.self_attn.unembed_out": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.16.attn_hc.base": false, "language_model.model.layers.16.attn_hc.fn": false, "language_model.model.layers.16.attn_hc.scale": false, "language_model.model.layers.16.ffn_hc.base": false, "language_model.model.layers.16.ffn_hc.fn": false, "language_model.model.layers.16.ffn_hc.scale": false, "language_model.model.layers.16.input_layernorm": false, "language_model.model.layers.16.mlp.gate": false, "language_model.model.layers.16.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.16.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.16.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.16.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.16.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.16.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.16.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.16.post_attention_layernorm": false, "language_model.model.layers.16.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.16.self_attn.forget_gate.A_log": false, "language_model.model.layers.16.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.16.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.16.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.16.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.16.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.16.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.16.self_attn.o_norm": false, "language_model.model.layers.16.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.16.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.16.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.17.attn_hc.base": false, "language_model.model.layers.17.attn_hc.fn": false, "language_model.model.layers.17.attn_hc.scale": false, "language_model.model.layers.17.ffn_hc.base": false, "language_model.model.layers.17.ffn_hc.fn": false, "language_model.model.layers.17.ffn_hc.scale": false, "language_model.model.layers.17.input_layernorm": false, "language_model.model.layers.17.mlp.gate": false, "language_model.model.layers.17.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.17.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.17.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.17.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.17.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.17.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.17.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.17.post_attention_layernorm": false, "language_model.model.layers.17.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.17.self_attn.forget_gate.A_log": false, "language_model.model.layers.17.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.17.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.17.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.17.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.17.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.17.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.17.self_attn.o_norm": false, "language_model.model.layers.17.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.17.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.17.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.18.attn_hc.base": false, "language_model.model.layers.18.attn_hc.fn": false, "language_model.model.layers.18.attn_hc.scale": false, "language_model.model.layers.18.ffn_hc.base": false, "language_model.model.layers.18.ffn_hc.fn": false, "language_model.model.layers.18.ffn_hc.scale": false, "language_model.model.layers.18.input_layernorm": false, "language_model.model.layers.18.mlp.gate": false, "language_model.model.layers.18.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.18.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.18.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.18.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.18.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.18.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.18.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.18.post_attention_layernorm": false, "language_model.model.layers.18.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.18.self_attn.forget_gate.A_log": false, "language_model.model.layers.18.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.18.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.18.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.18.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.18.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.18.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.18.self_attn.o_norm": false, "language_model.model.layers.18.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.18.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.18.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.19.attn_hc.base": false, "language_model.model.layers.19.attn_hc.fn": false, "language_model.model.layers.19.attn_hc.scale": false, "language_model.model.layers.19.ffn_hc.base": false, "language_model.model.layers.19.ffn_hc.fn": false, "language_model.model.layers.19.ffn_hc.scale": false, "language_model.model.layers.19.input_layernorm": false, "language_model.model.layers.19.mlp.gate": false, "language_model.model.layers.19.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.19.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.19.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.19.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.19.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.19.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.19.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.19.post_attention_layernorm": false, "language_model.model.layers.19.self_attn.embed_q": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.19.self_attn.indexer.index_kpool_compress_ape": false, "language_model.model.layers.19.self_attn.indexer.index_kpool_compress_gate": false, "language_model.model.layers.19.self_attn.indexer.k_norm": false, "language_model.model.layers.19.self_attn.indexer.k_norm.bias": false, "language_model.model.layers.19.self_attn.indexer.weights_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.19.self_attn.indexer.wk": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.19.self_attn.indexer.wq_b": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.19.self_attn.kv_a_layernorm": false, "language_model.model.layers.19.self_attn.kv_a_proj_with_mqa": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.19.self_attn.o_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.19.self_attn.q_a_layernorm": false, "language_model.model.layers.19.self_attn.q_a_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.19.self_attn.q_b_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.19.self_attn.unembed_out": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.2.attn_hc.base": false, "language_model.model.layers.2.attn_hc.fn": false, "language_model.model.layers.2.attn_hc.scale": false, "language_model.model.layers.2.ffn_hc.base": false, "language_model.model.layers.2.ffn_hc.fn": false, "language_model.model.layers.2.ffn_hc.scale": false, "language_model.model.layers.2.input_layernorm": false, "language_model.model.layers.2.mlp.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.2.mlp.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.2.mlp.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.2.post_attention_layernorm": false, "language_model.model.layers.2.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.2.self_attn.forget_gate.A_log": false, "language_model.model.layers.2.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.2.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.2.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.2.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.2.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.2.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.2.self_attn.o_norm": false, "language_model.model.layers.2.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.2.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.2.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.20.attn_hc.base": false, "language_model.model.layers.20.attn_hc.fn": false, "language_model.model.layers.20.attn_hc.scale": false, "language_model.model.layers.20.ffn_hc.base": false, "language_model.model.layers.20.ffn_hc.fn": false, "language_model.model.layers.20.ffn_hc.scale": false, "language_model.model.layers.20.input_layernorm": false, "language_model.model.layers.20.mlp.gate": false, "language_model.model.layers.20.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.20.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.20.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.20.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.20.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.20.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.20.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.20.post_attention_layernorm": false, "language_model.model.layers.20.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.20.self_attn.forget_gate.A_log": false, "language_model.model.layers.20.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.20.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.20.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.20.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.20.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.20.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.20.self_attn.o_norm": false, "language_model.model.layers.20.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.20.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.20.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.21.attn_hc.base": false, "language_model.model.layers.21.attn_hc.fn": false, "language_model.model.layers.21.attn_hc.scale": false, "language_model.model.layers.21.ffn_hc.base": false, "language_model.model.layers.21.ffn_hc.fn": false, "language_model.model.layers.21.ffn_hc.scale": false, "language_model.model.layers.21.input_layernorm": false, "language_model.model.layers.21.mlp.gate": false, "language_model.model.layers.21.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.21.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.21.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.21.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.21.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.21.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.21.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.21.post_attention_layernorm": false, "language_model.model.layers.21.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.21.self_attn.forget_gate.A_log": false, "language_model.model.layers.21.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.21.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.21.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.21.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.21.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.21.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.21.self_attn.o_norm": false, "language_model.model.layers.21.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.21.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.21.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.22.attn_hc.base": false, "language_model.model.layers.22.attn_hc.fn": false, "language_model.model.layers.22.attn_hc.scale": false, "language_model.model.layers.22.ffn_hc.base": false, "language_model.model.layers.22.ffn_hc.fn": false, "language_model.model.layers.22.ffn_hc.scale": false, "language_model.model.layers.22.input_layernorm": false, "language_model.model.layers.22.mlp.gate": false, "language_model.model.layers.22.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.22.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.22.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.22.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.22.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.22.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.22.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.22.post_attention_layernorm": false, "language_model.model.layers.22.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.22.self_attn.forget_gate.A_log": false, "language_model.model.layers.22.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.22.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.22.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.22.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.22.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.22.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.22.self_attn.o_norm": false, "language_model.model.layers.22.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.22.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.22.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.23.attn_hc.base": false, "language_model.model.layers.23.attn_hc.fn": false, "language_model.model.layers.23.attn_hc.scale": false, "language_model.model.layers.23.ffn_hc.base": false, "language_model.model.layers.23.ffn_hc.fn": false, "language_model.model.layers.23.ffn_hc.scale": false, "language_model.model.layers.23.input_layernorm": false, "language_model.model.layers.23.mlp.gate": false, "language_model.model.layers.23.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.23.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.23.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.23.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.23.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.23.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.23.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.23.post_attention_layernorm": false, "language_model.model.layers.23.self_attn.embed_q": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.23.self_attn.indexer.index_kpool_compress_ape": false, "language_model.model.layers.23.self_attn.indexer.index_kpool_compress_gate": false, "language_model.model.layers.23.self_attn.indexer.k_norm": false, "language_model.model.layers.23.self_attn.indexer.k_norm.bias": false, "language_model.model.layers.23.self_attn.indexer.weights_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.23.self_attn.indexer.wk": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.23.self_attn.indexer.wq_b": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.23.self_attn.kv_a_layernorm": false, "language_model.model.layers.23.self_attn.kv_a_proj_with_mqa": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.23.self_attn.o_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.23.self_attn.q_a_layernorm": false, "language_model.model.layers.23.self_attn.q_a_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.23.self_attn.q_b_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.23.self_attn.unembed_out": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.24.attn_hc.base": false, "language_model.model.layers.24.attn_hc.fn": false, "language_model.model.layers.24.attn_hc.scale": false, "language_model.model.layers.24.ffn_hc.base": false, "language_model.model.layers.24.ffn_hc.fn": false, "language_model.model.layers.24.ffn_hc.scale": false, "language_model.model.layers.24.input_layernorm": false, "language_model.model.layers.24.mlp.gate": false, "language_model.model.layers.24.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.24.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.24.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.24.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.24.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.24.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.24.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.24.post_attention_layernorm": false, "language_model.model.layers.24.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.24.self_attn.forget_gate.A_log": false, "language_model.model.layers.24.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.24.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.24.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.24.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.24.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.24.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.24.self_attn.o_norm": false, "language_model.model.layers.24.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.24.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.24.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.25.attn_hc.base": false, "language_model.model.layers.25.attn_hc.fn": false, "language_model.model.layers.25.attn_hc.scale": false, "language_model.model.layers.25.ffn_hc.base": false, "language_model.model.layers.25.ffn_hc.fn": false, "language_model.model.layers.25.ffn_hc.scale": false, "language_model.model.layers.25.input_layernorm": false, "language_model.model.layers.25.mlp.gate": false, "language_model.model.layers.25.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.25.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.25.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.25.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.25.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.25.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.25.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.25.post_attention_layernorm": false, "language_model.model.layers.25.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.25.self_attn.forget_gate.A_log": false, "language_model.model.layers.25.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.25.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.25.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.25.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.25.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.25.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.25.self_attn.o_norm": false, "language_model.model.layers.25.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.25.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.25.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.26.attn_hc.base": false, "language_model.model.layers.26.attn_hc.fn": false, "language_model.model.layers.26.attn_hc.scale": false, "language_model.model.layers.26.ffn_hc.base": false, "language_model.model.layers.26.ffn_hc.fn": false, "language_model.model.layers.26.ffn_hc.scale": false, "language_model.model.layers.26.input_layernorm": false, "language_model.model.layers.26.mlp.gate": false, "language_model.model.layers.26.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.26.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.26.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.26.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.26.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.26.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.26.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.26.post_attention_layernorm": false, "language_model.model.layers.26.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.26.self_attn.forget_gate.A_log": false, "language_model.model.layers.26.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.26.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.26.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.26.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.26.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.26.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.26.self_attn.o_norm": false, "language_model.model.layers.26.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.26.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.26.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.27.attn_hc.base": false, "language_model.model.layers.27.attn_hc.fn": false, "language_model.model.layers.27.attn_hc.scale": false, "language_model.model.layers.27.ffn_hc.base": false, "language_model.model.layers.27.ffn_hc.fn": false, "language_model.model.layers.27.ffn_hc.scale": false, "language_model.model.layers.27.input_layernorm": false, "language_model.model.layers.27.mlp.gate": false, "language_model.model.layers.27.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.27.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.27.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.27.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.27.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.27.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.27.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.27.post_attention_layernorm": false, "language_model.model.layers.27.self_attn.embed_q": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.27.self_attn.indexer.index_kpool_compress_ape": false, "language_model.model.layers.27.self_attn.indexer.index_kpool_compress_gate": false, "language_model.model.layers.27.self_attn.indexer.k_norm": false, "language_model.model.layers.27.self_attn.indexer.k_norm.bias": false, "language_model.model.layers.27.self_attn.indexer.weights_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.27.self_attn.indexer.wk": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.27.self_attn.indexer.wq_b": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.27.self_attn.kv_a_layernorm": false, "language_model.model.layers.27.self_attn.kv_a_proj_with_mqa": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.27.self_attn.o_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.27.self_attn.q_a_layernorm": false, "language_model.model.layers.27.self_attn.q_a_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.27.self_attn.q_b_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.27.self_attn.unembed_out": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.28.attn_hc.base": false, "language_model.model.layers.28.attn_hc.fn": false, "language_model.model.layers.28.attn_hc.scale": false, "language_model.model.layers.28.ffn_hc.base": false, "language_model.model.layers.28.ffn_hc.fn": false, "language_model.model.layers.28.ffn_hc.scale": false, "language_model.model.layers.28.input_layernorm": false, "language_model.model.layers.28.mlp.gate": false, "language_model.model.layers.28.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.28.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.28.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.28.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.28.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.28.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.28.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.28.post_attention_layernorm": false, "language_model.model.layers.28.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.28.self_attn.forget_gate.A_log": false, "language_model.model.layers.28.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.28.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.28.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.28.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.28.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.28.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.28.self_attn.o_norm": false, "language_model.model.layers.28.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.28.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.28.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.29.attn_hc.base": false, "language_model.model.layers.29.attn_hc.fn": false, "language_model.model.layers.29.attn_hc.scale": false, "language_model.model.layers.29.ffn_hc.base": false, "language_model.model.layers.29.ffn_hc.fn": false, "language_model.model.layers.29.ffn_hc.scale": false, "language_model.model.layers.29.input_layernorm": false, "language_model.model.layers.29.mlp.gate": false, "language_model.model.layers.29.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.29.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.29.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.29.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.29.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.29.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.29.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.29.post_attention_layernorm": false, "language_model.model.layers.29.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.29.self_attn.forget_gate.A_log": false, "language_model.model.layers.29.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.29.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.29.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.29.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.29.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.29.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.29.self_attn.o_norm": false, "language_model.model.layers.29.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.29.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.29.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.3.attn_hc.base": false, "language_model.model.layers.3.attn_hc.fn": false, "language_model.model.layers.3.attn_hc.scale": false, "language_model.model.layers.3.ffn_hc.base": false, "language_model.model.layers.3.ffn_hc.fn": false, "language_model.model.layers.3.ffn_hc.scale": false, "language_model.model.layers.3.input_layernorm": false, "language_model.model.layers.3.mlp.gate": false, "language_model.model.layers.3.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.3.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.3.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.3.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.3.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.3.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.3.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.3.post_attention_layernorm": false, "language_model.model.layers.3.self_attn.embed_q": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.3.self_attn.indexer.index_kpool_compress_ape": false, "language_model.model.layers.3.self_attn.indexer.index_kpool_compress_gate": false, "language_model.model.layers.3.self_attn.indexer.k_norm": false, "language_model.model.layers.3.self_attn.indexer.k_norm.bias": false, "language_model.model.layers.3.self_attn.indexer.weights_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.3.self_attn.indexer.wk": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.3.self_attn.indexer.wq_b": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.3.self_attn.kv_a_layernorm": false, "language_model.model.layers.3.self_attn.kv_a_proj_with_mqa": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.3.self_attn.o_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.3.self_attn.q_a_layernorm": false, "language_model.model.layers.3.self_attn.q_a_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.3.self_attn.q_b_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.3.self_attn.unembed_out": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.30.attn_hc.base": false, "language_model.model.layers.30.attn_hc.fn": false, "language_model.model.layers.30.attn_hc.scale": false, "language_model.model.layers.30.ffn_hc.base": false, "language_model.model.layers.30.ffn_hc.fn": false, "language_model.model.layers.30.ffn_hc.scale": false, "language_model.model.layers.30.input_layernorm": false, "language_model.model.layers.30.mlp.gate": false, "language_model.model.layers.30.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.30.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.30.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.30.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.30.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.30.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.30.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.30.post_attention_layernorm": false, "language_model.model.layers.30.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.30.self_attn.forget_gate.A_log": false, "language_model.model.layers.30.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.30.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.30.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.30.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.30.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.30.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.30.self_attn.o_norm": false, "language_model.model.layers.30.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.30.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.30.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.31.attn_hc.base": false, "language_model.model.layers.31.attn_hc.fn": false, "language_model.model.layers.31.attn_hc.scale": false, "language_model.model.layers.31.ffn_hc.base": false, "language_model.model.layers.31.ffn_hc.fn": false, "language_model.model.layers.31.ffn_hc.scale": false, "language_model.model.layers.31.input_layernorm": false, "language_model.model.layers.31.mlp.gate": false, "language_model.model.layers.31.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.31.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.31.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.31.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.31.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.31.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.31.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.31.post_attention_layernorm": false, "language_model.model.layers.31.self_attn.embed_q": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.31.self_attn.indexer.index_kpool_compress_ape": false, "language_model.model.layers.31.self_attn.indexer.index_kpool_compress_gate": false, "language_model.model.layers.31.self_attn.indexer.k_norm": false, "language_model.model.layers.31.self_attn.indexer.k_norm.bias": false, "language_model.model.layers.31.self_attn.indexer.weights_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.31.self_attn.indexer.wk": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.31.self_attn.indexer.wq_b": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.31.self_attn.kv_a_layernorm": false, "language_model.model.layers.31.self_attn.kv_a_proj_with_mqa": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.31.self_attn.o_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.31.self_attn.q_a_layernorm": false, "language_model.model.layers.31.self_attn.q_a_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.31.self_attn.q_b_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.31.self_attn.unembed_out": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.32.attn_hc.base": false, "language_model.model.layers.32.attn_hc.fn": false, "language_model.model.layers.32.attn_hc.scale": false, "language_model.model.layers.32.ffn_hc.base": false, "language_model.model.layers.32.ffn_hc.fn": false, "language_model.model.layers.32.ffn_hc.scale": false, "language_model.model.layers.32.input_layernorm": false, "language_model.model.layers.32.mlp.gate": false, "language_model.model.layers.32.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.32.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.32.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.32.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.32.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.32.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.32.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.32.post_attention_layernorm": false, "language_model.model.layers.32.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.32.self_attn.forget_gate.A_log": false, "language_model.model.layers.32.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.32.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.32.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.32.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.32.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.32.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.32.self_attn.o_norm": false, "language_model.model.layers.32.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.32.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.32.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.33.attn_hc.base": false, "language_model.model.layers.33.attn_hc.fn": false, "language_model.model.layers.33.attn_hc.scale": false, "language_model.model.layers.33.ffn_hc.base": false, "language_model.model.layers.33.ffn_hc.fn": false, "language_model.model.layers.33.ffn_hc.scale": false, "language_model.model.layers.33.input_layernorm": false, "language_model.model.layers.33.mlp.gate": false, "language_model.model.layers.33.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.33.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.33.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.33.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.33.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.33.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.33.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.33.post_attention_layernorm": false, "language_model.model.layers.33.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.33.self_attn.forget_gate.A_log": false, "language_model.model.layers.33.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.33.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.33.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.33.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.33.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.33.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.33.self_attn.o_norm": false, "language_model.model.layers.33.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.33.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.33.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.34.attn_hc.base": false, "language_model.model.layers.34.attn_hc.fn": false, "language_model.model.layers.34.attn_hc.scale": false, "language_model.model.layers.34.ffn_hc.base": false, "language_model.model.layers.34.ffn_hc.fn": false, "language_model.model.layers.34.ffn_hc.scale": false, "language_model.model.layers.34.input_layernorm": false, "language_model.model.layers.34.mlp.gate": false, "language_model.model.layers.34.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.34.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.34.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.34.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.34.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.34.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.34.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.34.post_attention_layernorm": false, "language_model.model.layers.34.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.34.self_attn.forget_gate.A_log": false, "language_model.model.layers.34.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.34.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.34.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.34.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.34.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.34.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.34.self_attn.o_norm": false, "language_model.model.layers.34.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.34.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.34.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.35.attn_hc.base": false, "language_model.model.layers.35.attn_hc.fn": false, "language_model.model.layers.35.attn_hc.scale": false, "language_model.model.layers.35.ffn_hc.base": false, "language_model.model.layers.35.ffn_hc.fn": false, "language_model.model.layers.35.ffn_hc.scale": false, "language_model.model.layers.35.input_layernorm": false, "language_model.model.layers.35.mlp.gate": false, "language_model.model.layers.35.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.35.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.35.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.35.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.35.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.35.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.35.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.35.post_attention_layernorm": false, "language_model.model.layers.35.self_attn.embed_q": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.35.self_attn.indexer.index_kpool_compress_ape": false, "language_model.model.layers.35.self_attn.indexer.index_kpool_compress_gate": false, "language_model.model.layers.35.self_attn.indexer.k_norm": false, "language_model.model.layers.35.self_attn.indexer.k_norm.bias": false, "language_model.model.layers.35.self_attn.indexer.weights_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.35.self_attn.indexer.wk": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.35.self_attn.indexer.wq_b": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.35.self_attn.kv_a_layernorm": false, "language_model.model.layers.35.self_attn.kv_a_proj_with_mqa": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.35.self_attn.o_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.35.self_attn.q_a_layernorm": false, "language_model.model.layers.35.self_attn.q_a_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.35.self_attn.q_b_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.35.self_attn.unembed_out": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.36.attn_hc.base": false, "language_model.model.layers.36.attn_hc.fn": false, "language_model.model.layers.36.attn_hc.scale": false, "language_model.model.layers.36.ffn_hc.base": false, "language_model.model.layers.36.ffn_hc.fn": false, "language_model.model.layers.36.ffn_hc.scale": false, "language_model.model.layers.36.input_layernorm": false, "language_model.model.layers.36.mlp.gate": false, "language_model.model.layers.36.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.36.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.36.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.36.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.36.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.36.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.36.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.36.post_attention_layernorm": false, "language_model.model.layers.36.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.36.self_attn.forget_gate.A_log": false, "language_model.model.layers.36.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.36.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.36.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.36.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.36.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.36.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.36.self_attn.o_norm": false, "language_model.model.layers.36.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.36.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.36.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.37.attn_hc.base": false, "language_model.model.layers.37.attn_hc.fn": false, "language_model.model.layers.37.attn_hc.scale": false, "language_model.model.layers.37.ffn_hc.base": false, "language_model.model.layers.37.ffn_hc.fn": false, "language_model.model.layers.37.ffn_hc.scale": false, "language_model.model.layers.37.input_layernorm": false, "language_model.model.layers.37.mlp.gate": false, "language_model.model.layers.37.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.37.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.37.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.37.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.37.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.37.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.37.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.37.post_attention_layernorm": false, "language_model.model.layers.37.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.37.self_attn.forget_gate.A_log": false, "language_model.model.layers.37.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.37.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.37.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.37.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.37.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.37.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.37.self_attn.o_norm": false, "language_model.model.layers.37.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.37.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.37.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.38.attn_hc.base": false, "language_model.model.layers.38.attn_hc.fn": false, "language_model.model.layers.38.attn_hc.scale": false, "language_model.model.layers.38.ffn_hc.base": false, "language_model.model.layers.38.ffn_hc.fn": false, "language_model.model.layers.38.ffn_hc.scale": false, "language_model.model.layers.38.input_layernorm": false, "language_model.model.layers.38.mlp.gate": false, "language_model.model.layers.38.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.38.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.38.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.38.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.38.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.38.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.38.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.38.post_attention_layernorm": false, "language_model.model.layers.38.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.38.self_attn.forget_gate.A_log": false, "language_model.model.layers.38.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.38.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.38.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.38.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.38.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.38.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.38.self_attn.o_norm": false, "language_model.model.layers.38.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.38.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.38.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.39.attn_hc.base": false, "language_model.model.layers.39.attn_hc.fn": false, "language_model.model.layers.39.attn_hc.scale": false, "language_model.model.layers.39.ffn_hc.base": false, "language_model.model.layers.39.ffn_hc.fn": false, "language_model.model.layers.39.ffn_hc.scale": false, "language_model.model.layers.39.input_layernorm": false, "language_model.model.layers.39.mlp.gate": false, "language_model.model.layers.39.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.39.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.39.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.39.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.39.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.39.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.39.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.39.post_attention_layernorm": false, "language_model.model.layers.39.self_attn.embed_q": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.39.self_attn.indexer.index_kpool_compress_ape": false, "language_model.model.layers.39.self_attn.indexer.index_kpool_compress_gate": false, "language_model.model.layers.39.self_attn.indexer.k_norm": false, "language_model.model.layers.39.self_attn.indexer.k_norm.bias": false, "language_model.model.layers.39.self_attn.indexer.weights_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.39.self_attn.indexer.wk": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.39.self_attn.indexer.wq_b": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.39.self_attn.kv_a_layernorm": false, "language_model.model.layers.39.self_attn.kv_a_proj_with_mqa": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.39.self_attn.o_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.39.self_attn.q_a_layernorm": false, "language_model.model.layers.39.self_attn.q_a_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.39.self_attn.q_b_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.39.self_attn.unembed_out": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.4.attn_hc.base": false, "language_model.model.layers.4.attn_hc.fn": false, "language_model.model.layers.4.attn_hc.scale": false, "language_model.model.layers.4.ffn_hc.base": false, "language_model.model.layers.4.ffn_hc.fn": false, "language_model.model.layers.4.ffn_hc.scale": false, "language_model.model.layers.4.input_layernorm": false, "language_model.model.layers.4.mlp.gate": false, "language_model.model.layers.4.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.4.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.4.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.4.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.4.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.4.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.4.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.4.post_attention_layernorm": false, "language_model.model.layers.4.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.4.self_attn.forget_gate.A_log": false, "language_model.model.layers.4.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.4.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.4.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.4.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.4.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.4.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.4.self_attn.o_norm": false, "language_model.model.layers.4.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.4.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.4.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.40.attn_hc.base": false, "language_model.model.layers.40.attn_hc.fn": false, "language_model.model.layers.40.attn_hc.scale": false, "language_model.model.layers.40.ffn_hc.base": false, "language_model.model.layers.40.ffn_hc.fn": false, "language_model.model.layers.40.ffn_hc.scale": false, "language_model.model.layers.40.input_layernorm": false, "language_model.model.layers.40.mlp.gate": false, "language_model.model.layers.40.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.40.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.40.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.40.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.40.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.40.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.40.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.40.post_attention_layernorm": false, "language_model.model.layers.40.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.40.self_attn.forget_gate.A_log": false, "language_model.model.layers.40.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.40.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.40.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.40.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.40.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.40.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.40.self_attn.o_norm": false, "language_model.model.layers.40.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.40.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.40.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.41.attn_hc.base": false, "language_model.model.layers.41.attn_hc.fn": false, "language_model.model.layers.41.attn_hc.scale": false, "language_model.model.layers.41.ffn_hc.base": false, "language_model.model.layers.41.ffn_hc.fn": false, "language_model.model.layers.41.ffn_hc.scale": false, "language_model.model.layers.41.input_layernorm": false, "language_model.model.layers.41.mlp.gate": false, "language_model.model.layers.41.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.41.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.41.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.41.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.41.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.41.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.41.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.41.post_attention_layernorm": false, "language_model.model.layers.41.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.41.self_attn.forget_gate.A_log": false, "language_model.model.layers.41.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.41.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.41.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.41.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.41.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.41.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.41.self_attn.o_norm": false, "language_model.model.layers.41.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.41.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.41.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.42.attn_hc.base": false, "language_model.model.layers.42.attn_hc.fn": false, "language_model.model.layers.42.attn_hc.scale": false, "language_model.model.layers.42.ffn_hc.base": false, "language_model.model.layers.42.ffn_hc.fn": false, "language_model.model.layers.42.ffn_hc.scale": false, "language_model.model.layers.42.input_layernorm": false, "language_model.model.layers.42.mlp.gate": false, "language_model.model.layers.42.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.42.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.42.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.42.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.42.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.42.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.42.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.42.post_attention_layernorm": false, "language_model.model.layers.42.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.42.self_attn.forget_gate.A_log": false, "language_model.model.layers.42.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.42.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.42.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.42.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.42.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.42.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.42.self_attn.o_norm": false, "language_model.model.layers.42.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.42.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.42.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.43.attn_hc.base": false, "language_model.model.layers.43.attn_hc.fn": false, "language_model.model.layers.43.attn_hc.scale": false, "language_model.model.layers.43.ffn_hc.base": false, "language_model.model.layers.43.ffn_hc.fn": false, "language_model.model.layers.43.ffn_hc.scale": false, "language_model.model.layers.43.input_layernorm": false, "language_model.model.layers.43.mlp.gate": false, "language_model.model.layers.43.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.43.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.43.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.43.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.43.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.43.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.43.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.43.post_attention_layernorm": false, "language_model.model.layers.43.self_attn.embed_q": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.43.self_attn.indexer.index_kpool_compress_ape": false, "language_model.model.layers.43.self_attn.indexer.index_kpool_compress_gate": false, "language_model.model.layers.43.self_attn.indexer.k_norm": false, "language_model.model.layers.43.self_attn.indexer.k_norm.bias": false, "language_model.model.layers.43.self_attn.indexer.weights_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.43.self_attn.indexer.wk": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.43.self_attn.indexer.wq_b": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.43.self_attn.kv_a_layernorm": false, "language_model.model.layers.43.self_attn.kv_a_proj_with_mqa": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.43.self_attn.o_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.43.self_attn.q_a_layernorm": false, "language_model.model.layers.43.self_attn.q_a_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.43.self_attn.q_b_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.43.self_attn.unembed_out": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.44.attn_hc.base": false, "language_model.model.layers.44.attn_hc.fn": false, "language_model.model.layers.44.attn_hc.scale": false, "language_model.model.layers.44.ffn_hc.base": false, "language_model.model.layers.44.ffn_hc.fn": false, "language_model.model.layers.44.ffn_hc.scale": false, "language_model.model.layers.44.input_layernorm": false, "language_model.model.layers.44.mlp.gate": false, "language_model.model.layers.44.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.44.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.44.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.44.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.44.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.44.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.44.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.44.post_attention_layernorm": false, "language_model.model.layers.44.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.44.self_attn.forget_gate.A_log": false, "language_model.model.layers.44.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.44.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.44.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.44.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.44.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.44.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.44.self_attn.o_norm": false, "language_model.model.layers.44.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.44.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.44.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.5.attn_hc.base": false, "language_model.model.layers.5.attn_hc.fn": false, "language_model.model.layers.5.attn_hc.scale": false, "language_model.model.layers.5.ffn_hc.base": false, "language_model.model.layers.5.ffn_hc.fn": false, "language_model.model.layers.5.ffn_hc.scale": false, "language_model.model.layers.5.input_layernorm": false, "language_model.model.layers.5.mlp.gate": false, "language_model.model.layers.5.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.5.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.5.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.5.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.5.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.5.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.5.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.5.post_attention_layernorm": false, "language_model.model.layers.5.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.5.self_attn.forget_gate.A_log": false, "language_model.model.layers.5.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.5.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.5.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.5.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.5.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.5.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.5.self_attn.o_norm": false, "language_model.model.layers.5.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.5.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.5.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.6.attn_hc.base": false, "language_model.model.layers.6.attn_hc.fn": false, "language_model.model.layers.6.attn_hc.scale": false, "language_model.model.layers.6.ffn_hc.base": false, "language_model.model.layers.6.ffn_hc.fn": false, "language_model.model.layers.6.ffn_hc.scale": false, "language_model.model.layers.6.input_layernorm": false, "language_model.model.layers.6.mlp.gate": false, "language_model.model.layers.6.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.6.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.6.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.6.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.6.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.6.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.6.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.6.post_attention_layernorm": false, "language_model.model.layers.6.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.6.self_attn.forget_gate.A_log": false, "language_model.model.layers.6.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.6.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.6.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.6.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.6.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.6.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.6.self_attn.o_norm": false, "language_model.model.layers.6.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.6.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.6.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.7.attn_hc.base": false, "language_model.model.layers.7.attn_hc.fn": false, "language_model.model.layers.7.attn_hc.scale": false, "language_model.model.layers.7.ffn_hc.base": false, "language_model.model.layers.7.ffn_hc.fn": false, "language_model.model.layers.7.ffn_hc.scale": false, "language_model.model.layers.7.input_layernorm": false, "language_model.model.layers.7.mlp.gate": false, "language_model.model.layers.7.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.7.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.7.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.7.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.7.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.7.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.7.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.7.post_attention_layernorm": false, "language_model.model.layers.7.self_attn.embed_q": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.7.self_attn.indexer.index_kpool_compress_ape": false, "language_model.model.layers.7.self_attn.indexer.index_kpool_compress_gate": false, "language_model.model.layers.7.self_attn.indexer.k_norm": false, "language_model.model.layers.7.self_attn.indexer.k_norm.bias": false, "language_model.model.layers.7.self_attn.indexer.weights_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.7.self_attn.indexer.wk": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.7.self_attn.indexer.wq_b": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.7.self_attn.kv_a_layernorm": false, "language_model.model.layers.7.self_attn.kv_a_proj_with_mqa": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.7.self_attn.o_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.7.self_attn.q_a_layernorm": false, "language_model.model.layers.7.self_attn.q_a_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.7.self_attn.q_b_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.7.self_attn.unembed_out": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.8.attn_hc.base": false, "language_model.model.layers.8.attn_hc.fn": false, "language_model.model.layers.8.attn_hc.scale": false, "language_model.model.layers.8.ffn_hc.base": false, "language_model.model.layers.8.ffn_hc.fn": false, "language_model.model.layers.8.ffn_hc.scale": false, "language_model.model.layers.8.input_layernorm": false, "language_model.model.layers.8.mlp.gate": false, "language_model.model.layers.8.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.8.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.8.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.8.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.8.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.8.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.8.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.8.post_attention_layernorm": false, "language_model.model.layers.8.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.8.self_attn.forget_gate.A_log": false, "language_model.model.layers.8.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.8.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.8.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.8.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.8.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.8.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.8.self_attn.o_norm": false, "language_model.model.layers.8.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.8.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.8.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.9.attn_hc.base": false, "language_model.model.layers.9.attn_hc.fn": false, "language_model.model.layers.9.attn_hc.scale": false, "language_model.model.layers.9.ffn_hc.base": false, "language_model.model.layers.9.ffn_hc.fn": false, "language_model.model.layers.9.ffn_hc.scale": false, "language_model.model.layers.9.input_layernorm": false, "language_model.model.layers.9.mlp.gate": false, "language_model.model.layers.9.mlp.gate.e_score_correction_bias": false, "language_model.model.layers.9.mlp.shared_experts.down_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.9.mlp.shared_experts.gate_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.9.mlp.shared_experts.up_proj": { "group_size": 64, "bits": 6, "mode": "affine" }, "language_model.model.layers.9.mlp.switch_mlp.down_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.9.mlp.switch_mlp.gate_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.9.mlp.switch_mlp.up_proj": { "group_size": 64, "bits": 4, "mode": "affine" }, "language_model.model.layers.9.post_attention_layernorm": false, "language_model.model.layers.9.self_attn.b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.9.self_attn.forget_gate.A_log": false, "language_model.model.layers.9.self_attn.forget_gate.dt_bias": false, "language_model.model.layers.9.self_attn.forget_gate.f_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.9.self_attn.forget_gate.f_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.9.self_attn.g_a_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.9.self_attn.g_b_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.9.self_attn.k_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.9.self_attn.o_norm": false, "language_model.model.layers.9.self_attn.o_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.9.self_attn.q_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.layers.9.self_attn.v_proj": { "group_size": 64, "bits": 8, "mode": "affine" }, "language_model.model.norm": false } }