Sync hf_quant_config.json with the thinkingmachines release

hf_quant_config.json

@@ -0,0 +1,578 @@
+{
+  "quantization": {
+    "quant_algo": "NVFP4",
+    "kv_cache_quant_algo": "none",
+    "exclude_modules": [
+      "model.audio.encoder",
+      "model.audio.final_norm",
+      "model.llm.embed",
+      "model.llm.embed_norm",
+      "model.llm.norm",
+      "model.llm.unembed",
+      "model.visual.final_norm",
+      "model.visual.layers.linear_0",
+      "model.visual.layers.linear_1",
+      "model.visual.layers.linear_2",
+      "model.visual.layers.linear_3",
+      "model.visual.layers.norm_0",
+      "model.visual.layers.norm_1",
+      "model.visual.layers.norm_2",
+      "model.llm.layers.0.attn",
+      "model.llm.layers.0.attn_norm",
+      "model.llm.layers.0.attn_sconv",
+      "model.llm.layers.0.mlp.global_scale",
+      "model.llm.layers.0.mlp.w13_dn",
+      "model.llm.layers.0.mlp.w2_md",
+      "model.llm.layers.0.mlp_norm",
+      "model.llm.layers.0.mlp_sconv",
+      "model.llm.layers.1.attn",
+      "model.llm.layers.1.attn_norm",
+      "model.llm.layers.1.attn_sconv",
+      "model.llm.layers.1.mlp.global_scale",
+      "model.llm.layers.1.mlp.w13_dn",
+      "model.llm.layers.1.mlp.w2_md",
+      "model.llm.layers.1.mlp_norm",
+      "model.llm.layers.1.mlp_sconv",
+      "model.llm.layers.10.attn",
+      "model.llm.layers.10.attn_norm",
+      "model.llm.layers.10.attn_sconv",
+      "model.llm.layers.10.mlp.gate",
+      "model.llm.layers.10.mlp.shared_experts",
+      "model.llm.layers.10.mlp_norm",
+      "model.llm.layers.10.mlp_sconv",
+      "model.llm.layers.11.attn",
+      "model.llm.layers.11.attn_norm",
+      "model.llm.layers.11.attn_sconv",
+      "model.llm.layers.11.mlp.gate",
+      "model.llm.layers.11.mlp.shared_experts",
+      "model.llm.layers.11.mlp_norm",
+      "model.llm.layers.11.mlp_sconv",
+      "model.llm.layers.12.attn",
+      "model.llm.layers.12.attn_norm",
+      "model.llm.layers.12.attn_sconv",
+      "model.llm.layers.12.mlp.gate",
+      "model.llm.layers.12.mlp.shared_experts",
+      "model.llm.layers.12.mlp_norm",
+      "model.llm.layers.12.mlp_sconv",
+      "model.llm.layers.13.attn",
+      "model.llm.layers.13.attn_norm",
+      "model.llm.layers.13.attn_sconv",
+      "model.llm.layers.13.mlp.gate",
+      "model.llm.layers.13.mlp.shared_experts",
+      "model.llm.layers.13.mlp_norm",
+      "model.llm.layers.13.mlp_sconv",
+      "model.llm.layers.14.attn",
+      "model.llm.layers.14.attn_norm",
+      "model.llm.layers.14.attn_sconv",
+      "model.llm.layers.14.mlp.gate",
+      "model.llm.layers.14.mlp.shared_experts",
+      "model.llm.layers.14.mlp_norm",
+      "model.llm.layers.14.mlp_sconv",
+      "model.llm.layers.15.attn",
+      "model.llm.layers.15.attn_norm",
+      "model.llm.layers.15.attn_sconv",
+      "model.llm.layers.15.mlp.gate",
+      "model.llm.layers.15.mlp.shared_experts",
+      "model.llm.layers.15.mlp_norm",
+      "model.llm.layers.15.mlp_sconv",
+      "model.llm.layers.16.attn",
+      "model.llm.layers.16.attn_norm",
+      "model.llm.layers.16.attn_sconv",
+      "model.llm.layers.16.mlp.gate",
+      "model.llm.layers.16.mlp.shared_experts",
+      "model.llm.layers.16.mlp_norm",
+      "model.llm.layers.16.mlp_sconv",
+      "model.llm.layers.17.attn",
+      "model.llm.layers.17.attn_norm",
+      "model.llm.layers.17.attn_sconv",
+      "model.llm.layers.17.mlp.gate",
+      "model.llm.layers.17.mlp.shared_experts",
+      "model.llm.layers.17.mlp_norm",
+      "model.llm.layers.17.mlp_sconv",
+      "model.llm.layers.18.attn",
+      "model.llm.layers.18.attn_norm",
+      "model.llm.layers.18.attn_sconv",
+      "model.llm.layers.18.mlp.gate",
+      "model.llm.layers.18.mlp.shared_experts",
+      "model.llm.layers.18.mlp_norm",
+      "model.llm.layers.18.mlp_sconv",
+      "model.llm.layers.19.attn",
+      "model.llm.layers.19.attn_norm",
+      "model.llm.layers.19.attn_sconv",
+      "model.llm.layers.19.mlp.gate",
+      "model.llm.layers.19.mlp.shared_experts",
+      "model.llm.layers.19.mlp_norm",
+      "model.llm.layers.19.mlp_sconv",
+      "model.llm.layers.2.attn",
+      "model.llm.layers.2.attn_norm",
+      "model.llm.layers.2.attn_sconv",
+      "model.llm.layers.2.mlp.experts",
+      "model.llm.layers.2.mlp.gate",
+      "model.llm.layers.2.mlp.shared_experts",
+      "model.llm.layers.2.mlp_norm",
+      "model.llm.layers.2.mlp_sconv",
+      "model.llm.layers.20.attn",
+      "model.llm.layers.20.attn_norm",
+      "model.llm.layers.20.attn_sconv",
+      "model.llm.layers.20.mlp.gate",
+      "model.llm.layers.20.mlp.shared_experts",
+      "model.llm.layers.20.mlp_norm",
+      "model.llm.layers.20.mlp_sconv",
+      "model.llm.layers.21.attn",
+      "model.llm.layers.21.attn_norm",
+      "model.llm.layers.21.attn_sconv",
+      "model.llm.layers.21.mlp.gate",
+      "model.llm.layers.21.mlp.shared_experts",
+      "model.llm.layers.21.mlp_norm",
+      "model.llm.layers.21.mlp_sconv",
+      "model.llm.layers.22.attn",
+      "model.llm.layers.22.attn_norm",
+      "model.llm.layers.22.attn_sconv",
+      "model.llm.layers.22.mlp.gate",
+      "model.llm.layers.22.mlp.shared_experts",
+      "model.llm.layers.22.mlp_norm",
+      "model.llm.layers.22.mlp_sconv",
+      "model.llm.layers.23.attn",
+      "model.llm.layers.23.attn_norm",
+      "model.llm.layers.23.attn_sconv",
+      "model.llm.layers.23.mlp.gate",
+      "model.llm.layers.23.mlp.shared_experts",
+      "model.llm.layers.23.mlp_norm",
+      "model.llm.layers.23.mlp_sconv",
+      "model.llm.layers.24.attn",
+      "model.llm.layers.24.attn_norm",
+      "model.llm.layers.24.attn_sconv",
+      "model.llm.layers.24.mlp.gate",
+      "model.llm.layers.24.mlp.shared_experts",
+      "model.llm.layers.24.mlp_norm",
+      "model.llm.layers.24.mlp_sconv",
+      "model.llm.layers.25.attn",
+      "model.llm.layers.25.attn_norm",
+      "model.llm.layers.25.attn_sconv",
+      "model.llm.layers.25.mlp.gate",
+      "model.llm.layers.25.mlp.shared_experts",
+      "model.llm.layers.25.mlp_norm",
+      "model.llm.layers.25.mlp_sconv",
+      "model.llm.layers.26.attn",
+      "model.llm.layers.26.attn_norm",
+      "model.llm.layers.26.attn_sconv",
+      "model.llm.layers.26.mlp.gate",
+      "model.llm.layers.26.mlp.shared_experts",
+      "model.llm.layers.26.mlp_norm",
+      "model.llm.layers.26.mlp_sconv",
+      "model.llm.layers.27.attn",
+      "model.llm.layers.27.attn_norm",
+      "model.llm.layers.27.attn_sconv",
+      "model.llm.layers.27.mlp.gate",
+      "model.llm.layers.27.mlp.shared_experts",
+      "model.llm.layers.27.mlp_norm",
+      "model.llm.layers.27.mlp_sconv",
+      "model.llm.layers.28.attn",
+      "model.llm.layers.28.attn_norm",
+      "model.llm.layers.28.attn_sconv",
+      "model.llm.layers.28.mlp.gate",
+      "model.llm.layers.28.mlp.shared_experts",
+      "model.llm.layers.28.mlp_norm",
+      "model.llm.layers.28.mlp_sconv",
+      "model.llm.layers.29.attn",
+      "model.llm.layers.29.attn_norm",
+      "model.llm.layers.29.attn_sconv",
+      "model.llm.layers.29.mlp.gate",
+      "model.llm.layers.29.mlp.shared_experts",
+      "model.llm.layers.29.mlp_norm",
+      "model.llm.layers.29.mlp_sconv",
+      "model.llm.layers.3.attn",
+      "model.llm.layers.3.attn_norm",
+      "model.llm.layers.3.attn_sconv",
+      "model.llm.layers.3.mlp.gate",
+      "model.llm.layers.3.mlp.shared_experts",
+      "model.llm.layers.3.mlp_norm",
+      "model.llm.layers.3.mlp_sconv",
+      "model.llm.layers.30.attn",
+      "model.llm.layers.30.attn_norm",
+      "model.llm.layers.30.attn_sconv",
+      "model.llm.layers.30.mlp.gate",
+      "model.llm.layers.30.mlp.shared_experts",
+      "model.llm.layers.30.mlp_norm",
+      "model.llm.layers.30.mlp_sconv",
+      "model.llm.layers.31.attn",
+      "model.llm.layers.31.attn_norm",
+      "model.llm.layers.31.attn_sconv",
+      "model.llm.layers.31.mlp.gate",
+      "model.llm.layers.31.mlp.shared_experts",
+      "model.llm.layers.31.mlp_norm",
+      "model.llm.layers.31.mlp_sconv",
+      "model.llm.layers.32.attn",
+      "model.llm.layers.32.attn_norm",
+      "model.llm.layers.32.attn_sconv",
+      "model.llm.layers.32.mlp.gate",
+      "model.llm.layers.32.mlp.shared_experts",
+      "model.llm.layers.32.mlp_norm",
+      "model.llm.layers.32.mlp_sconv",
+      "model.llm.layers.33.attn",
+      "model.llm.layers.33.attn_norm",
+      "model.llm.layers.33.attn_sconv",
+      "model.llm.layers.33.mlp.gate",
+      "model.llm.layers.33.mlp.shared_experts",
+      "model.llm.layers.33.mlp_norm",
+      "model.llm.layers.33.mlp_sconv",
+      "model.llm.layers.34.attn",
+      "model.llm.layers.34.attn_norm",
+      "model.llm.layers.34.attn_sconv",
+      "model.llm.layers.34.mlp.gate",
+      "model.llm.layers.34.mlp.shared_experts",
+      "model.llm.layers.34.mlp_norm",
+      "model.llm.layers.34.mlp_sconv",
+      "model.llm.layers.35.attn",
+      "model.llm.layers.35.attn_norm",
+      "model.llm.layers.35.attn_sconv",
+      "model.llm.layers.35.mlp.gate",
+      "model.llm.layers.35.mlp.shared_experts",
+      "model.llm.layers.35.mlp_norm",
+      "model.llm.layers.35.mlp_sconv",
+      "model.llm.layers.36.attn",
+      "model.llm.layers.36.attn_norm",
+      "model.llm.layers.36.attn_sconv",
+      "model.llm.layers.36.mlp.gate",
+      "model.llm.layers.36.mlp.shared_experts",
+      "model.llm.layers.36.mlp_norm",
+      "model.llm.layers.36.mlp_sconv",
+      "model.llm.layers.37.attn",
+      "model.llm.layers.37.attn_norm",
+      "model.llm.layers.37.attn_sconv",
+      "model.llm.layers.37.mlp.gate",
+      "model.llm.layers.37.mlp.shared_experts",
+      "model.llm.layers.37.mlp_norm",
+      "model.llm.layers.37.mlp_sconv",
+      "model.llm.layers.38.attn",
+      "model.llm.layers.38.attn_norm",
+      "model.llm.layers.38.attn_sconv",
+      "model.llm.layers.38.mlp.gate",
+      "model.llm.layers.38.mlp.shared_experts",
+      "model.llm.layers.38.mlp_norm",
+      "model.llm.layers.38.mlp_sconv",
+      "model.llm.layers.39.attn",
+      "model.llm.layers.39.attn_norm",
+      "model.llm.layers.39.attn_sconv",
+      "model.llm.layers.39.mlp.gate",
+      "model.llm.layers.39.mlp.shared_experts",
+      "model.llm.layers.39.mlp_norm",
+      "model.llm.layers.39.mlp_sconv",
+      "model.llm.layers.4.attn",
+      "model.llm.layers.4.attn_norm",
+      "model.llm.layers.4.attn_sconv",
+      "model.llm.layers.4.mlp.gate",
+      "model.llm.layers.4.mlp.shared_experts",
+      "model.llm.layers.4.mlp_norm",
+      "model.llm.layers.4.mlp_sconv",
+      "model.llm.layers.40.attn",
+      "model.llm.layers.40.attn_norm",
+      "model.llm.layers.40.attn_sconv",
+      "model.llm.layers.40.mlp.gate",
+      "model.llm.layers.40.mlp.shared_experts",
+      "model.llm.layers.40.mlp_norm",
+      "model.llm.layers.40.mlp_sconv",
+      "model.llm.layers.41.attn",
+      "model.llm.layers.41.attn_norm",
+      "model.llm.layers.41.attn_sconv",
+      "model.llm.layers.41.mlp.gate",
+      "model.llm.layers.41.mlp.shared_experts",
+      "model.llm.layers.41.mlp_norm",
+      "model.llm.layers.41.mlp_sconv",
+      "model.llm.layers.42.attn",
+      "model.llm.layers.42.attn_norm",
+      "model.llm.layers.42.attn_sconv",
+      "model.llm.layers.42.mlp.gate",
+      "model.llm.layers.42.mlp.shared_experts",
+      "model.llm.layers.42.mlp_norm",
+      "model.llm.layers.42.mlp_sconv",
+      "model.llm.layers.43.attn",
+      "model.llm.layers.43.attn_norm",
+      "model.llm.layers.43.attn_sconv",
+      "model.llm.layers.43.mlp.gate",
+      "model.llm.layers.43.mlp.shared_experts",
+      "model.llm.layers.43.mlp_norm",
+      "model.llm.layers.43.mlp_sconv",
+      "model.llm.layers.44.attn",
+      "model.llm.layers.44.attn_norm",
+      "model.llm.layers.44.attn_sconv",
+      "model.llm.layers.44.mlp.gate",
+      "model.llm.layers.44.mlp.shared_experts",
+      "model.llm.layers.44.mlp_norm",
+      "model.llm.layers.44.mlp_sconv",
+      "model.llm.layers.45.attn",
+      "model.llm.layers.45.attn_norm",
+      "model.llm.layers.45.attn_sconv",
+      "model.llm.layers.45.mlp.gate",
+      "model.llm.layers.45.mlp.shared_experts",
+      "model.llm.layers.45.mlp_norm",
+      "model.llm.layers.45.mlp_sconv",
+      "model.llm.layers.46.attn",
+      "model.llm.layers.46.attn_norm",
+      "model.llm.layers.46.attn_sconv",
+      "model.llm.layers.46.mlp.gate",
+      "model.llm.layers.46.mlp.shared_experts",
+      "model.llm.layers.46.mlp_norm",
+      "model.llm.layers.46.mlp_sconv",
+      "model.llm.layers.47.attn",
+      "model.llm.layers.47.attn_norm",
+      "model.llm.layers.47.attn_sconv",
+      "model.llm.layers.47.mlp.gate",
+      "model.llm.layers.47.mlp.shared_experts",
+      "model.llm.layers.47.mlp_norm",
+      "model.llm.layers.47.mlp_sconv",
+      "model.llm.layers.48.attn",
+      "model.llm.layers.48.attn_norm",
+      "model.llm.layers.48.attn_sconv",
+      "model.llm.layers.48.mlp.gate",
+      "model.llm.layers.48.mlp.shared_experts",
+      "model.llm.layers.48.mlp_norm",
+      "model.llm.layers.48.mlp_sconv",
+      "model.llm.layers.49.attn",
+      "model.llm.layers.49.attn_norm",
+      "model.llm.layers.49.attn_sconv",
+      "model.llm.layers.49.mlp.gate",
+      "model.llm.layers.49.mlp.shared_experts",
+      "model.llm.layers.49.mlp_norm",
+      "model.llm.layers.49.mlp_sconv",
+      "model.llm.layers.5.attn",
+      "model.llm.layers.5.attn_norm",
+      "model.llm.layers.5.attn_sconv",
+      "model.llm.layers.5.mlp.gate",
+      "model.llm.layers.5.mlp.shared_experts",
+      "model.llm.layers.5.mlp_norm",
+      "model.llm.layers.5.mlp_sconv",
+      "model.llm.layers.50.attn",
+      "model.llm.layers.50.attn_norm",
+      "model.llm.layers.50.attn_sconv",
+      "model.llm.layers.50.mlp.gate",
+      "model.llm.layers.50.mlp.shared_experts",
+      "model.llm.layers.50.mlp_norm",
+      "model.llm.layers.50.mlp_sconv",
+      "model.llm.layers.51.attn",
+      "model.llm.layers.51.attn_norm",
+      "model.llm.layers.51.attn_sconv",
+      "model.llm.layers.51.mlp.gate",
+      "model.llm.layers.51.mlp.shared_experts",
+      "model.llm.layers.51.mlp_norm",
+      "model.llm.layers.51.mlp_sconv",
+      "model.llm.layers.52.attn",
+      "model.llm.layers.52.attn_norm",
+      "model.llm.layers.52.attn_sconv",
+      "model.llm.layers.52.mlp.gate",
+      "model.llm.layers.52.mlp.shared_experts",
+      "model.llm.layers.52.mlp_norm",
+      "model.llm.layers.52.mlp_sconv",
+      "model.llm.layers.53.attn",
+      "model.llm.layers.53.attn_norm",
+      "model.llm.layers.53.attn_sconv",
+      "model.llm.layers.53.mlp.gate",
+      "model.llm.layers.53.mlp.shared_experts",
+      "model.llm.layers.53.mlp_norm",
+      "model.llm.layers.53.mlp_sconv",
+      "model.llm.layers.54.attn",
+      "model.llm.layers.54.attn_norm",
+      "model.llm.layers.54.attn_sconv",
+      "model.llm.layers.54.mlp.gate",
+      "model.llm.layers.54.mlp.shared_experts",
+      "model.llm.layers.54.mlp_norm",
+      "model.llm.layers.54.mlp_sconv",
+      "model.llm.layers.55.attn",
+      "model.llm.layers.55.attn_norm",
+      "model.llm.layers.55.attn_sconv",
+      "model.llm.layers.55.mlp.gate",
+      "model.llm.layers.55.mlp.shared_experts",
+      "model.llm.layers.55.mlp_norm",
+      "model.llm.layers.55.mlp_sconv",
+      "model.llm.layers.56.attn",
+      "model.llm.layers.56.attn_norm",
+      "model.llm.layers.56.attn_sconv",
+      "model.llm.layers.56.mlp.gate",
+      "model.llm.layers.56.mlp.shared_experts",
+      "model.llm.layers.56.mlp_norm",
+      "model.llm.layers.56.mlp_sconv",
+      "model.llm.layers.57.attn",
+      "model.llm.layers.57.attn_norm",
+      "model.llm.layers.57.attn_sconv",
+      "model.llm.layers.57.mlp.gate",
+      "model.llm.layers.57.mlp.shared_experts",
+      "model.llm.layers.57.mlp_norm",
+      "model.llm.layers.57.mlp_sconv",
+      "model.llm.layers.58.attn",
+      "model.llm.layers.58.attn_norm",
+      "model.llm.layers.58.attn_sconv",
+      "model.llm.layers.58.mlp.gate",
+      "model.llm.layers.58.mlp.shared_experts",
+      "model.llm.layers.58.mlp_norm",
+      "model.llm.layers.58.mlp_sconv",
+      "model.llm.layers.59.attn",
+      "model.llm.layers.59.attn_norm",
+      "model.llm.layers.59.attn_sconv",
+      "model.llm.layers.59.mlp.gate",
+      "model.llm.layers.59.mlp.shared_experts",
+      "model.llm.layers.59.mlp_norm",
+      "model.llm.layers.59.mlp_sconv",
+      "model.llm.layers.6.attn",
+      "model.llm.layers.6.attn_norm",
+      "model.llm.layers.6.attn_sconv",
+      "model.llm.layers.6.mlp.gate",
+      "model.llm.layers.6.mlp.shared_experts",
+      "model.llm.layers.6.mlp_norm",
+      "model.llm.layers.6.mlp_sconv",
+      "model.llm.layers.60.attn",
+      "model.llm.layers.60.attn_norm",
+      "model.llm.layers.60.attn_sconv",
+      "model.llm.layers.60.mlp.gate",
+      "model.llm.layers.60.mlp.shared_experts",
+      "model.llm.layers.60.mlp_norm",
+      "model.llm.layers.60.mlp_sconv",
+      "model.llm.layers.61.attn",
+      "model.llm.layers.61.attn_norm",
+      "model.llm.layers.61.attn_sconv",
+      "model.llm.layers.61.mlp.gate",
+      "model.llm.layers.61.mlp.shared_experts",
+      "model.llm.layers.61.mlp_norm",
+      "model.llm.layers.61.mlp_sconv",
+      "model.llm.layers.62.attn",
+      "model.llm.layers.62.attn_norm",
+      "model.llm.layers.62.attn_sconv",
+      "model.llm.layers.62.mlp.gate",
+      "model.llm.layers.62.mlp.shared_experts",
+      "model.llm.layers.62.mlp_norm",
+      "model.llm.layers.62.mlp_sconv",
+      "model.llm.layers.63.attn",
+      "model.llm.layers.63.attn_norm",
+      "model.llm.layers.63.attn_sconv",
+      "model.llm.layers.63.mlp.gate",
+      "model.llm.layers.63.mlp.shared_experts",
+      "model.llm.layers.63.mlp_norm",
+      "model.llm.layers.63.mlp_sconv",
+      "model.llm.layers.64.attn",
+      "model.llm.layers.64.attn_norm",
+      "model.llm.layers.64.attn_sconv",
+      "model.llm.layers.64.mlp.gate",
+      "model.llm.layers.64.mlp.shared_experts",
+      "model.llm.layers.64.mlp_norm",
+      "model.llm.layers.64.mlp_sconv",
+      "model.llm.layers.65.attn",
+      "model.llm.layers.65.attn_norm",
+      "model.llm.layers.65.attn_sconv",
+      "model.llm.layers.65.mlp.gate",
+      "model.llm.layers.65.mlp.shared_experts",
+      "model.llm.layers.65.mlp_norm",
+      "model.llm.layers.65.mlp_sconv",
+      "model.llm.layers.7.attn",
+      "model.llm.layers.7.attn_norm",
+      "model.llm.layers.7.attn_sconv",
+      "model.llm.layers.7.mlp.gate",
+      "model.llm.layers.7.mlp.shared_experts",
+      "model.llm.layers.7.mlp_norm",
+      "model.llm.layers.7.mlp_sconv",
+      "model.llm.layers.8.attn",
+      "model.llm.layers.8.attn_norm",
+      "model.llm.layers.8.attn_sconv",
+      "model.llm.layers.8.mlp.gate",
+      "model.llm.layers.8.mlp.shared_experts",
+      "model.llm.layers.8.mlp_norm",
+      "model.llm.layers.8.mlp_sconv",
+      "model.llm.layers.9.attn",
+      "model.llm.layers.9.attn_norm",
+      "model.llm.layers.9.attn_sconv",
+      "model.llm.layers.9.mlp.gate",
+      "model.llm.layers.9.mlp.shared_experts",
+      "model.llm.layers.9.mlp_norm",
+      "model.llm.layers.9.mlp_sconv"
+    ],
+    "modelopt_quant_config": {
+      "quant_cfg": {
+        "*weight_quantizer": {
+          "num_bits": [
+            2,
+            1
+          ],
+          "block_sizes": {
+            "-1": 16,
+            "type": "dynamic",
+            "scale_bits": [
+              4,
+              3
+            ]
+          },
+          "axis": null,
+          "enable": true
+        },
+        "*input_quantizer": {
+          "num_bits": [
+            2,
+            1
+          ],
+          "block_sizes": {
+            "-1": 16,
+            "type": "dynamic",
+            "scale_bits": [
+              4,
+              3
+            ]
+          },
+          "axis": null,
+          "enable": true
+        },
+        "nn.BatchNorm1d": {
+          "*": {
+            "enable": false
+          }
+        },
+        "nn.BatchNorm2d": {
+          "*": {
+            "enable": false
+          }
+        },
+        "nn.BatchNorm3d": {
+          "*": {
+            "enable": false
+          }
+        },
+        "nn.LeakyReLU": {
+          "*": {
+            "enable": false
+          }
+        },
+        "*lm_head*": {
+          "enable": false
+        },
+        "*proj_out.*": {
+          "enable": false
+        },
+        "*block_sparse_moe.gate*": {
+          "enable": false
+        },
+        "*router*": {
+          "enable": false
+        },
+        "*mlp.gate.*": {
+          "enable": false
+        },
+        "*mlp.shared_expert_gate.*": {
+          "enable": false
+        },
+        "*linear_attn.conv1d*": {
+          "enable": false
+        },
+        "*mixer.conv1d*": {
+          "enable": false
+        },
+        "*output_layer*": {
+          "enable": false
+        },
+        "output.*": {
+          "enable": false
+        },
+        "default": {
+          "enable": false
+        }
+      },
+      "algorithm": "max"
+    },
+    "group_size": 16
+  }
+}