N8Programs's picture
Ship lossless text-only MLX-LM package
6994986 verified
Raw
History Blame Contribute Delete
90.3 kB
{
"architectures": [
"Qwen3_5MoeForConditionalGeneration"
],
"dtype": "bfloat16",
"mlx_hybrid_format": {
"activations": "model_dtype_weight_only_quantized_matmul",
"format": "modelopt_fp8_nvfp4_v1",
"fp8_storage": "mxfp8_carrier_with_unit_e8m0_scales",
"nvfp4_storage": "native_e2m1_e4m3_with_output_tensor_scale",
"source_activation_scales_retained": false,
"source_quantization_config": {
"config_groups": {
"group_0": {
"input_activations": {
"dynamic": false,
"num_bits": 8,
"type": "float"
},
"targets": [
"model.language_model.layers.0.linear_attn.in_proj_qkv",
"model.language_model.layers.0.linear_attn.in_proj_z",
"model.language_model.layers.0.linear_attn.out_proj",
"model.language_model.layers.1.linear_attn.in_proj_qkv",
"model.language_model.layers.1.linear_attn.in_proj_z",
"model.language_model.layers.1.linear_attn.out_proj",
"model.language_model.layers.10.linear_attn.in_proj_qkv",
"model.language_model.layers.10.linear_attn.in_proj_z",
"model.language_model.layers.10.linear_attn.out_proj",
"model.language_model.layers.11.self_attn.k_proj",
"model.language_model.layers.11.self_attn.o_proj",
"model.language_model.layers.11.self_attn.q_proj",
"model.language_model.layers.11.self_attn.v_proj",
"model.language_model.layers.12.linear_attn.in_proj_qkv",
"model.language_model.layers.12.linear_attn.in_proj_z",
"model.language_model.layers.12.linear_attn.out_proj",
"model.language_model.layers.13.linear_attn.in_proj_qkv",
"model.language_model.layers.13.linear_attn.in_proj_z",
"model.language_model.layers.13.linear_attn.out_proj",
"model.language_model.layers.14.linear_attn.in_proj_qkv",
"model.language_model.layers.14.linear_attn.in_proj_z",
"model.language_model.layers.14.linear_attn.out_proj",
"model.language_model.layers.15.self_attn.k_proj",
"model.language_model.layers.15.self_attn.o_proj",
"model.language_model.layers.15.self_attn.q_proj",
"model.language_model.layers.15.self_attn.v_proj",
"model.language_model.layers.16.linear_attn.in_proj_qkv",
"model.language_model.layers.16.linear_attn.in_proj_z",
"model.language_model.layers.16.linear_attn.out_proj",
"model.language_model.layers.17.linear_attn.in_proj_qkv",
"model.language_model.layers.17.linear_attn.in_proj_z",
"model.language_model.layers.17.linear_attn.out_proj",
"model.language_model.layers.18.linear_attn.in_proj_qkv",
"model.language_model.layers.18.linear_attn.in_proj_z",
"model.language_model.layers.18.linear_attn.out_proj",
"model.language_model.layers.19.self_attn.k_proj",
"model.language_model.layers.19.self_attn.o_proj",
"model.language_model.layers.19.self_attn.q_proj",
"model.language_model.layers.19.self_attn.v_proj",
"model.language_model.layers.2.linear_attn.in_proj_qkv",
"model.language_model.layers.2.linear_attn.in_proj_z",
"model.language_model.layers.2.linear_attn.out_proj",
"model.language_model.layers.20.linear_attn.in_proj_qkv",
"model.language_model.layers.20.linear_attn.in_proj_z",
"model.language_model.layers.20.linear_attn.out_proj",
"model.language_model.layers.21.linear_attn.in_proj_qkv",
"model.language_model.layers.21.linear_attn.in_proj_z",
"model.language_model.layers.21.linear_attn.out_proj",
"model.language_model.layers.22.linear_attn.in_proj_qkv",
"model.language_model.layers.22.linear_attn.in_proj_z",
"model.language_model.layers.22.linear_attn.out_proj",
"model.language_model.layers.23.self_attn.k_proj",
"model.language_model.layers.23.self_attn.o_proj",
"model.language_model.layers.23.self_attn.q_proj",
"model.language_model.layers.23.self_attn.v_proj",
"model.language_model.layers.24.linear_attn.in_proj_qkv",
"model.language_model.layers.24.linear_attn.in_proj_z",
"model.language_model.layers.24.linear_attn.out_proj",
"model.language_model.layers.25.linear_attn.in_proj_qkv",
"model.language_model.layers.25.linear_attn.in_proj_z",
"model.language_model.layers.25.linear_attn.out_proj",
"model.language_model.layers.26.linear_attn.in_proj_qkv",
"model.language_model.layers.26.linear_attn.in_proj_z",
"model.language_model.layers.26.linear_attn.out_proj",
"model.language_model.layers.27.self_attn.k_proj",
"model.language_model.layers.27.self_attn.o_proj",
"model.language_model.layers.27.self_attn.q_proj",
"model.language_model.layers.27.self_attn.v_proj",
"model.language_model.layers.28.linear_attn.in_proj_qkv",
"model.language_model.layers.28.linear_attn.in_proj_z",
"model.language_model.layers.28.linear_attn.out_proj",
"model.language_model.layers.29.linear_attn.in_proj_qkv",
"model.language_model.layers.29.linear_attn.in_proj_z",
"model.language_model.layers.29.linear_attn.out_proj",
"model.language_model.layers.3.self_attn.k_proj",
"model.language_model.layers.3.self_attn.o_proj",
"model.language_model.layers.3.self_attn.q_proj",
"model.language_model.layers.3.self_attn.v_proj",
"model.language_model.layers.30.linear_attn.in_proj_qkv",
"model.language_model.layers.30.linear_attn.in_proj_z",
"model.language_model.layers.30.linear_attn.out_proj",
"model.language_model.layers.31.self_attn.k_proj",
"model.language_model.layers.31.self_attn.o_proj",
"model.language_model.layers.31.self_attn.q_proj",
"model.language_model.layers.31.self_attn.v_proj",
"model.language_model.layers.32.linear_attn.in_proj_qkv",
"model.language_model.layers.32.linear_attn.in_proj_z",
"model.language_model.layers.32.linear_attn.out_proj",
"model.language_model.layers.33.linear_attn.in_proj_qkv",
"model.language_model.layers.33.linear_attn.in_proj_z",
"model.language_model.layers.33.linear_attn.out_proj",
"model.language_model.layers.34.linear_attn.in_proj_qkv",
"model.language_model.layers.34.linear_attn.in_proj_z",
"model.language_model.layers.34.linear_attn.out_proj",
"model.language_model.layers.35.self_attn.k_proj",
"model.language_model.layers.35.self_attn.o_proj",
"model.language_model.layers.35.self_attn.q_proj",
"model.language_model.layers.35.self_attn.v_proj",
"model.language_model.layers.36.linear_attn.in_proj_qkv",
"model.language_model.layers.36.linear_attn.in_proj_z",
"model.language_model.layers.36.linear_attn.out_proj",
"model.language_model.layers.37.linear_attn.in_proj_qkv",
"model.language_model.layers.37.linear_attn.in_proj_z",
"model.language_model.layers.37.linear_attn.out_proj",
"model.language_model.layers.38.linear_attn.in_proj_qkv",
"model.language_model.layers.38.linear_attn.in_proj_z",
"model.language_model.layers.38.linear_attn.out_proj",
"model.language_model.layers.39.self_attn.k_proj",
"model.language_model.layers.39.self_attn.o_proj",
"model.language_model.layers.39.self_attn.q_proj",
"model.language_model.layers.39.self_attn.v_proj",
"model.language_model.layers.4.linear_attn.in_proj_qkv",
"model.language_model.layers.4.linear_attn.in_proj_z",
"model.language_model.layers.4.linear_attn.out_proj",
"model.language_model.layers.5.linear_attn.in_proj_qkv",
"model.language_model.layers.5.linear_attn.in_proj_z",
"model.language_model.layers.5.linear_attn.out_proj",
"model.language_model.layers.6.linear_attn.in_proj_qkv",
"model.language_model.layers.6.linear_attn.in_proj_z",
"model.language_model.layers.6.linear_attn.out_proj",
"model.language_model.layers.7.self_attn.k_proj",
"model.language_model.layers.7.self_attn.o_proj",
"model.language_model.layers.7.self_attn.q_proj",
"model.language_model.layers.7.self_attn.v_proj",
"model.language_model.layers.8.linear_attn.in_proj_qkv",
"model.language_model.layers.8.linear_attn.in_proj_z",
"model.language_model.layers.8.linear_attn.out_proj",
"model.language_model.layers.9.linear_attn.in_proj_qkv",
"model.language_model.layers.9.linear_attn.in_proj_z",
"model.language_model.layers.9.linear_attn.out_proj"
],
"weights": {
"dynamic": false,
"num_bits": 8,
"type": "float"
}
},
"group_1": {
"input_activations": {
"dynamic": false,
"group_size": 16,
"num_bits": 4,
"type": "float"
},
"targets": [
"lm_head",
"model.language_model.layers.0.mlp.experts",
"model.language_model.layers.0.mlp.shared_expert.down_proj",
"model.language_model.layers.0.mlp.shared_expert.gate_proj",
"model.language_model.layers.0.mlp.shared_expert.up_proj",
"model.language_model.layers.1.mlp.experts",
"model.language_model.layers.1.mlp.shared_expert.down_proj",
"model.language_model.layers.1.mlp.shared_expert.gate_proj",
"model.language_model.layers.1.mlp.shared_expert.up_proj",
"model.language_model.layers.10.mlp.experts",
"model.language_model.layers.10.mlp.shared_expert.down_proj",
"model.language_model.layers.10.mlp.shared_expert.gate_proj",
"model.language_model.layers.10.mlp.shared_expert.up_proj",
"model.language_model.layers.11.mlp.experts",
"model.language_model.layers.11.mlp.shared_expert.down_proj",
"model.language_model.layers.11.mlp.shared_expert.gate_proj",
"model.language_model.layers.11.mlp.shared_expert.up_proj",
"model.language_model.layers.12.mlp.experts",
"model.language_model.layers.12.mlp.shared_expert.down_proj",
"model.language_model.layers.12.mlp.shared_expert.gate_proj",
"model.language_model.layers.12.mlp.shared_expert.up_proj",
"model.language_model.layers.13.mlp.experts",
"model.language_model.layers.13.mlp.shared_expert.down_proj",
"model.language_model.layers.13.mlp.shared_expert.gate_proj",
"model.language_model.layers.13.mlp.shared_expert.up_proj",
"model.language_model.layers.14.mlp.experts",
"model.language_model.layers.14.mlp.shared_expert.down_proj",
"model.language_model.layers.14.mlp.shared_expert.gate_proj",
"model.language_model.layers.14.mlp.shared_expert.up_proj",
"model.language_model.layers.15.mlp.experts",
"model.language_model.layers.15.mlp.shared_expert.down_proj",
"model.language_model.layers.15.mlp.shared_expert.gate_proj",
"model.language_model.layers.15.mlp.shared_expert.up_proj",
"model.language_model.layers.16.mlp.experts",
"model.language_model.layers.16.mlp.shared_expert.down_proj",
"model.language_model.layers.16.mlp.shared_expert.gate_proj",
"model.language_model.layers.16.mlp.shared_expert.up_proj",
"model.language_model.layers.17.mlp.experts",
"model.language_model.layers.17.mlp.shared_expert.down_proj",
"model.language_model.layers.17.mlp.shared_expert.gate_proj",
"model.language_model.layers.17.mlp.shared_expert.up_proj",
"model.language_model.layers.18.mlp.experts",
"model.language_model.layers.18.mlp.shared_expert.down_proj",
"model.language_model.layers.18.mlp.shared_expert.gate_proj",
"model.language_model.layers.18.mlp.shared_expert.up_proj",
"model.language_model.layers.19.mlp.experts",
"model.language_model.layers.19.mlp.shared_expert.down_proj",
"model.language_model.layers.19.mlp.shared_expert.gate_proj",
"model.language_model.layers.19.mlp.shared_expert.up_proj",
"model.language_model.layers.2.mlp.experts",
"model.language_model.layers.2.mlp.shared_expert.down_proj",
"model.language_model.layers.2.mlp.shared_expert.gate_proj",
"model.language_model.layers.2.mlp.shared_expert.up_proj",
"model.language_model.layers.20.mlp.experts",
"model.language_model.layers.20.mlp.shared_expert.down_proj",
"model.language_model.layers.20.mlp.shared_expert.gate_proj",
"model.language_model.layers.20.mlp.shared_expert.up_proj",
"model.language_model.layers.21.mlp.experts",
"model.language_model.layers.21.mlp.shared_expert.down_proj",
"model.language_model.layers.21.mlp.shared_expert.gate_proj",
"model.language_model.layers.21.mlp.shared_expert.up_proj",
"model.language_model.layers.22.mlp.experts",
"model.language_model.layers.22.mlp.shared_expert.down_proj",
"model.language_model.layers.22.mlp.shared_expert.gate_proj",
"model.language_model.layers.22.mlp.shared_expert.up_proj",
"model.language_model.layers.23.mlp.experts",
"model.language_model.layers.23.mlp.shared_expert.down_proj",
"model.language_model.layers.23.mlp.shared_expert.gate_proj",
"model.language_model.layers.23.mlp.shared_expert.up_proj",
"model.language_model.layers.24.mlp.experts",
"model.language_model.layers.24.mlp.shared_expert.down_proj",
"model.language_model.layers.24.mlp.shared_expert.gate_proj",
"model.language_model.layers.24.mlp.shared_expert.up_proj",
"model.language_model.layers.25.mlp.experts",
"model.language_model.layers.25.mlp.shared_expert.down_proj",
"model.language_model.layers.25.mlp.shared_expert.gate_proj",
"model.language_model.layers.25.mlp.shared_expert.up_proj",
"model.language_model.layers.26.mlp.experts",
"model.language_model.layers.26.mlp.shared_expert.down_proj",
"model.language_model.layers.26.mlp.shared_expert.gate_proj",
"model.language_model.layers.26.mlp.shared_expert.up_proj",
"model.language_model.layers.27.mlp.experts",
"model.language_model.layers.27.mlp.shared_expert.down_proj",
"model.language_model.layers.27.mlp.shared_expert.gate_proj",
"model.language_model.layers.27.mlp.shared_expert.up_proj",
"model.language_model.layers.28.mlp.experts",
"model.language_model.layers.28.mlp.shared_expert.down_proj",
"model.language_model.layers.28.mlp.shared_expert.gate_proj",
"model.language_model.layers.28.mlp.shared_expert.up_proj",
"model.language_model.layers.29.mlp.experts",
"model.language_model.layers.29.mlp.shared_expert.down_proj",
"model.language_model.layers.29.mlp.shared_expert.gate_proj",
"model.language_model.layers.29.mlp.shared_expert.up_proj",
"model.language_model.layers.3.mlp.experts",
"model.language_model.layers.3.mlp.shared_expert.down_proj",
"model.language_model.layers.3.mlp.shared_expert.gate_proj",
"model.language_model.layers.3.mlp.shared_expert.up_proj",
"model.language_model.layers.30.mlp.experts",
"model.language_model.layers.30.mlp.shared_expert.down_proj",
"model.language_model.layers.30.mlp.shared_expert.gate_proj",
"model.language_model.layers.30.mlp.shared_expert.up_proj",
"model.language_model.layers.31.mlp.experts",
"model.language_model.layers.31.mlp.shared_expert.down_proj",
"model.language_model.layers.31.mlp.shared_expert.gate_proj",
"model.language_model.layers.31.mlp.shared_expert.up_proj",
"model.language_model.layers.32.mlp.experts",
"model.language_model.layers.32.mlp.shared_expert.down_proj",
"model.language_model.layers.32.mlp.shared_expert.gate_proj",
"model.language_model.layers.32.mlp.shared_expert.up_proj",
"model.language_model.layers.33.mlp.experts",
"model.language_model.layers.33.mlp.shared_expert.down_proj",
"model.language_model.layers.33.mlp.shared_expert.gate_proj",
"model.language_model.layers.33.mlp.shared_expert.up_proj",
"model.language_model.layers.34.mlp.experts",
"model.language_model.layers.34.mlp.shared_expert.down_proj",
"model.language_model.layers.34.mlp.shared_expert.gate_proj",
"model.language_model.layers.34.mlp.shared_expert.up_proj",
"model.language_model.layers.35.mlp.experts",
"model.language_model.layers.35.mlp.shared_expert.down_proj",
"model.language_model.layers.35.mlp.shared_expert.gate_proj",
"model.language_model.layers.35.mlp.shared_expert.up_proj",
"model.language_model.layers.36.mlp.experts",
"model.language_model.layers.36.mlp.shared_expert.down_proj",
"model.language_model.layers.36.mlp.shared_expert.gate_proj",
"model.language_model.layers.36.mlp.shared_expert.up_proj",
"model.language_model.layers.37.mlp.experts",
"model.language_model.layers.37.mlp.shared_expert.down_proj",
"model.language_model.layers.37.mlp.shared_expert.gate_proj",
"model.language_model.layers.37.mlp.shared_expert.up_proj",
"model.language_model.layers.38.mlp.experts",
"model.language_model.layers.38.mlp.shared_expert.down_proj",
"model.language_model.layers.38.mlp.shared_expert.gate_proj",
"model.language_model.layers.38.mlp.shared_expert.up_proj",
"model.language_model.layers.39.mlp.experts",
"model.language_model.layers.39.mlp.shared_expert.down_proj",
"model.language_model.layers.39.mlp.shared_expert.gate_proj",
"model.language_model.layers.39.mlp.shared_expert.up_proj",
"model.language_model.layers.4.mlp.experts",
"model.language_model.layers.4.mlp.shared_expert.down_proj",
"model.language_model.layers.4.mlp.shared_expert.gate_proj",
"model.language_model.layers.4.mlp.shared_expert.up_proj",
"model.language_model.layers.5.mlp.experts",
"model.language_model.layers.5.mlp.shared_expert.down_proj",
"model.language_model.layers.5.mlp.shared_expert.gate_proj",
"model.language_model.layers.5.mlp.shared_expert.up_proj",
"model.language_model.layers.6.mlp.experts",
"model.language_model.layers.6.mlp.shared_expert.down_proj",
"model.language_model.layers.6.mlp.shared_expert.gate_proj",
"model.language_model.layers.6.mlp.shared_expert.up_proj",
"model.language_model.layers.7.mlp.experts",
"model.language_model.layers.7.mlp.shared_expert.down_proj",
"model.language_model.layers.7.mlp.shared_expert.gate_proj",
"model.language_model.layers.7.mlp.shared_expert.up_proj",
"model.language_model.layers.8.mlp.experts",
"model.language_model.layers.8.mlp.shared_expert.down_proj",
"model.language_model.layers.8.mlp.shared_expert.gate_proj",
"model.language_model.layers.8.mlp.shared_expert.up_proj",
"model.language_model.layers.9.mlp.experts",
"model.language_model.layers.9.mlp.shared_expert.down_proj",
"model.language_model.layers.9.mlp.shared_expert.gate_proj",
"model.language_model.layers.9.mlp.shared_expert.up_proj"
],
"weights": {
"dynamic": false,
"group_size": 16,
"num_bits": 4,
"type": "float"
}
}
},
"ignore": [
"mtp.layers.0*",
"mtp*"
],
"producer": {
"name": "modelopt",
"version": "0.37.0"
},
"quant_algo": "MIXED_PRECISION",
"quant_method": "modelopt",
"quantized_layers": {
"lm_head": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.0.linear_attn.in_proj_qkv": {
"quant_algo": "FP8"
},
"model.language_model.layers.0.linear_attn.in_proj_z": {
"quant_algo": "FP8"
},
"model.language_model.layers.0.linear_attn.out_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.0.mlp.experts": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.0.mlp.shared_expert.down_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.0.mlp.shared_expert.gate_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.0.mlp.shared_expert.up_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.1.linear_attn.in_proj_qkv": {
"quant_algo": "FP8"
},
"model.language_model.layers.1.linear_attn.in_proj_z": {
"quant_algo": "FP8"
},
"model.language_model.layers.1.linear_attn.out_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.1.mlp.experts": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.1.mlp.shared_expert.down_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.1.mlp.shared_expert.gate_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.1.mlp.shared_expert.up_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.10.linear_attn.in_proj_qkv": {
"quant_algo": "FP8"
},
"model.language_model.layers.10.linear_attn.in_proj_z": {
"quant_algo": "FP8"
},
"model.language_model.layers.10.linear_attn.out_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.10.mlp.experts": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.10.mlp.shared_expert.down_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.10.mlp.shared_expert.gate_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.10.mlp.shared_expert.up_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.11.mlp.experts": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.11.mlp.shared_expert.down_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.11.mlp.shared_expert.gate_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.11.mlp.shared_expert.up_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.11.self_attn.k_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.11.self_attn.o_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.11.self_attn.q_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.11.self_attn.v_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.12.linear_attn.in_proj_qkv": {
"quant_algo": "FP8"
},
"model.language_model.layers.12.linear_attn.in_proj_z": {
"quant_algo": "FP8"
},
"model.language_model.layers.12.linear_attn.out_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.12.mlp.experts": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.12.mlp.shared_expert.down_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.12.mlp.shared_expert.gate_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.12.mlp.shared_expert.up_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.13.linear_attn.in_proj_qkv": {
"quant_algo": "FP8"
},
"model.language_model.layers.13.linear_attn.in_proj_z": {
"quant_algo": "FP8"
},
"model.language_model.layers.13.linear_attn.out_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.13.mlp.experts": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.13.mlp.shared_expert.down_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.13.mlp.shared_expert.gate_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.13.mlp.shared_expert.up_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.14.linear_attn.in_proj_qkv": {
"quant_algo": "FP8"
},
"model.language_model.layers.14.linear_attn.in_proj_z": {
"quant_algo": "FP8"
},
"model.language_model.layers.14.linear_attn.out_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.14.mlp.experts": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.14.mlp.shared_expert.down_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.14.mlp.shared_expert.gate_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.14.mlp.shared_expert.up_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.15.mlp.experts": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.15.mlp.shared_expert.down_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.15.mlp.shared_expert.gate_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.15.mlp.shared_expert.up_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.15.self_attn.k_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.15.self_attn.o_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.15.self_attn.q_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.15.self_attn.v_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.16.linear_attn.in_proj_qkv": {
"quant_algo": "FP8"
},
"model.language_model.layers.16.linear_attn.in_proj_z": {
"quant_algo": "FP8"
},
"model.language_model.layers.16.linear_attn.out_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.16.mlp.experts": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.16.mlp.shared_expert.down_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.16.mlp.shared_expert.gate_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.16.mlp.shared_expert.up_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.17.linear_attn.in_proj_qkv": {
"quant_algo": "FP8"
},
"model.language_model.layers.17.linear_attn.in_proj_z": {
"quant_algo": "FP8"
},
"model.language_model.layers.17.linear_attn.out_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.17.mlp.experts": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.17.mlp.shared_expert.down_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.17.mlp.shared_expert.gate_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.17.mlp.shared_expert.up_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.18.linear_attn.in_proj_qkv": {
"quant_algo": "FP8"
},
"model.language_model.layers.18.linear_attn.in_proj_z": {
"quant_algo": "FP8"
},
"model.language_model.layers.18.linear_attn.out_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.18.mlp.experts": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.18.mlp.shared_expert.down_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.18.mlp.shared_expert.gate_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.18.mlp.shared_expert.up_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.19.mlp.experts": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.19.mlp.shared_expert.down_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.19.mlp.shared_expert.gate_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.19.mlp.shared_expert.up_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.19.self_attn.k_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.19.self_attn.o_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.19.self_attn.q_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.19.self_attn.v_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.2.linear_attn.in_proj_qkv": {
"quant_algo": "FP8"
},
"model.language_model.layers.2.linear_attn.in_proj_z": {
"quant_algo": "FP8"
},
"model.language_model.layers.2.linear_attn.out_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.2.mlp.experts": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.2.mlp.shared_expert.down_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.2.mlp.shared_expert.gate_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.2.mlp.shared_expert.up_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.20.linear_attn.in_proj_qkv": {
"quant_algo": "FP8"
},
"model.language_model.layers.20.linear_attn.in_proj_z": {
"quant_algo": "FP8"
},
"model.language_model.layers.20.linear_attn.out_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.20.mlp.experts": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.20.mlp.shared_expert.down_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.20.mlp.shared_expert.gate_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.20.mlp.shared_expert.up_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.21.linear_attn.in_proj_qkv": {
"quant_algo": "FP8"
},
"model.language_model.layers.21.linear_attn.in_proj_z": {
"quant_algo": "FP8"
},
"model.language_model.layers.21.linear_attn.out_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.21.mlp.experts": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.21.mlp.shared_expert.down_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.21.mlp.shared_expert.gate_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.21.mlp.shared_expert.up_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.22.linear_attn.in_proj_qkv": {
"quant_algo": "FP8"
},
"model.language_model.layers.22.linear_attn.in_proj_z": {
"quant_algo": "FP8"
},
"model.language_model.layers.22.linear_attn.out_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.22.mlp.experts": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.22.mlp.shared_expert.down_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.22.mlp.shared_expert.gate_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.22.mlp.shared_expert.up_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.23.mlp.experts": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.23.mlp.shared_expert.down_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.23.mlp.shared_expert.gate_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.23.mlp.shared_expert.up_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.23.self_attn.k_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.23.self_attn.o_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.23.self_attn.q_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.23.self_attn.v_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.24.linear_attn.in_proj_qkv": {
"quant_algo": "FP8"
},
"model.language_model.layers.24.linear_attn.in_proj_z": {
"quant_algo": "FP8"
},
"model.language_model.layers.24.linear_attn.out_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.24.mlp.experts": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.24.mlp.shared_expert.down_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.24.mlp.shared_expert.gate_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.24.mlp.shared_expert.up_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.25.linear_attn.in_proj_qkv": {
"quant_algo": "FP8"
},
"model.language_model.layers.25.linear_attn.in_proj_z": {
"quant_algo": "FP8"
},
"model.language_model.layers.25.linear_attn.out_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.25.mlp.experts": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.25.mlp.shared_expert.down_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.25.mlp.shared_expert.gate_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.25.mlp.shared_expert.up_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.26.linear_attn.in_proj_qkv": {
"quant_algo": "FP8"
},
"model.language_model.layers.26.linear_attn.in_proj_z": {
"quant_algo": "FP8"
},
"model.language_model.layers.26.linear_attn.out_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.26.mlp.experts": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.26.mlp.shared_expert.down_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.26.mlp.shared_expert.gate_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.26.mlp.shared_expert.up_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.27.mlp.experts": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.27.mlp.shared_expert.down_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.27.mlp.shared_expert.gate_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.27.mlp.shared_expert.up_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.27.self_attn.k_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.27.self_attn.o_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.27.self_attn.q_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.27.self_attn.v_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.28.linear_attn.in_proj_qkv": {
"quant_algo": "FP8"
},
"model.language_model.layers.28.linear_attn.in_proj_z": {
"quant_algo": "FP8"
},
"model.language_model.layers.28.linear_attn.out_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.28.mlp.experts": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.28.mlp.shared_expert.down_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.28.mlp.shared_expert.gate_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.28.mlp.shared_expert.up_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.29.linear_attn.in_proj_qkv": {
"quant_algo": "FP8"
},
"model.language_model.layers.29.linear_attn.in_proj_z": {
"quant_algo": "FP8"
},
"model.language_model.layers.29.linear_attn.out_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.29.mlp.experts": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.29.mlp.shared_expert.down_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.29.mlp.shared_expert.gate_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.29.mlp.shared_expert.up_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.3.mlp.experts": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.3.mlp.shared_expert.down_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.3.mlp.shared_expert.gate_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.3.mlp.shared_expert.up_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.3.self_attn.k_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.3.self_attn.o_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.3.self_attn.q_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.3.self_attn.v_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.30.linear_attn.in_proj_qkv": {
"quant_algo": "FP8"
},
"model.language_model.layers.30.linear_attn.in_proj_z": {
"quant_algo": "FP8"
},
"model.language_model.layers.30.linear_attn.out_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.30.mlp.experts": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.30.mlp.shared_expert.down_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.30.mlp.shared_expert.gate_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.30.mlp.shared_expert.up_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.31.mlp.experts": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.31.mlp.shared_expert.down_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.31.mlp.shared_expert.gate_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.31.mlp.shared_expert.up_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.31.self_attn.k_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.31.self_attn.o_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.31.self_attn.q_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.31.self_attn.v_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.32.linear_attn.in_proj_qkv": {
"quant_algo": "FP8"
},
"model.language_model.layers.32.linear_attn.in_proj_z": {
"quant_algo": "FP8"
},
"model.language_model.layers.32.linear_attn.out_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.32.mlp.experts": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.32.mlp.shared_expert.down_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.32.mlp.shared_expert.gate_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.32.mlp.shared_expert.up_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.33.linear_attn.in_proj_qkv": {
"quant_algo": "FP8"
},
"model.language_model.layers.33.linear_attn.in_proj_z": {
"quant_algo": "FP8"
},
"model.language_model.layers.33.linear_attn.out_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.33.mlp.experts": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.33.mlp.shared_expert.down_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.33.mlp.shared_expert.gate_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.33.mlp.shared_expert.up_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.34.linear_attn.in_proj_qkv": {
"quant_algo": "FP8"
},
"model.language_model.layers.34.linear_attn.in_proj_z": {
"quant_algo": "FP8"
},
"model.language_model.layers.34.linear_attn.out_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.34.mlp.experts": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.34.mlp.shared_expert.down_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.34.mlp.shared_expert.gate_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.34.mlp.shared_expert.up_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.35.mlp.experts": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.35.mlp.shared_expert.down_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.35.mlp.shared_expert.gate_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.35.mlp.shared_expert.up_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.35.self_attn.k_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.35.self_attn.o_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.35.self_attn.q_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.35.self_attn.v_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.36.linear_attn.in_proj_qkv": {
"quant_algo": "FP8"
},
"model.language_model.layers.36.linear_attn.in_proj_z": {
"quant_algo": "FP8"
},
"model.language_model.layers.36.linear_attn.out_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.36.mlp.experts": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.36.mlp.shared_expert.down_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.36.mlp.shared_expert.gate_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.36.mlp.shared_expert.up_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.37.linear_attn.in_proj_qkv": {
"quant_algo": "FP8"
},
"model.language_model.layers.37.linear_attn.in_proj_z": {
"quant_algo": "FP8"
},
"model.language_model.layers.37.linear_attn.out_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.37.mlp.experts": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.37.mlp.shared_expert.down_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.37.mlp.shared_expert.gate_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.37.mlp.shared_expert.up_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.38.linear_attn.in_proj_qkv": {
"quant_algo": "FP8"
},
"model.language_model.layers.38.linear_attn.in_proj_z": {
"quant_algo": "FP8"
},
"model.language_model.layers.38.linear_attn.out_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.38.mlp.experts": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.38.mlp.shared_expert.down_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.38.mlp.shared_expert.gate_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.38.mlp.shared_expert.up_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.39.mlp.experts": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.39.mlp.shared_expert.down_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.39.mlp.shared_expert.gate_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.39.mlp.shared_expert.up_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.39.self_attn.k_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.39.self_attn.o_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.39.self_attn.q_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.39.self_attn.v_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.4.linear_attn.in_proj_qkv": {
"quant_algo": "FP8"
},
"model.language_model.layers.4.linear_attn.in_proj_z": {
"quant_algo": "FP8"
},
"model.language_model.layers.4.linear_attn.out_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.4.mlp.experts": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.4.mlp.shared_expert.down_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.4.mlp.shared_expert.gate_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.4.mlp.shared_expert.up_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.5.linear_attn.in_proj_qkv": {
"quant_algo": "FP8"
},
"model.language_model.layers.5.linear_attn.in_proj_z": {
"quant_algo": "FP8"
},
"model.language_model.layers.5.linear_attn.out_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.5.mlp.experts": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.5.mlp.shared_expert.down_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.5.mlp.shared_expert.gate_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.5.mlp.shared_expert.up_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.6.linear_attn.in_proj_qkv": {
"quant_algo": "FP8"
},
"model.language_model.layers.6.linear_attn.in_proj_z": {
"quant_algo": "FP8"
},
"model.language_model.layers.6.linear_attn.out_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.6.mlp.experts": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.6.mlp.shared_expert.down_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.6.mlp.shared_expert.gate_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.6.mlp.shared_expert.up_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.7.mlp.experts": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.7.mlp.shared_expert.down_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.7.mlp.shared_expert.gate_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.7.mlp.shared_expert.up_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.7.self_attn.k_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.7.self_attn.o_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.7.self_attn.q_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.7.self_attn.v_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.8.linear_attn.in_proj_qkv": {
"quant_algo": "FP8"
},
"model.language_model.layers.8.linear_attn.in_proj_z": {
"quant_algo": "FP8"
},
"model.language_model.layers.8.linear_attn.out_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.8.mlp.experts": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.8.mlp.shared_expert.down_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.8.mlp.shared_expert.gate_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.8.mlp.shared_expert.up_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.9.linear_attn.in_proj_qkv": {
"quant_algo": "FP8"
},
"model.language_model.layers.9.linear_attn.in_proj_z": {
"quant_algo": "FP8"
},
"model.language_model.layers.9.linear_attn.out_proj": {
"quant_algo": "FP8"
},
"model.language_model.layers.9.mlp.experts": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.9.mlp.shared_expert.down_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.9.mlp.shared_expert.gate_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
},
"model.language_model.layers.9.mlp.shared_expert.up_proj": {
"group_size": 16,
"quant_algo": "W4A16_NVFP4"
}
}
}
},
"mlx_modelopt_quantization": {
"language_model.lm_head": "scaled_nvfp4",
"language_model.model.layers.0.linear_attn.in_proj_qkv": "scaled_mxfp8",
"language_model.model.layers.0.linear_attn.in_proj_z": "scaled_mxfp8",
"language_model.model.layers.0.linear_attn.out_proj": "scaled_mxfp8",
"language_model.model.layers.0.mlp.shared_expert.down_proj": "scaled_nvfp4",
"language_model.model.layers.0.mlp.shared_expert.gate_proj": "scaled_nvfp4",
"language_model.model.layers.0.mlp.shared_expert.up_proj": "scaled_nvfp4",
"language_model.model.layers.0.mlp.switch_mlp.down_proj": "scaled_nvfp4_switch",
"language_model.model.layers.0.mlp.switch_mlp.gate_proj": "scaled_nvfp4_switch",
"language_model.model.layers.0.mlp.switch_mlp.up_proj": "scaled_nvfp4_switch",
"language_model.model.layers.1.linear_attn.in_proj_qkv": "scaled_mxfp8",
"language_model.model.layers.1.linear_attn.in_proj_z": "scaled_mxfp8",
"language_model.model.layers.1.linear_attn.out_proj": "scaled_mxfp8",
"language_model.model.layers.1.mlp.shared_expert.down_proj": "scaled_nvfp4",
"language_model.model.layers.1.mlp.shared_expert.gate_proj": "scaled_nvfp4",
"language_model.model.layers.1.mlp.shared_expert.up_proj": "scaled_nvfp4",
"language_model.model.layers.1.mlp.switch_mlp.down_proj": "scaled_nvfp4_switch",
"language_model.model.layers.1.mlp.switch_mlp.gate_proj": "scaled_nvfp4_switch",
"language_model.model.layers.1.mlp.switch_mlp.up_proj": "scaled_nvfp4_switch",
"language_model.model.layers.10.linear_attn.in_proj_qkv": "scaled_mxfp8",
"language_model.model.layers.10.linear_attn.in_proj_z": "scaled_mxfp8",
"language_model.model.layers.10.linear_attn.out_proj": "scaled_mxfp8",
"language_model.model.layers.10.mlp.shared_expert.down_proj": "scaled_nvfp4",
"language_model.model.layers.10.mlp.shared_expert.gate_proj": "scaled_nvfp4",
"language_model.model.layers.10.mlp.shared_expert.up_proj": "scaled_nvfp4",
"language_model.model.layers.10.mlp.switch_mlp.down_proj": "scaled_nvfp4_switch",
"language_model.model.layers.10.mlp.switch_mlp.gate_proj": "scaled_nvfp4_switch",
"language_model.model.layers.10.mlp.switch_mlp.up_proj": "scaled_nvfp4_switch",
"language_model.model.layers.11.mlp.shared_expert.down_proj": "scaled_nvfp4",
"language_model.model.layers.11.mlp.shared_expert.gate_proj": "scaled_nvfp4",
"language_model.model.layers.11.mlp.shared_expert.up_proj": "scaled_nvfp4",
"language_model.model.layers.11.mlp.switch_mlp.down_proj": "scaled_nvfp4_switch",
"language_model.model.layers.11.mlp.switch_mlp.gate_proj": "scaled_nvfp4_switch",
"language_model.model.layers.11.mlp.switch_mlp.up_proj": "scaled_nvfp4_switch",
"language_model.model.layers.11.self_attn.k_proj": "scaled_mxfp8",
"language_model.model.layers.11.self_attn.o_proj": "scaled_mxfp8",
"language_model.model.layers.11.self_attn.q_proj": "scaled_mxfp8",
"language_model.model.layers.11.self_attn.v_proj": "scaled_mxfp8",
"language_model.model.layers.12.linear_attn.in_proj_qkv": "scaled_mxfp8",
"language_model.model.layers.12.linear_attn.in_proj_z": "scaled_mxfp8",
"language_model.model.layers.12.linear_attn.out_proj": "scaled_mxfp8",
"language_model.model.layers.12.mlp.shared_expert.down_proj": "scaled_nvfp4",
"language_model.model.layers.12.mlp.shared_expert.gate_proj": "scaled_nvfp4",
"language_model.model.layers.12.mlp.shared_expert.up_proj": "scaled_nvfp4",
"language_model.model.layers.12.mlp.switch_mlp.down_proj": "scaled_nvfp4_switch",
"language_model.model.layers.12.mlp.switch_mlp.gate_proj": "scaled_nvfp4_switch",
"language_model.model.layers.12.mlp.switch_mlp.up_proj": "scaled_nvfp4_switch",
"language_model.model.layers.13.linear_attn.in_proj_qkv": "scaled_mxfp8",
"language_model.model.layers.13.linear_attn.in_proj_z": "scaled_mxfp8",
"language_model.model.layers.13.linear_attn.out_proj": "scaled_mxfp8",
"language_model.model.layers.13.mlp.shared_expert.down_proj": "scaled_nvfp4",
"language_model.model.layers.13.mlp.shared_expert.gate_proj": "scaled_nvfp4",
"language_model.model.layers.13.mlp.shared_expert.up_proj": "scaled_nvfp4",
"language_model.model.layers.13.mlp.switch_mlp.down_proj": "scaled_nvfp4_switch",
"language_model.model.layers.13.mlp.switch_mlp.gate_proj": "scaled_nvfp4_switch",
"language_model.model.layers.13.mlp.switch_mlp.up_proj": "scaled_nvfp4_switch",
"language_model.model.layers.14.linear_attn.in_proj_qkv": "scaled_mxfp8",
"language_model.model.layers.14.linear_attn.in_proj_z": "scaled_mxfp8",
"language_model.model.layers.14.linear_attn.out_proj": "scaled_mxfp8",
"language_model.model.layers.14.mlp.shared_expert.down_proj": "scaled_nvfp4",
"language_model.model.layers.14.mlp.shared_expert.gate_proj": "scaled_nvfp4",
"language_model.model.layers.14.mlp.shared_expert.up_proj": "scaled_nvfp4",
"language_model.model.layers.14.mlp.switch_mlp.down_proj": "scaled_nvfp4_switch",
"language_model.model.layers.14.mlp.switch_mlp.gate_proj": "scaled_nvfp4_switch",
"language_model.model.layers.14.mlp.switch_mlp.up_proj": "scaled_nvfp4_switch",
"language_model.model.layers.15.mlp.shared_expert.down_proj": "scaled_nvfp4",
"language_model.model.layers.15.mlp.shared_expert.gate_proj": "scaled_nvfp4",
"language_model.model.layers.15.mlp.shared_expert.up_proj": "scaled_nvfp4",
"language_model.model.layers.15.mlp.switch_mlp.down_proj": "scaled_nvfp4_switch",
"language_model.model.layers.15.mlp.switch_mlp.gate_proj": "scaled_nvfp4_switch",
"language_model.model.layers.15.mlp.switch_mlp.up_proj": "scaled_nvfp4_switch",
"language_model.model.layers.15.self_attn.k_proj": "scaled_mxfp8",
"language_model.model.layers.15.self_attn.o_proj": "scaled_mxfp8",
"language_model.model.layers.15.self_attn.q_proj": "scaled_mxfp8",
"language_model.model.layers.15.self_attn.v_proj": "scaled_mxfp8",
"language_model.model.layers.16.linear_attn.in_proj_qkv": "scaled_mxfp8",
"language_model.model.layers.16.linear_attn.in_proj_z": "scaled_mxfp8",
"language_model.model.layers.16.linear_attn.out_proj": "scaled_mxfp8",
"language_model.model.layers.16.mlp.shared_expert.down_proj": "scaled_nvfp4",
"language_model.model.layers.16.mlp.shared_expert.gate_proj": "scaled_nvfp4",
"language_model.model.layers.16.mlp.shared_expert.up_proj": "scaled_nvfp4",
"language_model.model.layers.16.mlp.switch_mlp.down_proj": "scaled_nvfp4_switch",
"language_model.model.layers.16.mlp.switch_mlp.gate_proj": "scaled_nvfp4_switch",
"language_model.model.layers.16.mlp.switch_mlp.up_proj": "scaled_nvfp4_switch",
"language_model.model.layers.17.linear_attn.in_proj_qkv": "scaled_mxfp8",
"language_model.model.layers.17.linear_attn.in_proj_z": "scaled_mxfp8",
"language_model.model.layers.17.linear_attn.out_proj": "scaled_mxfp8",
"language_model.model.layers.17.mlp.shared_expert.down_proj": "scaled_nvfp4",
"language_model.model.layers.17.mlp.shared_expert.gate_proj": "scaled_nvfp4",
"language_model.model.layers.17.mlp.shared_expert.up_proj": "scaled_nvfp4",
"language_model.model.layers.17.mlp.switch_mlp.down_proj": "scaled_nvfp4_switch",
"language_model.model.layers.17.mlp.switch_mlp.gate_proj": "scaled_nvfp4_switch",
"language_model.model.layers.17.mlp.switch_mlp.up_proj": "scaled_nvfp4_switch",
"language_model.model.layers.18.linear_attn.in_proj_qkv": "scaled_mxfp8",
"language_model.model.layers.18.linear_attn.in_proj_z": "scaled_mxfp8",
"language_model.model.layers.18.linear_attn.out_proj": "scaled_mxfp8",
"language_model.model.layers.18.mlp.shared_expert.down_proj": "scaled_nvfp4",
"language_model.model.layers.18.mlp.shared_expert.gate_proj": "scaled_nvfp4",
"language_model.model.layers.18.mlp.shared_expert.up_proj": "scaled_nvfp4",
"language_model.model.layers.18.mlp.switch_mlp.down_proj": "scaled_nvfp4_switch",
"language_model.model.layers.18.mlp.switch_mlp.gate_proj": "scaled_nvfp4_switch",
"language_model.model.layers.18.mlp.switch_mlp.up_proj": "scaled_nvfp4_switch",
"language_model.model.layers.19.mlp.shared_expert.down_proj": "scaled_nvfp4",
"language_model.model.layers.19.mlp.shared_expert.gate_proj": "scaled_nvfp4",
"language_model.model.layers.19.mlp.shared_expert.up_proj": "scaled_nvfp4",
"language_model.model.layers.19.mlp.switch_mlp.down_proj": "scaled_nvfp4_switch",
"language_model.model.layers.19.mlp.switch_mlp.gate_proj": "scaled_nvfp4_switch",
"language_model.model.layers.19.mlp.switch_mlp.up_proj": "scaled_nvfp4_switch",
"language_model.model.layers.19.self_attn.k_proj": "scaled_mxfp8",
"language_model.model.layers.19.self_attn.o_proj": "scaled_mxfp8",
"language_model.model.layers.19.self_attn.q_proj": "scaled_mxfp8",
"language_model.model.layers.19.self_attn.v_proj": "scaled_mxfp8",
"language_model.model.layers.2.linear_attn.in_proj_qkv": "scaled_mxfp8",
"language_model.model.layers.2.linear_attn.in_proj_z": "scaled_mxfp8",
"language_model.model.layers.2.linear_attn.out_proj": "scaled_mxfp8",
"language_model.model.layers.2.mlp.shared_expert.down_proj": "scaled_nvfp4",
"language_model.model.layers.2.mlp.shared_expert.gate_proj": "scaled_nvfp4",
"language_model.model.layers.2.mlp.shared_expert.up_proj": "scaled_nvfp4",
"language_model.model.layers.2.mlp.switch_mlp.down_proj": "scaled_nvfp4_switch",
"language_model.model.layers.2.mlp.switch_mlp.gate_proj": "scaled_nvfp4_switch",
"language_model.model.layers.2.mlp.switch_mlp.up_proj": "scaled_nvfp4_switch",
"language_model.model.layers.20.linear_attn.in_proj_qkv": "scaled_mxfp8",
"language_model.model.layers.20.linear_attn.in_proj_z": "scaled_mxfp8",
"language_model.model.layers.20.linear_attn.out_proj": "scaled_mxfp8",
"language_model.model.layers.20.mlp.shared_expert.down_proj": "scaled_nvfp4",
"language_model.model.layers.20.mlp.shared_expert.gate_proj": "scaled_nvfp4",
"language_model.model.layers.20.mlp.shared_expert.up_proj": "scaled_nvfp4",
"language_model.model.layers.20.mlp.switch_mlp.down_proj": "scaled_nvfp4_switch",
"language_model.model.layers.20.mlp.switch_mlp.gate_proj": "scaled_nvfp4_switch",
"language_model.model.layers.20.mlp.switch_mlp.up_proj": "scaled_nvfp4_switch",
"language_model.model.layers.21.linear_attn.in_proj_qkv": "scaled_mxfp8",
"language_model.model.layers.21.linear_attn.in_proj_z": "scaled_mxfp8",
"language_model.model.layers.21.linear_attn.out_proj": "scaled_mxfp8",
"language_model.model.layers.21.mlp.shared_expert.down_proj": "scaled_nvfp4",
"language_model.model.layers.21.mlp.shared_expert.gate_proj": "scaled_nvfp4",
"language_model.model.layers.21.mlp.shared_expert.up_proj": "scaled_nvfp4",
"language_model.model.layers.21.mlp.switch_mlp.down_proj": "scaled_nvfp4_switch",
"language_model.model.layers.21.mlp.switch_mlp.gate_proj": "scaled_nvfp4_switch",
"language_model.model.layers.21.mlp.switch_mlp.up_proj": "scaled_nvfp4_switch",
"language_model.model.layers.22.linear_attn.in_proj_qkv": "scaled_mxfp8",
"language_model.model.layers.22.linear_attn.in_proj_z": "scaled_mxfp8",
"language_model.model.layers.22.linear_attn.out_proj": "scaled_mxfp8",
"language_model.model.layers.22.mlp.shared_expert.down_proj": "scaled_nvfp4",
"language_model.model.layers.22.mlp.shared_expert.gate_proj": "scaled_nvfp4",
"language_model.model.layers.22.mlp.shared_expert.up_proj": "scaled_nvfp4",
"language_model.model.layers.22.mlp.switch_mlp.down_proj": "scaled_nvfp4_switch",
"language_model.model.layers.22.mlp.switch_mlp.gate_proj": "scaled_nvfp4_switch",
"language_model.model.layers.22.mlp.switch_mlp.up_proj": "scaled_nvfp4_switch",
"language_model.model.layers.23.mlp.shared_expert.down_proj": "scaled_nvfp4",
"language_model.model.layers.23.mlp.shared_expert.gate_proj": "scaled_nvfp4",
"language_model.model.layers.23.mlp.shared_expert.up_proj": "scaled_nvfp4",
"language_model.model.layers.23.mlp.switch_mlp.down_proj": "scaled_nvfp4_switch",
"language_model.model.layers.23.mlp.switch_mlp.gate_proj": "scaled_nvfp4_switch",
"language_model.model.layers.23.mlp.switch_mlp.up_proj": "scaled_nvfp4_switch",
"language_model.model.layers.23.self_attn.k_proj": "scaled_mxfp8",
"language_model.model.layers.23.self_attn.o_proj": "scaled_mxfp8",
"language_model.model.layers.23.self_attn.q_proj": "scaled_mxfp8",
"language_model.model.layers.23.self_attn.v_proj": "scaled_mxfp8",
"language_model.model.layers.24.linear_attn.in_proj_qkv": "scaled_mxfp8",
"language_model.model.layers.24.linear_attn.in_proj_z": "scaled_mxfp8",
"language_model.model.layers.24.linear_attn.out_proj": "scaled_mxfp8",
"language_model.model.layers.24.mlp.shared_expert.down_proj": "scaled_nvfp4",
"language_model.model.layers.24.mlp.shared_expert.gate_proj": "scaled_nvfp4",
"language_model.model.layers.24.mlp.shared_expert.up_proj": "scaled_nvfp4",
"language_model.model.layers.24.mlp.switch_mlp.down_proj": "scaled_nvfp4_switch",
"language_model.model.layers.24.mlp.switch_mlp.gate_proj": "scaled_nvfp4_switch",
"language_model.model.layers.24.mlp.switch_mlp.up_proj": "scaled_nvfp4_switch",
"language_model.model.layers.25.linear_attn.in_proj_qkv": "scaled_mxfp8",
"language_model.model.layers.25.linear_attn.in_proj_z": "scaled_mxfp8",
"language_model.model.layers.25.linear_attn.out_proj": "scaled_mxfp8",
"language_model.model.layers.25.mlp.shared_expert.down_proj": "scaled_nvfp4",
"language_model.model.layers.25.mlp.shared_expert.gate_proj": "scaled_nvfp4",
"language_model.model.layers.25.mlp.shared_expert.up_proj": "scaled_nvfp4",
"language_model.model.layers.25.mlp.switch_mlp.down_proj": "scaled_nvfp4_switch",
"language_model.model.layers.25.mlp.switch_mlp.gate_proj": "scaled_nvfp4_switch",
"language_model.model.layers.25.mlp.switch_mlp.up_proj": "scaled_nvfp4_switch",
"language_model.model.layers.26.linear_attn.in_proj_qkv": "scaled_mxfp8",
"language_model.model.layers.26.linear_attn.in_proj_z": "scaled_mxfp8",
"language_model.model.layers.26.linear_attn.out_proj": "scaled_mxfp8",
"language_model.model.layers.26.mlp.shared_expert.down_proj": "scaled_nvfp4",
"language_model.model.layers.26.mlp.shared_expert.gate_proj": "scaled_nvfp4",
"language_model.model.layers.26.mlp.shared_expert.up_proj": "scaled_nvfp4",
"language_model.model.layers.26.mlp.switch_mlp.down_proj": "scaled_nvfp4_switch",
"language_model.model.layers.26.mlp.switch_mlp.gate_proj": "scaled_nvfp4_switch",
"language_model.model.layers.26.mlp.switch_mlp.up_proj": "scaled_nvfp4_switch",
"language_model.model.layers.27.mlp.shared_expert.down_proj": "scaled_nvfp4",
"language_model.model.layers.27.mlp.shared_expert.gate_proj": "scaled_nvfp4",
"language_model.model.layers.27.mlp.shared_expert.up_proj": "scaled_nvfp4",
"language_model.model.layers.27.mlp.switch_mlp.down_proj": "scaled_nvfp4_switch",
"language_model.model.layers.27.mlp.switch_mlp.gate_proj": "scaled_nvfp4_switch",
"language_model.model.layers.27.mlp.switch_mlp.up_proj": "scaled_nvfp4_switch",
"language_model.model.layers.27.self_attn.k_proj": "scaled_mxfp8",
"language_model.model.layers.27.self_attn.o_proj": "scaled_mxfp8",
"language_model.model.layers.27.self_attn.q_proj": "scaled_mxfp8",
"language_model.model.layers.27.self_attn.v_proj": "scaled_mxfp8",
"language_model.model.layers.28.linear_attn.in_proj_qkv": "scaled_mxfp8",
"language_model.model.layers.28.linear_attn.in_proj_z": "scaled_mxfp8",
"language_model.model.layers.28.linear_attn.out_proj": "scaled_mxfp8",
"language_model.model.layers.28.mlp.shared_expert.down_proj": "scaled_nvfp4",
"language_model.model.layers.28.mlp.shared_expert.gate_proj": "scaled_nvfp4",
"language_model.model.layers.28.mlp.shared_expert.up_proj": "scaled_nvfp4",
"language_model.model.layers.28.mlp.switch_mlp.down_proj": "scaled_nvfp4_switch",
"language_model.model.layers.28.mlp.switch_mlp.gate_proj": "scaled_nvfp4_switch",
"language_model.model.layers.28.mlp.switch_mlp.up_proj": "scaled_nvfp4_switch",
"language_model.model.layers.29.linear_attn.in_proj_qkv": "scaled_mxfp8",
"language_model.model.layers.29.linear_attn.in_proj_z": "scaled_mxfp8",
"language_model.model.layers.29.linear_attn.out_proj": "scaled_mxfp8",
"language_model.model.layers.29.mlp.shared_expert.down_proj": "scaled_nvfp4",
"language_model.model.layers.29.mlp.shared_expert.gate_proj": "scaled_nvfp4",
"language_model.model.layers.29.mlp.shared_expert.up_proj": "scaled_nvfp4",
"language_model.model.layers.29.mlp.switch_mlp.down_proj": "scaled_nvfp4_switch",
"language_model.model.layers.29.mlp.switch_mlp.gate_proj": "scaled_nvfp4_switch",
"language_model.model.layers.29.mlp.switch_mlp.up_proj": "scaled_nvfp4_switch",
"language_model.model.layers.3.mlp.shared_expert.down_proj": "scaled_nvfp4",
"language_model.model.layers.3.mlp.shared_expert.gate_proj": "scaled_nvfp4",
"language_model.model.layers.3.mlp.shared_expert.up_proj": "scaled_nvfp4",
"language_model.model.layers.3.mlp.switch_mlp.down_proj": "scaled_nvfp4_switch",
"language_model.model.layers.3.mlp.switch_mlp.gate_proj": "scaled_nvfp4_switch",
"language_model.model.layers.3.mlp.switch_mlp.up_proj": "scaled_nvfp4_switch",
"language_model.model.layers.3.self_attn.k_proj": "scaled_mxfp8",
"language_model.model.layers.3.self_attn.o_proj": "scaled_mxfp8",
"language_model.model.layers.3.self_attn.q_proj": "scaled_mxfp8",
"language_model.model.layers.3.self_attn.v_proj": "scaled_mxfp8",
"language_model.model.layers.30.linear_attn.in_proj_qkv": "scaled_mxfp8",
"language_model.model.layers.30.linear_attn.in_proj_z": "scaled_mxfp8",
"language_model.model.layers.30.linear_attn.out_proj": "scaled_mxfp8",
"language_model.model.layers.30.mlp.shared_expert.down_proj": "scaled_nvfp4",
"language_model.model.layers.30.mlp.shared_expert.gate_proj": "scaled_nvfp4",
"language_model.model.layers.30.mlp.shared_expert.up_proj": "scaled_nvfp4",
"language_model.model.layers.30.mlp.switch_mlp.down_proj": "scaled_nvfp4_switch",
"language_model.model.layers.30.mlp.switch_mlp.gate_proj": "scaled_nvfp4_switch",
"language_model.model.layers.30.mlp.switch_mlp.up_proj": "scaled_nvfp4_switch",
"language_model.model.layers.31.mlp.shared_expert.down_proj": "scaled_nvfp4",
"language_model.model.layers.31.mlp.shared_expert.gate_proj": "scaled_nvfp4",
"language_model.model.layers.31.mlp.shared_expert.up_proj": "scaled_nvfp4",
"language_model.model.layers.31.mlp.switch_mlp.down_proj": "scaled_nvfp4_switch",
"language_model.model.layers.31.mlp.switch_mlp.gate_proj": "scaled_nvfp4_switch",
"language_model.model.layers.31.mlp.switch_mlp.up_proj": "scaled_nvfp4_switch",
"language_model.model.layers.31.self_attn.k_proj": "scaled_mxfp8",
"language_model.model.layers.31.self_attn.o_proj": "scaled_mxfp8",
"language_model.model.layers.31.self_attn.q_proj": "scaled_mxfp8",
"language_model.model.layers.31.self_attn.v_proj": "scaled_mxfp8",
"language_model.model.layers.32.linear_attn.in_proj_qkv": "scaled_mxfp8",
"language_model.model.layers.32.linear_attn.in_proj_z": "scaled_mxfp8",
"language_model.model.layers.32.linear_attn.out_proj": "scaled_mxfp8",
"language_model.model.layers.32.mlp.shared_expert.down_proj": "scaled_nvfp4",
"language_model.model.layers.32.mlp.shared_expert.gate_proj": "scaled_nvfp4",
"language_model.model.layers.32.mlp.shared_expert.up_proj": "scaled_nvfp4",
"language_model.model.layers.32.mlp.switch_mlp.down_proj": "scaled_nvfp4_switch",
"language_model.model.layers.32.mlp.switch_mlp.gate_proj": "scaled_nvfp4_switch",
"language_model.model.layers.32.mlp.switch_mlp.up_proj": "scaled_nvfp4_switch",
"language_model.model.layers.33.linear_attn.in_proj_qkv": "scaled_mxfp8",
"language_model.model.layers.33.linear_attn.in_proj_z": "scaled_mxfp8",
"language_model.model.layers.33.linear_attn.out_proj": "scaled_mxfp8",
"language_model.model.layers.33.mlp.shared_expert.down_proj": "scaled_nvfp4",
"language_model.model.layers.33.mlp.shared_expert.gate_proj": "scaled_nvfp4",
"language_model.model.layers.33.mlp.shared_expert.up_proj": "scaled_nvfp4",
"language_model.model.layers.33.mlp.switch_mlp.down_proj": "scaled_nvfp4_switch",
"language_model.model.layers.33.mlp.switch_mlp.gate_proj": "scaled_nvfp4_switch",
"language_model.model.layers.33.mlp.switch_mlp.up_proj": "scaled_nvfp4_switch",
"language_model.model.layers.34.linear_attn.in_proj_qkv": "scaled_mxfp8",
"language_model.model.layers.34.linear_attn.in_proj_z": "scaled_mxfp8",
"language_model.model.layers.34.linear_attn.out_proj": "scaled_mxfp8",
"language_model.model.layers.34.mlp.shared_expert.down_proj": "scaled_nvfp4",
"language_model.model.layers.34.mlp.shared_expert.gate_proj": "scaled_nvfp4",
"language_model.model.layers.34.mlp.shared_expert.up_proj": "scaled_nvfp4",
"language_model.model.layers.34.mlp.switch_mlp.down_proj": "scaled_nvfp4_switch",
"language_model.model.layers.34.mlp.switch_mlp.gate_proj": "scaled_nvfp4_switch",
"language_model.model.layers.34.mlp.switch_mlp.up_proj": "scaled_nvfp4_switch",
"language_model.model.layers.35.mlp.shared_expert.down_proj": "scaled_nvfp4",
"language_model.model.layers.35.mlp.shared_expert.gate_proj": "scaled_nvfp4",
"language_model.model.layers.35.mlp.shared_expert.up_proj": "scaled_nvfp4",
"language_model.model.layers.35.mlp.switch_mlp.down_proj": "scaled_nvfp4_switch",
"language_model.model.layers.35.mlp.switch_mlp.gate_proj": "scaled_nvfp4_switch",
"language_model.model.layers.35.mlp.switch_mlp.up_proj": "scaled_nvfp4_switch",
"language_model.model.layers.35.self_attn.k_proj": "scaled_mxfp8",
"language_model.model.layers.35.self_attn.o_proj": "scaled_mxfp8",
"language_model.model.layers.35.self_attn.q_proj": "scaled_mxfp8",
"language_model.model.layers.35.self_attn.v_proj": "scaled_mxfp8",
"language_model.model.layers.36.linear_attn.in_proj_qkv": "scaled_mxfp8",
"language_model.model.layers.36.linear_attn.in_proj_z": "scaled_mxfp8",
"language_model.model.layers.36.linear_attn.out_proj": "scaled_mxfp8",
"language_model.model.layers.36.mlp.shared_expert.down_proj": "scaled_nvfp4",
"language_model.model.layers.36.mlp.shared_expert.gate_proj": "scaled_nvfp4",
"language_model.model.layers.36.mlp.shared_expert.up_proj": "scaled_nvfp4",
"language_model.model.layers.36.mlp.switch_mlp.down_proj": "scaled_nvfp4_switch",
"language_model.model.layers.36.mlp.switch_mlp.gate_proj": "scaled_nvfp4_switch",
"language_model.model.layers.36.mlp.switch_mlp.up_proj": "scaled_nvfp4_switch",
"language_model.model.layers.37.linear_attn.in_proj_qkv": "scaled_mxfp8",
"language_model.model.layers.37.linear_attn.in_proj_z": "scaled_mxfp8",
"language_model.model.layers.37.linear_attn.out_proj": "scaled_mxfp8",
"language_model.model.layers.37.mlp.shared_expert.down_proj": "scaled_nvfp4",
"language_model.model.layers.37.mlp.shared_expert.gate_proj": "scaled_nvfp4",
"language_model.model.layers.37.mlp.shared_expert.up_proj": "scaled_nvfp4",
"language_model.model.layers.37.mlp.switch_mlp.down_proj": "scaled_nvfp4_switch",
"language_model.model.layers.37.mlp.switch_mlp.gate_proj": "scaled_nvfp4_switch",
"language_model.model.layers.37.mlp.switch_mlp.up_proj": "scaled_nvfp4_switch",
"language_model.model.layers.38.linear_attn.in_proj_qkv": "scaled_mxfp8",
"language_model.model.layers.38.linear_attn.in_proj_z": "scaled_mxfp8",
"language_model.model.layers.38.linear_attn.out_proj": "scaled_mxfp8",
"language_model.model.layers.38.mlp.shared_expert.down_proj": "scaled_nvfp4",
"language_model.model.layers.38.mlp.shared_expert.gate_proj": "scaled_nvfp4",
"language_model.model.layers.38.mlp.shared_expert.up_proj": "scaled_nvfp4",
"language_model.model.layers.38.mlp.switch_mlp.down_proj": "scaled_nvfp4_switch",
"language_model.model.layers.38.mlp.switch_mlp.gate_proj": "scaled_nvfp4_switch",
"language_model.model.layers.38.mlp.switch_mlp.up_proj": "scaled_nvfp4_switch",
"language_model.model.layers.39.mlp.shared_expert.down_proj": "scaled_nvfp4",
"language_model.model.layers.39.mlp.shared_expert.gate_proj": "scaled_nvfp4",
"language_model.model.layers.39.mlp.shared_expert.up_proj": "scaled_nvfp4",
"language_model.model.layers.39.mlp.switch_mlp.down_proj": "scaled_nvfp4_switch",
"language_model.model.layers.39.mlp.switch_mlp.gate_proj": "scaled_nvfp4_switch",
"language_model.model.layers.39.mlp.switch_mlp.up_proj": "scaled_nvfp4_switch",
"language_model.model.layers.39.self_attn.k_proj": "scaled_mxfp8",
"language_model.model.layers.39.self_attn.o_proj": "scaled_mxfp8",
"language_model.model.layers.39.self_attn.q_proj": "scaled_mxfp8",
"language_model.model.layers.39.self_attn.v_proj": "scaled_mxfp8",
"language_model.model.layers.4.linear_attn.in_proj_qkv": "scaled_mxfp8",
"language_model.model.layers.4.linear_attn.in_proj_z": "scaled_mxfp8",
"language_model.model.layers.4.linear_attn.out_proj": "scaled_mxfp8",
"language_model.model.layers.4.mlp.shared_expert.down_proj": "scaled_nvfp4",
"language_model.model.layers.4.mlp.shared_expert.gate_proj": "scaled_nvfp4",
"language_model.model.layers.4.mlp.shared_expert.up_proj": "scaled_nvfp4",
"language_model.model.layers.4.mlp.switch_mlp.down_proj": "scaled_nvfp4_switch",
"language_model.model.layers.4.mlp.switch_mlp.gate_proj": "scaled_nvfp4_switch",
"language_model.model.layers.4.mlp.switch_mlp.up_proj": "scaled_nvfp4_switch",
"language_model.model.layers.5.linear_attn.in_proj_qkv": "scaled_mxfp8",
"language_model.model.layers.5.linear_attn.in_proj_z": "scaled_mxfp8",
"language_model.model.layers.5.linear_attn.out_proj": "scaled_mxfp8",
"language_model.model.layers.5.mlp.shared_expert.down_proj": "scaled_nvfp4",
"language_model.model.layers.5.mlp.shared_expert.gate_proj": "scaled_nvfp4",
"language_model.model.layers.5.mlp.shared_expert.up_proj": "scaled_nvfp4",
"language_model.model.layers.5.mlp.switch_mlp.down_proj": "scaled_nvfp4_switch",
"language_model.model.layers.5.mlp.switch_mlp.gate_proj": "scaled_nvfp4_switch",
"language_model.model.layers.5.mlp.switch_mlp.up_proj": "scaled_nvfp4_switch",
"language_model.model.layers.6.linear_attn.in_proj_qkv": "scaled_mxfp8",
"language_model.model.layers.6.linear_attn.in_proj_z": "scaled_mxfp8",
"language_model.model.layers.6.linear_attn.out_proj": "scaled_mxfp8",
"language_model.model.layers.6.mlp.shared_expert.down_proj": "scaled_nvfp4",
"language_model.model.layers.6.mlp.shared_expert.gate_proj": "scaled_nvfp4",
"language_model.model.layers.6.mlp.shared_expert.up_proj": "scaled_nvfp4",
"language_model.model.layers.6.mlp.switch_mlp.down_proj": "scaled_nvfp4_switch",
"language_model.model.layers.6.mlp.switch_mlp.gate_proj": "scaled_nvfp4_switch",
"language_model.model.layers.6.mlp.switch_mlp.up_proj": "scaled_nvfp4_switch",
"language_model.model.layers.7.mlp.shared_expert.down_proj": "scaled_nvfp4",
"language_model.model.layers.7.mlp.shared_expert.gate_proj": "scaled_nvfp4",
"language_model.model.layers.7.mlp.shared_expert.up_proj": "scaled_nvfp4",
"language_model.model.layers.7.mlp.switch_mlp.down_proj": "scaled_nvfp4_switch",
"language_model.model.layers.7.mlp.switch_mlp.gate_proj": "scaled_nvfp4_switch",
"language_model.model.layers.7.mlp.switch_mlp.up_proj": "scaled_nvfp4_switch",
"language_model.model.layers.7.self_attn.k_proj": "scaled_mxfp8",
"language_model.model.layers.7.self_attn.o_proj": "scaled_mxfp8",
"language_model.model.layers.7.self_attn.q_proj": "scaled_mxfp8",
"language_model.model.layers.7.self_attn.v_proj": "scaled_mxfp8",
"language_model.model.layers.8.linear_attn.in_proj_qkv": "scaled_mxfp8",
"language_model.model.layers.8.linear_attn.in_proj_z": "scaled_mxfp8",
"language_model.model.layers.8.linear_attn.out_proj": "scaled_mxfp8",
"language_model.model.layers.8.mlp.shared_expert.down_proj": "scaled_nvfp4",
"language_model.model.layers.8.mlp.shared_expert.gate_proj": "scaled_nvfp4",
"language_model.model.layers.8.mlp.shared_expert.up_proj": "scaled_nvfp4",
"language_model.model.layers.8.mlp.switch_mlp.down_proj": "scaled_nvfp4_switch",
"language_model.model.layers.8.mlp.switch_mlp.gate_proj": "scaled_nvfp4_switch",
"language_model.model.layers.8.mlp.switch_mlp.up_proj": "scaled_nvfp4_switch",
"language_model.model.layers.9.linear_attn.in_proj_qkv": "scaled_mxfp8",
"language_model.model.layers.9.linear_attn.in_proj_z": "scaled_mxfp8",
"language_model.model.layers.9.linear_attn.out_proj": "scaled_mxfp8",
"language_model.model.layers.9.mlp.shared_expert.down_proj": "scaled_nvfp4",
"language_model.model.layers.9.mlp.shared_expert.gate_proj": "scaled_nvfp4",
"language_model.model.layers.9.mlp.shared_expert.up_proj": "scaled_nvfp4",
"language_model.model.layers.9.mlp.switch_mlp.down_proj": "scaled_nvfp4_switch",
"language_model.model.layers.9.mlp.switch_mlp.gate_proj": "scaled_nvfp4_switch",
"language_model.model.layers.9.mlp.switch_mlp.up_proj": "scaled_nvfp4_switch"
},
"model_file": "modeling_mlx_qwen36_modelopt_hybrid.py",
"model_type": "qwen3_5_moe",
"text_config": {
"attention_bias": false,
"attention_dropout": 0.0,
"attn_output_gate": true,
"bos_token_id": 248044,
"dtype": "bfloat16",
"eos_token_id": 248044,
"full_attention_interval": 4,
"head_dim": 256,
"hidden_act": "silu",
"hidden_size": 2048,
"initializer_range": 0.02,
"layer_types": [
"linear_attention",
"linear_attention",
"linear_attention",
"full_attention",
"linear_attention",
"linear_attention",
"linear_attention",
"full_attention",
"linear_attention",
"linear_attention",
"linear_attention",
"full_attention",
"linear_attention",
"linear_attention",
"linear_attention",
"full_attention",
"linear_attention",
"linear_attention",
"linear_attention",
"full_attention",
"linear_attention",
"linear_attention",
"linear_attention",
"full_attention",
"linear_attention",
"linear_attention",
"linear_attention",
"full_attention",
"linear_attention",
"linear_attention",
"linear_attention",
"full_attention",
"linear_attention",
"linear_attention",
"linear_attention",
"full_attention",
"linear_attention",
"linear_attention",
"linear_attention",
"full_attention"
],
"linear_conv_kernel_dim": 4,
"linear_key_head_dim": 128,
"linear_num_key_heads": 16,
"linear_num_value_heads": 32,
"linear_value_head_dim": 128,
"mamba_ssm_dtype": "float32",
"max_position_embeddings": 262144,
"model_type": "qwen3_5_moe_text",
"moe_intermediate_size": 512,
"mtp_num_hidden_layers": 1,
"mtp_use_dedicated_embeddings": false,
"num_attention_heads": 16,
"num_experts": 256,
"num_experts_per_tok": 8,
"num_hidden_layers": 40,
"num_key_value_heads": 2,
"output_router_logits": false,
"pad_token_id": null,
"partial_rotary_factor": 0.25,
"rms_norm_eps": 1e-06,
"rope_parameters": {
"mrope_interleaved": true,
"mrope_section": [
11,
11,
10
],
"partial_rotary_factor": 0.25,
"rope_theta": 10000000,
"rope_type": "default"
},
"router_aux_loss_coef": 0.001,
"shared_expert_intermediate_size": 512,
"tie_word_embeddings": false,
"use_cache": true,
"vocab_size": 248320
},
"tie_word_embeddings": false,
"transformers_version": "5.7.0.dev0"
}