Merge branch 'main' into patches
Signed-off-by: MOVZX <movzx@yahoo.com>
This commit is contained in:
@@ -74,3 +74,12 @@ vllm serve XiaomiMiMo/MiMo-7B-Base \
|
||||
is a good default to start with.
|
||||
- If your model does not support MTP, use another method such as EAGLE or draft
|
||||
model speculation.
|
||||
- When the MTP head shares the target's unquantized lm_head (most MTP models),
|
||||
`"draft_lm_head_quantization": "fp8"` or `"nvfp4"` in `--speculative-config`
|
||||
gives the drafter a quantized copy of that head (Model Runner V2). The target
|
||||
still verifies with its own head, so outputs are unchanged; only the
|
||||
acceptance length can move. It helps most on bandwidth-bound GPUs with a
|
||||
large vocabulary. The copy costs about one byte (FP8) or 0.56 bytes (NVFP4)
|
||||
per lm_head weight, e.g. 0.6 or 0.35 GiB for a 248k x 2560 head; it is
|
||||
counted in the model weights, so with `--kv-cache-memory-bytes` leave room
|
||||
for it.
|
||||
|
||||
@@ -125,8 +125,10 @@ generation.
|
||||
## Speculative decoding
|
||||
|
||||
Watermarking requires speculative decoding to use probabilistic draft sampling,
|
||||
standard rejection sampling, and an autoregressive model-based method (`dspark`,
|
||||
`eagle`, `eagle3`, or `mtp`). Parallel drafting is supported only by `dspark`.
|
||||
standard rejection sampling, and a model-based method (`dflash`, `dspark`,
|
||||
`eagle`, `eagle3`, or `mtp`). Parallel drafting is supported only by `dflash`
|
||||
and `dspark`. DFlash drafters with a candidate head (`DFlash2DraftModel`,
|
||||
`LiLiCorrDraftModel`) are not supported.
|
||||
|
||||
A watermarking algorithm without native speculative-decoding support is
|
||||
rejected before model loading. Set
|
||||
|
||||
Reference in New Issue
Block a user