ZixiQi commited on
Commit
c7b1d3b
·
verified ·
1 Parent(s): 4a97a1b

Trim Performance section to MT-Bench + SPEED-Bench acceptance; drop MHA comparison

Browse files
Files changed (1) hide show
  1. README.md +6 -12
README.md CHANGED
@@ -25,15 +25,9 @@ The draft is a 1-layer dense Llama (`LlamaForCausalLMEagle3`) on MiniMax-M3's `h
25
 
26
  ## Performance
27
 
28
- Mean accepted length measured end-to-end against `MiniMaxAI/MiniMax-M3-MXFP8` served with vLLM at `tensor-parallel-size=4`, `num_speculative_tokens=3`, `--enforce-eager`, greedy draft sampling (`topk=1`). The MHA version is shown for reference under identical conditions.
29
-
30
- | Category | Dataset | n | GQA (this model) | MHA reference |
31
- |---|---|---:|---:|---:|
32
- | Dialogue | [MT-Bench](https://github.com/lm-sys/FastChat/tree/main/fastchat/llm_judge) | 80 | 2.689 | 2.698 |
33
- | Math | [GSM8K](https://github.com/openai/grade-school-math) | 200 | 3.500 | 3.518 |
34
- | Code | [HumanEval](https://huggingface.co/datasets/openai/openai_humaneval) | 164 | 3.489 | 3.499 |
35
- | Math | [MATH500](https://huggingface.co/datasets/HuggingFaceH4/MATH-500) | 500 | 3.510 | 3.517 |
36
- | Math | [AIME](https://huggingface.co/datasets/Maxwell-Jia/AIME_2024) | 30 | 3.282 | 3.291 |
37
- | | **Mean** | | **3.294** | 3.305 |
38
-
39
- GQA matches the MHA version's acceptance (**~99.7%** of mean accepted length) while using **16× fewer** key/value heads.
 
25
 
26
  ## Performance
27
 
28
+ Mean accepted length and draft accept rate measured end-to-end against `MiniMaxAI/MiniMax-M3-MXFP8` served with vLLM at `tensor-parallel-size=4`, `num_speculative_tokens=3`, greedy sampling (`temperature=0`, `top_p=1.0`), `max-concurrency=16`.
29
+
30
+ | Dataset | n | Mean accepted length | Draft accept rate |
31
+ |---|---:|---:|---:|
32
+ | MT-Bench | 64 | 2.668 | 55.62% |
33
+ | SPEED-Bench (qualitative) | 64 | 2.561 | 52.04% |