k-l-lambda commited on
Commit
bab7ecd
·
verified ·
1 Parent(s): 9b80167

best158 (step95400, K2.6 val-best): overwrite ckpt100; README updated with real accept_len + distribution-shift note

Browse files
Files changed (2) hide show
  1. README.md +0 -54
  2. model.safetensors +1 -1
README.md CHANGED
@@ -1,54 +0,0 @@
1
- ---
2
- license: other
3
- base_model: moonshotai/Kimi-K2.6
4
- tags:
5
- - text-generation
6
- - speculative-decoding
7
- - eagle3
8
- - kimi-k2.6
9
- - mla
10
- - torchspec
11
- ---
12
-
13
- # kimi-k2.6-eagle3-mla
14
-
15
- Eagle3 MTP draft model with MLA (Multi-Latent Attention) for accelerating
16
- inference of [Kimi-K2.6](https://huggingface.co/moonshotai/Kimi-K2.6).
17
-
18
- This is a fine-tuned draft, anchored to the official
19
- [lightseekorg/kimi-k2.6-eagle3-mla](https://huggingface.co/lightseekorg/kimi-k2.6-eagle3-mla)
20
- initialization. It targets multi-hop (downstream-position) acceptance while
21
- preserving the first-hop gain, evaluated by runtime accept-length on a frozen
22
- full-context held-out set.
23
-
24
- ## Fine-tune setup
25
-
26
- - **Init**: lightseekorg/kimi-k2.6-eagle3-mla (official MLA weights)
27
- - **Objective**: Eagle3 distillation + multi-step TTT supervision
28
- (`ttt_steps=4`, `ttt_step_loss_decay=1.0`, off-policy downstream tokens)
29
- - **Anti-over-specialization**: L2-SP weight-space anchor toward the init
30
- (penalize trainable-param drift; lambda=1e-4)
31
- - **Optimizer**: lr 2e-5, cosine schedule
32
- - **Checkpoint**: best by held-out runtime accept-length
33
-
34
- ## Performance
35
-
36
- Primary metric is **accept_length** — average tokens accepted per speculation
37
- step with `num_speculative_tokens=3` (higher is better). Per-position numbers
38
- are conditional acceptance rates at hop 0/1/2. Evaluated on a frozen
39
- full-context held-out judge set (912 prompts, greedy), vLLM 0.20.0, 8x H200,
40
- TP=8, max-model-len 32768.
41
-
42
- | Model | accept_len | pos-0 | pos-1 | pos-2 |
43
- |-------|-----------:|------:|------:|------:|
44
- | lightseek (official init) | 2.30 | 0.633 | 0.404 | 0.264 |
45
- | this model | **2.345** | **0.648** | **0.419** | **0.278** |
46
-
47
- This draft improves first-hop acceptance over the official init while also
48
- lifting the downstream positions (pos-1, pos-2), yielding a higher overall
49
- accept length.
50
-
51
- ## Usage
52
-
53
- Serve with vLLM as the speculative draft for Kimi-K2.6, with
54
- `num_speculative_tokens=3` in the speculative-config.
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:57cf1a3b1cd6153619f37a42410e6123fd899f15d2f95d230dafe68399825d04
3
  size 6031210296
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:5b5829babeb37d5b1f982b3f6f759e45370f82e5d7b7938398e3a48ecb5e32c4
3
  size 6031210296