palmfuture commited on
Commit
f32841f
·
verified ·
1 Parent(s): bddd3f8

Add MTP weights + speculative decoding (backward compatible)

Browse files

- mtp.safetensors: 19 MTP tensors in per-expert split format
(768 expert keys + 17 non-expert, BF16, +1.6 GB)
- model.safetensors.index.json: weight_map updated with 19 MTP keys
(total_size now 24.4 GB)
- config.json: add mlp_only_layers, modules_to_not_convert
(Qwen official schema parity)
- README: document vLLM + SGLang MTP configs with verified metrics

Verified on 4× RTX 3060 12GB:
- vLLM 0.19.1 + MTP: 56-82 t/s, 70-89% acceptance
- vLLM 0.19.1 baseline: works (no regression)
- SGLang 0.5.10 + EAGLE: 52-141 t/s (peak batch), 34-70% acceptance
- SGLang 0.5.10 baseline: works (no regression)

vLLM users with MTP: do NOT pass --quantization moe_wna16 flag
(triggers KeyError in MTP loader; auto-detect required).

Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -47,7 +47,7 @@ This v2 release ships **MTP (Multi-Token Prediction) speculative decoding weight
47
  | Hidden layers | 40 |
48
  | Context length | 262,144 tokens |
49
  | Quantization | GPTQ v2, 4-bit, group_size=128, symmetric |
50
- | Quantized size | 22.7 GB (incl. MTP weights) |
51
  | KV cache support | fp16, bf16, fp8_e4m3 (storage-only on Ampere) |
52
  | MTP head | Included (BF16, 785 keys, split per-expert format) |
53
 
 
47
  | Hidden layers | 40 |
48
  | Context length | 262,144 tokens |
49
  | Quantization | GPTQ v2, 4-bit, group_size=128, symmetric |
50
+ | Quantized size | 24.4 GB (incl. MTP weights) |
51
  | KV cache support | fp16, bf16, fp8_e4m3 (storage-only on Ampere) |
52
  | MTP head | Included (BF16, 785 keys, split per-expert format) |
53