Expand description
2026-09-25: MoE and MTP-head weight loaders: the mlp MoE block, the Mistral w1/w2/w3 MoE block, and the MTP head.
Owner: model-layers (weight loading). Invariants:
MtpWeights::fp8_expertsisSomeonly when the experts are unstacked, the variant is notBf16Raw, and expert 0 is FP8 withweight_scale_inv.
Functionsยง
- load_
moe_ mistral - 2026-09-25: Load a Mistral MoE block (
w1/w2/w3naming: w1 = gate_proj, w2 = down_proj, w3 = up_proj) throughquantized_v2. - load_
mtp - 2026-09-25: Load the MTP head (
mtp.*keys) as BF16DenseWeights.