| { |
| "architectures": [ |
| "DeepseekV4ForCausalLM" |
| ], |
| "attention_bias": false, |
| "attention_dropout": 0.0, |
| "bos_token_id": 0, |
| "compress_ratios": [ |
| 0, |
| 0, |
| 4, |
| 128, |
| 4, |
| 128, |
| 4, |
| 128, |
| 4, |
| 128, |
| 4, |
| 128, |
| 4, |
| 128, |
| 4, |
| 128, |
| 4, |
| 128, |
| 4, |
| 128, |
| 4, |
| 128, |
| 4, |
| 128, |
| 4, |
| 128, |
| 4, |
| 128, |
| 4, |
| 128, |
| 4, |
| 128, |
| 4, |
| 128, |
| 4, |
| 128, |
| 4, |
| 128, |
| 4, |
| 128, |
| 4, |
| 128, |
| 4, |
| 0 |
| ], |
| "compress_rope_theta": 160000.0, |
| "eos_token_id": 1, |
| "hc_eps": 1e-06, |
| "hc_mult": 4, |
| "hc_sinkhorn_iters": 20, |
| "head_dim": 512, |
| "hidden_act": "silu", |
| "hidden_size": 4096, |
| "index_head_dim": 128, |
| "index_n_heads": 64, |
| "index_topk": 512, |
| "initializer_range": 0.02, |
| "max_position_embeddings": 1048576, |
| "mlx-sanitized": "0.30.7", |
| "model_type": "deepseek_v4", |
| "moe_intermediate_size": 2048, |
| "n_routed_experts": 256, |
| "n_shared_experts": 1, |
| "norm_topk_prob": true, |
| "num_attention_heads": 64, |
| "num_experts_per_tok": 6, |
| "num_hash_layers": 3, |
| "num_hidden_layers": 43, |
| "num_key_value_heads": 1, |
| "num_nextn_predict_layers": 1, |
| "o_groups": 8, |
| "o_lora_rank": 1024, |
| "q_lora_rank": 1024, |
| "qk_rope_head_dim": 64, |
| "quantization_config": { |
| "group_size": 32, |
| "model.layers.0.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.0.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.0.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.1.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.1.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.1.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.2.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.2.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.2.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.3.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.3.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.3.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.4.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.4.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.4.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.5.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.5.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.5.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.6.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.6.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.6.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.7.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.7.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.7.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.8.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.8.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.8.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.9.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.9.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.9.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.10.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.10.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.10.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.11.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.11.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.11.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.12.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.12.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.12.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.13.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.13.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.13.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.14.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.14.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.14.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.15.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.15.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.15.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.16.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.16.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.16.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.17.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.17.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.17.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.18.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.18.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.18.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.19.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.19.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.19.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.20.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.20.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.20.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.21.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.21.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.21.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.22.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.22.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.22.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.23.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.23.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.23.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.24.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.24.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.24.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.25.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.25.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.25.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.26.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.26.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.26.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.27.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.27.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.27.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.28.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.28.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.28.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.29.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.29.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.29.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.30.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.30.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.30.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.31.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.31.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.31.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.32.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.32.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.32.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.33.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.33.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.33.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.34.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.34.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.34.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.35.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.35.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.35.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.36.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.36.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.36.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.37.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.37.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.37.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.38.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.38.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.38.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.39.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.39.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.39.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.40.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.40.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.40.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.41.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.41.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.41.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.42.ffn.switch_mlp.gate_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.42.ffn.switch_mlp.up_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.layers.42.ffn.switch_mlp.down_proj": { |
| "mode": "mxfp4" |
| }, |
| "model.embed_tokens": {}, |
| "model.layers.0.attn.wq_a": {}, |
| "model.layers.0.attn.wq_b": {}, |
| "model.layers.0.attn.wkv": {}, |
| "model.layers.0.attn.wo_a": {}, |
| "model.layers.0.attn.wo_b": {}, |
| "model.layers.0.ffn.shared_experts.gate_proj": {}, |
| "model.layers.0.ffn.shared_experts.up_proj": {}, |
| "model.layers.0.ffn.shared_experts.down_proj": {}, |
| "model.layers.1.attn.wq_a": {}, |
| "model.layers.1.attn.wq_b": {}, |
| "model.layers.1.attn.wkv": {}, |
| "model.layers.1.attn.wo_a": {}, |
| "model.layers.1.attn.wo_b": {}, |
| "model.layers.1.ffn.shared_experts.gate_proj": {}, |
| "model.layers.1.ffn.shared_experts.up_proj": {}, |
| "model.layers.1.ffn.shared_experts.down_proj": {}, |
| "model.layers.2.attn.wq_a": {}, |
| "model.layers.2.attn.wq_b": {}, |
| "model.layers.2.attn.wkv": {}, |
| "model.layers.2.attn.wo_a": {}, |
| "model.layers.2.attn.wo_b": {}, |
| "model.layers.2.attn.compressor.wkv": {}, |
| "model.layers.2.attn.compressor.wgate": {}, |
| "model.layers.2.attn.indexer.wq_b": {}, |
| "model.layers.2.attn.indexer.weights_proj": {}, |
| "model.layers.2.attn.indexer.compressor.wkv": {}, |
| "model.layers.2.attn.indexer.compressor.wgate": {}, |
| "model.layers.2.ffn.shared_experts.gate_proj": {}, |
| "model.layers.2.ffn.shared_experts.up_proj": {}, |
| "model.layers.2.ffn.shared_experts.down_proj": {}, |
| "model.layers.3.attn.wq_a": {}, |
| "model.layers.3.attn.wq_b": {}, |
| "model.layers.3.attn.wkv": {}, |
| "model.layers.3.attn.wo_a": {}, |
| "model.layers.3.attn.wo_b": {}, |
| "model.layers.3.attn.compressor.wkv": {}, |
| "model.layers.3.attn.compressor.wgate": {}, |
| "model.layers.3.ffn.shared_experts.gate_proj": {}, |
| "model.layers.3.ffn.shared_experts.up_proj": {}, |
| "model.layers.3.ffn.shared_experts.down_proj": {}, |
| "model.layers.4.attn.wq_a": {}, |
| "model.layers.4.attn.wq_b": {}, |
| "model.layers.4.attn.wkv": {}, |
| "model.layers.4.attn.wo_a": {}, |
| "model.layers.4.attn.wo_b": {}, |
| "model.layers.4.attn.compressor.wkv": {}, |
| "model.layers.4.attn.compressor.wgate": {}, |
| "model.layers.4.attn.indexer.wq_b": {}, |
| "model.layers.4.attn.indexer.weights_proj": {}, |
| "model.layers.4.attn.indexer.compressor.wkv": {}, |
| "model.layers.4.attn.indexer.compressor.wgate": {}, |
| "model.layers.4.ffn.shared_experts.gate_proj": {}, |
| "model.layers.4.ffn.shared_experts.up_proj": {}, |
| "model.layers.4.ffn.shared_experts.down_proj": {}, |
| "model.layers.5.attn.wq_a": {}, |
| "model.layers.5.attn.wq_b": {}, |
| "model.layers.5.attn.wkv": {}, |
| "model.layers.5.attn.wo_a": {}, |
| "model.layers.5.attn.wo_b": {}, |
| "model.layers.5.attn.compressor.wkv": {}, |
| "model.layers.5.attn.compressor.wgate": {}, |
| "model.layers.5.ffn.shared_experts.gate_proj": {}, |
| "model.layers.5.ffn.shared_experts.up_proj": {}, |
| "model.layers.5.ffn.shared_experts.down_proj": {}, |
| "model.layers.6.attn.wq_a": {}, |
| "model.layers.6.attn.wq_b": {}, |
| "model.layers.6.attn.wkv": {}, |
| "model.layers.6.attn.wo_a": {}, |
| "model.layers.6.attn.wo_b": {}, |
| "model.layers.6.attn.compressor.wkv": {}, |
| "model.layers.6.attn.compressor.wgate": {}, |
| "model.layers.6.attn.indexer.wq_b": {}, |
| "model.layers.6.attn.indexer.weights_proj": {}, |
| "model.layers.6.attn.indexer.compressor.wkv": {}, |
| "model.layers.6.attn.indexer.compressor.wgate": {}, |
| "model.layers.6.ffn.shared_experts.gate_proj": {}, |
| "model.layers.6.ffn.shared_experts.up_proj": {}, |
| "model.layers.6.ffn.shared_experts.down_proj": {}, |
| "model.layers.7.attn.wq_a": {}, |
| "model.layers.7.attn.wq_b": {}, |
| "model.layers.7.attn.wkv": {}, |
| "model.layers.7.attn.wo_a": {}, |
| "model.layers.7.attn.wo_b": {}, |
| "model.layers.7.attn.compressor.wkv": {}, |
| "model.layers.7.attn.compressor.wgate": {}, |
| "model.layers.7.ffn.shared_experts.gate_proj": {}, |
| "model.layers.7.ffn.shared_experts.up_proj": {}, |
| "model.layers.7.ffn.shared_experts.down_proj": {}, |
| "model.layers.8.attn.wq_a": {}, |
| "model.layers.8.attn.wq_b": {}, |
| "model.layers.8.attn.wkv": {}, |
| "model.layers.8.attn.wo_a": {}, |
| "model.layers.8.attn.wo_b": {}, |
| "model.layers.8.attn.compressor.wkv": {}, |
| "model.layers.8.attn.compressor.wgate": {}, |
| "model.layers.8.attn.indexer.wq_b": {}, |
| "model.layers.8.attn.indexer.weights_proj": {}, |
| "model.layers.8.attn.indexer.compressor.wkv": {}, |
| "model.layers.8.attn.indexer.compressor.wgate": {}, |
| "model.layers.8.ffn.shared_experts.gate_proj": {}, |
| "model.layers.8.ffn.shared_experts.up_proj": {}, |
| "model.layers.8.ffn.shared_experts.down_proj": {}, |
| "model.layers.9.attn.wq_a": {}, |
| "model.layers.9.attn.wq_b": {}, |
| "model.layers.9.attn.wkv": {}, |
| "model.layers.9.attn.wo_a": {}, |
| "model.layers.9.attn.wo_b": {}, |
| "model.layers.9.attn.compressor.wkv": {}, |
| "model.layers.9.attn.compressor.wgate": {}, |
| "model.layers.9.ffn.shared_experts.gate_proj": {}, |
| "model.layers.9.ffn.shared_experts.up_proj": {}, |
| "model.layers.9.ffn.shared_experts.down_proj": {}, |
| "model.layers.10.attn.wq_a": {}, |
| "model.layers.10.attn.wq_b": {}, |
| "model.layers.10.attn.wkv": {}, |
| "model.layers.10.attn.wo_a": {}, |
| "model.layers.10.attn.wo_b": {}, |
| "model.layers.10.attn.compressor.wkv": {}, |
| "model.layers.10.attn.compressor.wgate": {}, |
| "model.layers.10.attn.indexer.wq_b": {}, |
| "model.layers.10.attn.indexer.weights_proj": {}, |
| "model.layers.10.attn.indexer.compressor.wkv": {}, |
| "model.layers.10.attn.indexer.compressor.wgate": {}, |
| "model.layers.10.ffn.shared_experts.gate_proj": {}, |
| "model.layers.10.ffn.shared_experts.up_proj": {}, |
| "model.layers.10.ffn.shared_experts.down_proj": {}, |
| "model.layers.11.attn.wq_a": {}, |
| "model.layers.11.attn.wq_b": {}, |
| "model.layers.11.attn.wkv": {}, |
| "model.layers.11.attn.wo_a": {}, |
| "model.layers.11.attn.wo_b": {}, |
| "model.layers.11.attn.compressor.wkv": {}, |
| "model.layers.11.attn.compressor.wgate": {}, |
| "model.layers.11.ffn.shared_experts.gate_proj": {}, |
| "model.layers.11.ffn.shared_experts.up_proj": {}, |
| "model.layers.11.ffn.shared_experts.down_proj": {}, |
| "model.layers.12.attn.wq_a": {}, |
| "model.layers.12.attn.wq_b": {}, |
| "model.layers.12.attn.wkv": {}, |
| "model.layers.12.attn.wo_a": {}, |
| "model.layers.12.attn.wo_b": {}, |
| "model.layers.12.attn.compressor.wkv": {}, |
| "model.layers.12.attn.compressor.wgate": {}, |
| "model.layers.12.attn.indexer.wq_b": {}, |
| "model.layers.12.attn.indexer.weights_proj": {}, |
| "model.layers.12.attn.indexer.compressor.wkv": {}, |
| "model.layers.12.attn.indexer.compressor.wgate": {}, |
| "model.layers.12.ffn.shared_experts.gate_proj": {}, |
| "model.layers.12.ffn.shared_experts.up_proj": {}, |
| "model.layers.12.ffn.shared_experts.down_proj": {}, |
| "model.layers.13.attn.wq_a": {}, |
| "model.layers.13.attn.wq_b": {}, |
| "model.layers.13.attn.wkv": {}, |
| "model.layers.13.attn.wo_a": {}, |
| "model.layers.13.attn.wo_b": {}, |
| "model.layers.13.attn.compressor.wkv": {}, |
| "model.layers.13.attn.compressor.wgate": {}, |
| "model.layers.13.ffn.shared_experts.gate_proj": {}, |
| "model.layers.13.ffn.shared_experts.up_proj": {}, |
| "model.layers.13.ffn.shared_experts.down_proj": {}, |
| "model.layers.14.attn.wq_a": {}, |
| "model.layers.14.attn.wq_b": {}, |
| "model.layers.14.attn.wkv": {}, |
| "model.layers.14.attn.wo_a": {}, |
| "model.layers.14.attn.wo_b": {}, |
| "model.layers.14.attn.compressor.wkv": {}, |
| "model.layers.14.attn.compressor.wgate": {}, |
| "model.layers.14.attn.indexer.wq_b": {}, |
| "model.layers.14.attn.indexer.weights_proj": {}, |
| "model.layers.14.attn.indexer.compressor.wkv": {}, |
| "model.layers.14.attn.indexer.compressor.wgate": {}, |
| "model.layers.14.ffn.shared_experts.gate_proj": {}, |
| "model.layers.14.ffn.shared_experts.up_proj": {}, |
| "model.layers.14.ffn.shared_experts.down_proj": {}, |
| "model.layers.15.attn.wq_a": {}, |
| "model.layers.15.attn.wq_b": {}, |
| "model.layers.15.attn.wkv": {}, |
| "model.layers.15.attn.wo_a": {}, |
| "model.layers.15.attn.wo_b": {}, |
| "model.layers.15.attn.compressor.wkv": {}, |
| "model.layers.15.attn.compressor.wgate": {}, |
| "model.layers.15.ffn.shared_experts.gate_proj": {}, |
| "model.layers.15.ffn.shared_experts.up_proj": {}, |
| "model.layers.15.ffn.shared_experts.down_proj": {}, |
| "model.layers.16.attn.wq_a": {}, |
| "model.layers.16.attn.wq_b": {}, |
| "model.layers.16.attn.wkv": {}, |
| "model.layers.16.attn.wo_a": {}, |
| "model.layers.16.attn.wo_b": {}, |
| "model.layers.16.attn.compressor.wkv": {}, |
| "model.layers.16.attn.compressor.wgate": {}, |
| "model.layers.16.attn.indexer.wq_b": {}, |
| "model.layers.16.attn.indexer.weights_proj": {}, |
| "model.layers.16.attn.indexer.compressor.wkv": {}, |
| "model.layers.16.attn.indexer.compressor.wgate": {}, |
| "model.layers.16.ffn.shared_experts.gate_proj": {}, |
| "model.layers.16.ffn.shared_experts.up_proj": {}, |
| "model.layers.16.ffn.shared_experts.down_proj": {}, |
| "model.layers.17.attn.wq_a": {}, |
| "model.layers.17.attn.wq_b": {}, |
| "model.layers.17.attn.wkv": {}, |
| "model.layers.17.attn.wo_a": {}, |
| "model.layers.17.attn.wo_b": {}, |
| "model.layers.17.attn.compressor.wkv": {}, |
| "model.layers.17.attn.compressor.wgate": {}, |
| "model.layers.17.ffn.shared_experts.gate_proj": {}, |
| "model.layers.17.ffn.shared_experts.up_proj": {}, |
| "model.layers.17.ffn.shared_experts.down_proj": {}, |
| "model.layers.18.attn.wq_a": {}, |
| "model.layers.18.attn.wq_b": {}, |
| "model.layers.18.attn.wkv": {}, |
| "model.layers.18.attn.wo_a": {}, |
| "model.layers.18.attn.wo_b": {}, |
| "model.layers.18.attn.compressor.wkv": {}, |
| "model.layers.18.attn.compressor.wgate": {}, |
| "model.layers.18.attn.indexer.wq_b": {}, |
| "model.layers.18.attn.indexer.weights_proj": {}, |
| "model.layers.18.attn.indexer.compressor.wkv": {}, |
| "model.layers.18.attn.indexer.compressor.wgate": {}, |
| "model.layers.18.ffn.shared_experts.gate_proj": {}, |
| "model.layers.18.ffn.shared_experts.up_proj": {}, |
| "model.layers.18.ffn.shared_experts.down_proj": {}, |
| "model.layers.19.attn.wq_a": {}, |
| "model.layers.19.attn.wq_b": {}, |
| "model.layers.19.attn.wkv": {}, |
| "model.layers.19.attn.wo_a": {}, |
| "model.layers.19.attn.wo_b": {}, |
| "model.layers.19.attn.compressor.wkv": {}, |
| "model.layers.19.attn.compressor.wgate": {}, |
| "model.layers.19.ffn.shared_experts.gate_proj": {}, |
| "model.layers.19.ffn.shared_experts.up_proj": {}, |
| "model.layers.19.ffn.shared_experts.down_proj": {}, |
| "model.layers.20.attn.wq_a": {}, |
| "model.layers.20.attn.wq_b": {}, |
| "model.layers.20.attn.wkv": {}, |
| "model.layers.20.attn.wo_a": {}, |
| "model.layers.20.attn.wo_b": {}, |
| "model.layers.20.attn.compressor.wkv": {}, |
| "model.layers.20.attn.compressor.wgate": {}, |
| "model.layers.20.attn.indexer.wq_b": {}, |
| "model.layers.20.attn.indexer.weights_proj": {}, |
| "model.layers.20.attn.indexer.compressor.wkv": {}, |
| "model.layers.20.attn.indexer.compressor.wgate": {}, |
| "model.layers.20.ffn.shared_experts.gate_proj": {}, |
| "model.layers.20.ffn.shared_experts.up_proj": {}, |
| "model.layers.20.ffn.shared_experts.down_proj": {}, |
| "model.layers.21.attn.wq_a": {}, |
| "model.layers.21.attn.wq_b": {}, |
| "model.layers.21.attn.wkv": {}, |
| "model.layers.21.attn.wo_a": {}, |
| "model.layers.21.attn.wo_b": {}, |
| "model.layers.21.attn.compressor.wkv": {}, |
| "model.layers.21.attn.compressor.wgate": {}, |
| "model.layers.21.ffn.shared_experts.gate_proj": {}, |
| "model.layers.21.ffn.shared_experts.up_proj": {}, |
| "model.layers.21.ffn.shared_experts.down_proj": {}, |
| "model.layers.22.attn.wq_a": {}, |
| "model.layers.22.attn.wq_b": {}, |
| "model.layers.22.attn.wkv": {}, |
| "model.layers.22.attn.wo_a": {}, |
| "model.layers.22.attn.wo_b": {}, |
| "model.layers.22.attn.compressor.wkv": {}, |
| "model.layers.22.attn.compressor.wgate": {}, |
| "model.layers.22.attn.indexer.wq_b": {}, |
| "model.layers.22.attn.indexer.weights_proj": {}, |
| "model.layers.22.attn.indexer.compressor.wkv": {}, |
| "model.layers.22.attn.indexer.compressor.wgate": {}, |
| "model.layers.22.ffn.shared_experts.gate_proj": {}, |
| "model.layers.22.ffn.shared_experts.up_proj": {}, |
| "model.layers.22.ffn.shared_experts.down_proj": {}, |
| "model.layers.23.attn.wq_a": {}, |
| "model.layers.23.attn.wq_b": {}, |
| "model.layers.23.attn.wkv": {}, |
| "model.layers.23.attn.wo_a": {}, |
| "model.layers.23.attn.wo_b": {}, |
| "model.layers.23.attn.compressor.wkv": {}, |
| "model.layers.23.attn.compressor.wgate": {}, |
| "model.layers.23.ffn.shared_experts.gate_proj": {}, |
| "model.layers.23.ffn.shared_experts.up_proj": {}, |
| "model.layers.23.ffn.shared_experts.down_proj": {}, |
| "model.layers.24.attn.wq_a": {}, |
| "model.layers.24.attn.wq_b": {}, |
| "model.layers.24.attn.wkv": {}, |
| "model.layers.24.attn.wo_a": {}, |
| "model.layers.24.attn.wo_b": {}, |
| "model.layers.24.attn.compressor.wkv": {}, |
| "model.layers.24.attn.compressor.wgate": {}, |
| "model.layers.24.attn.indexer.wq_b": {}, |
| "model.layers.24.attn.indexer.weights_proj": {}, |
| "model.layers.24.attn.indexer.compressor.wkv": {}, |
| "model.layers.24.attn.indexer.compressor.wgate": {}, |
| "model.layers.24.ffn.shared_experts.gate_proj": {}, |
| "model.layers.24.ffn.shared_experts.up_proj": {}, |
| "model.layers.24.ffn.shared_experts.down_proj": {}, |
| "model.layers.25.attn.wq_a": {}, |
| "model.layers.25.attn.wq_b": {}, |
| "model.layers.25.attn.wkv": {}, |
| "model.layers.25.attn.wo_a": {}, |
| "model.layers.25.attn.wo_b": {}, |
| "model.layers.25.attn.compressor.wkv": {}, |
| "model.layers.25.attn.compressor.wgate": {}, |
| "model.layers.25.ffn.shared_experts.gate_proj": {}, |
| "model.layers.25.ffn.shared_experts.up_proj": {}, |
| "model.layers.25.ffn.shared_experts.down_proj": {}, |
| "model.layers.26.attn.wq_a": {}, |
| "model.layers.26.attn.wq_b": {}, |
| "model.layers.26.attn.wkv": {}, |
| "model.layers.26.attn.wo_a": {}, |
| "model.layers.26.attn.wo_b": {}, |
| "model.layers.26.attn.compressor.wkv": {}, |
| "model.layers.26.attn.compressor.wgate": {}, |
| "model.layers.26.attn.indexer.wq_b": {}, |
| "model.layers.26.attn.indexer.weights_proj": {}, |
| "model.layers.26.attn.indexer.compressor.wkv": {}, |
| "model.layers.26.attn.indexer.compressor.wgate": {}, |
| "model.layers.26.ffn.shared_experts.gate_proj": {}, |
| "model.layers.26.ffn.shared_experts.up_proj": {}, |
| "model.layers.26.ffn.shared_experts.down_proj": {}, |
| "model.layers.27.attn.wq_a": {}, |
| "model.layers.27.attn.wq_b": {}, |
| "model.layers.27.attn.wkv": {}, |
| "model.layers.27.attn.wo_a": {}, |
| "model.layers.27.attn.wo_b": {}, |
| "model.layers.27.attn.compressor.wkv": {}, |
| "model.layers.27.attn.compressor.wgate": {}, |
| "model.layers.27.ffn.shared_experts.gate_proj": {}, |
| "model.layers.27.ffn.shared_experts.up_proj": {}, |
| "model.layers.27.ffn.shared_experts.down_proj": {}, |
| "model.layers.28.attn.wq_a": {}, |
| "model.layers.28.attn.wq_b": {}, |
| "model.layers.28.attn.wkv": {}, |
| "model.layers.28.attn.wo_a": {}, |
| "model.layers.28.attn.wo_b": {}, |
| "model.layers.28.attn.compressor.wkv": {}, |
| "model.layers.28.attn.compressor.wgate": {}, |
| "model.layers.28.attn.indexer.wq_b": {}, |
| "model.layers.28.attn.indexer.weights_proj": {}, |
| "model.layers.28.attn.indexer.compressor.wkv": {}, |
| "model.layers.28.attn.indexer.compressor.wgate": {}, |
| "model.layers.28.ffn.shared_experts.gate_proj": {}, |
| "model.layers.28.ffn.shared_experts.up_proj": {}, |
| "model.layers.28.ffn.shared_experts.down_proj": {}, |
| "model.layers.29.attn.wq_a": {}, |
| "model.layers.29.attn.wq_b": {}, |
| "model.layers.29.attn.wkv": {}, |
| "model.layers.29.attn.wo_a": {}, |
| "model.layers.29.attn.wo_b": {}, |
| "model.layers.29.attn.compressor.wkv": {}, |
| "model.layers.29.attn.compressor.wgate": {}, |
| "model.layers.29.ffn.shared_experts.gate_proj": {}, |
| "model.layers.29.ffn.shared_experts.up_proj": {}, |
| "model.layers.29.ffn.shared_experts.down_proj": {}, |
| "model.layers.30.attn.wq_a": {}, |
| "model.layers.30.attn.wq_b": {}, |
| "model.layers.30.attn.wkv": {}, |
| "model.layers.30.attn.wo_a": {}, |
| "model.layers.30.attn.wo_b": {}, |
| "model.layers.30.attn.compressor.wkv": {}, |
| "model.layers.30.attn.compressor.wgate": {}, |
| "model.layers.30.attn.indexer.wq_b": {}, |
| "model.layers.30.attn.indexer.weights_proj": {}, |
| "model.layers.30.attn.indexer.compressor.wkv": {}, |
| "model.layers.30.attn.indexer.compressor.wgate": {}, |
| "model.layers.30.ffn.shared_experts.gate_proj": {}, |
| "model.layers.30.ffn.shared_experts.up_proj": {}, |
| "model.layers.30.ffn.shared_experts.down_proj": {}, |
| "model.layers.31.attn.wq_a": {}, |
| "model.layers.31.attn.wq_b": {}, |
| "model.layers.31.attn.wkv": {}, |
| "model.layers.31.attn.wo_a": {}, |
| "model.layers.31.attn.wo_b": {}, |
| "model.layers.31.attn.compressor.wkv": {}, |
| "model.layers.31.attn.compressor.wgate": {}, |
| "model.layers.31.ffn.shared_experts.gate_proj": {}, |
| "model.layers.31.ffn.shared_experts.up_proj": {}, |
| "model.layers.31.ffn.shared_experts.down_proj": {}, |
| "model.layers.32.attn.wq_a": {}, |
| "model.layers.32.attn.wq_b": {}, |
| "model.layers.32.attn.wkv": {}, |
| "model.layers.32.attn.wo_a": {}, |
| "model.layers.32.attn.wo_b": {}, |
| "model.layers.32.attn.compressor.wkv": {}, |
| "model.layers.32.attn.compressor.wgate": {}, |
| "model.layers.32.attn.indexer.wq_b": {}, |
| "model.layers.32.attn.indexer.weights_proj": {}, |
| "model.layers.32.attn.indexer.compressor.wkv": {}, |
| "model.layers.32.attn.indexer.compressor.wgate": {}, |
| "model.layers.32.ffn.shared_experts.gate_proj": {}, |
| "model.layers.32.ffn.shared_experts.up_proj": {}, |
| "model.layers.32.ffn.shared_experts.down_proj": {}, |
| "model.layers.33.attn.wq_a": {}, |
| "model.layers.33.attn.wq_b": {}, |
| "model.layers.33.attn.wkv": {}, |
| "model.layers.33.attn.wo_a": {}, |
| "model.layers.33.attn.wo_b": {}, |
| "model.layers.33.attn.compressor.wkv": {}, |
| "model.layers.33.attn.compressor.wgate": {}, |
| "model.layers.33.ffn.shared_experts.gate_proj": {}, |
| "model.layers.33.ffn.shared_experts.up_proj": {}, |
| "model.layers.33.ffn.shared_experts.down_proj": {}, |
| "model.layers.34.attn.wq_a": {}, |
| "model.layers.34.attn.wq_b": {}, |
| "model.layers.34.attn.wkv": {}, |
| "model.layers.34.attn.wo_a": {}, |
| "model.layers.34.attn.wo_b": {}, |
| "model.layers.34.attn.compressor.wkv": {}, |
| "model.layers.34.attn.compressor.wgate": {}, |
| "model.layers.34.attn.indexer.wq_b": {}, |
| "model.layers.34.attn.indexer.weights_proj": {}, |
| "model.layers.34.attn.indexer.compressor.wkv": {}, |
| "model.layers.34.attn.indexer.compressor.wgate": {}, |
| "model.layers.34.ffn.shared_experts.gate_proj": {}, |
| "model.layers.34.ffn.shared_experts.up_proj": {}, |
| "model.layers.34.ffn.shared_experts.down_proj": {}, |
| "model.layers.35.attn.wq_a": {}, |
| "model.layers.35.attn.wq_b": {}, |
| "model.layers.35.attn.wkv": {}, |
| "model.layers.35.attn.wo_a": {}, |
| "model.layers.35.attn.wo_b": {}, |
| "model.layers.35.attn.compressor.wkv": {}, |
| "model.layers.35.attn.compressor.wgate": {}, |
| "model.layers.35.ffn.shared_experts.gate_proj": {}, |
| "model.layers.35.ffn.shared_experts.up_proj": {}, |
| "model.layers.35.ffn.shared_experts.down_proj": {}, |
| "model.layers.36.attn.wq_a": {}, |
| "model.layers.36.attn.wq_b": {}, |
| "model.layers.36.attn.wkv": {}, |
| "model.layers.36.attn.wo_a": {}, |
| "model.layers.36.attn.wo_b": {}, |
| "model.layers.36.attn.compressor.wkv": {}, |
| "model.layers.36.attn.compressor.wgate": {}, |
| "model.layers.36.attn.indexer.wq_b": {}, |
| "model.layers.36.attn.indexer.weights_proj": {}, |
| "model.layers.36.attn.indexer.compressor.wkv": {}, |
| "model.layers.36.attn.indexer.compressor.wgate": {}, |
| "model.layers.36.ffn.shared_experts.gate_proj": {}, |
| "model.layers.36.ffn.shared_experts.up_proj": {}, |
| "model.layers.36.ffn.shared_experts.down_proj": {}, |
| "model.layers.37.attn.wq_a": {}, |
| "model.layers.37.attn.wq_b": {}, |
| "model.layers.37.attn.wkv": {}, |
| "model.layers.37.attn.wo_a": {}, |
| "model.layers.37.attn.wo_b": {}, |
| "model.layers.37.attn.compressor.wkv": {}, |
| "model.layers.37.attn.compressor.wgate": {}, |
| "model.layers.37.ffn.shared_experts.gate_proj": {}, |
| "model.layers.37.ffn.shared_experts.up_proj": {}, |
| "model.layers.37.ffn.shared_experts.down_proj": {}, |
| "model.layers.38.attn.wq_a": {}, |
| "model.layers.38.attn.wq_b": {}, |
| "model.layers.38.attn.wkv": {}, |
| "model.layers.38.attn.wo_a": {}, |
| "model.layers.38.attn.wo_b": {}, |
| "model.layers.38.attn.compressor.wkv": {}, |
| "model.layers.38.attn.compressor.wgate": {}, |
| "model.layers.38.attn.indexer.wq_b": {}, |
| "model.layers.38.attn.indexer.weights_proj": {}, |
| "model.layers.38.attn.indexer.compressor.wkv": {}, |
| "model.layers.38.attn.indexer.compressor.wgate": {}, |
| "model.layers.38.ffn.shared_experts.gate_proj": {}, |
| "model.layers.38.ffn.shared_experts.up_proj": {}, |
| "model.layers.38.ffn.shared_experts.down_proj": {}, |
| "model.layers.39.attn.wq_a": {}, |
| "model.layers.39.attn.wq_b": {}, |
| "model.layers.39.attn.wkv": {}, |
| "model.layers.39.attn.wo_a": {}, |
| "model.layers.39.attn.wo_b": {}, |
| "model.layers.39.attn.compressor.wkv": {}, |
| "model.layers.39.attn.compressor.wgate": {}, |
| "model.layers.39.ffn.shared_experts.gate_proj": {}, |
| "model.layers.39.ffn.shared_experts.up_proj": {}, |
| "model.layers.39.ffn.shared_experts.down_proj": {}, |
| "model.layers.40.attn.wq_a": {}, |
| "model.layers.40.attn.wq_b": {}, |
| "model.layers.40.attn.wkv": {}, |
| "model.layers.40.attn.wo_a": {}, |
| "model.layers.40.attn.wo_b": {}, |
| "model.layers.40.attn.compressor.wkv": {}, |
| "model.layers.40.attn.compressor.wgate": {}, |
| "model.layers.40.attn.indexer.wq_b": {}, |
| "model.layers.40.attn.indexer.weights_proj": {}, |
| "model.layers.40.attn.indexer.compressor.wkv": {}, |
| "model.layers.40.attn.indexer.compressor.wgate": {}, |
| "model.layers.40.ffn.shared_experts.gate_proj": {}, |
| "model.layers.40.ffn.shared_experts.up_proj": {}, |
| "model.layers.40.ffn.shared_experts.down_proj": {}, |
| "model.layers.41.attn.wq_a": {}, |
| "model.layers.41.attn.wq_b": {}, |
| "model.layers.41.attn.wkv": {}, |
| "model.layers.41.attn.wo_a": {}, |
| "model.layers.41.attn.wo_b": {}, |
| "model.layers.41.attn.compressor.wkv": {}, |
| "model.layers.41.attn.compressor.wgate": {}, |
| "model.layers.41.ffn.shared_experts.gate_proj": {}, |
| "model.layers.41.ffn.shared_experts.up_proj": {}, |
| "model.layers.41.ffn.shared_experts.down_proj": {}, |
| "model.layers.42.attn.wq_a": {}, |
| "model.layers.42.attn.wq_b": {}, |
| "model.layers.42.attn.wkv": {}, |
| "model.layers.42.attn.wo_a": {}, |
| "model.layers.42.attn.wo_b": {}, |
| "model.layers.42.attn.compressor.wkv": {}, |
| "model.layers.42.attn.compressor.wgate": {}, |
| "model.layers.42.attn.indexer.wq_b": {}, |
| "model.layers.42.attn.indexer.weights_proj": {}, |
| "model.layers.42.attn.indexer.compressor.wkv": {}, |
| "model.layers.42.attn.indexer.compressor.wgate": {}, |
| "model.layers.42.ffn.shared_experts.gate_proj": {}, |
| "model.layers.42.ffn.shared_experts.up_proj": {}, |
| "model.layers.42.ffn.shared_experts.down_proj": {}, |
| "lm_head": {} |
| }, |
| "rms_norm_eps": 1e-06, |
| "rope_scaling": { |
| "beta_fast": 32, |
| "beta_slow": 1, |
| "factor": 16, |
| "original_max_position_embeddings": 65536, |
| "type": "yarn" |
| }, |
| "rope_theta": 10000.0, |
| "routed_scaling_factor": 1.5, |
| "scoring_func": "sqrtsoftplus", |
| "sliding_window": 128, |
| "swiglu_limit": 10.0, |
| "tie_word_embeddings": false, |
| "topk_method": "noaux_tc", |
| "torch_dtype": "bfloat16", |
| "transformers_version": "4.57.1", |
| "use_cache": true, |
| "vocab_size": 129280 |
| } |
