{ "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" }