Hybrid Attention Models#
Some models interleave more than one attention type across their layers — most commonly sliding-window attention on most layers and full attention on a few. vLLM serves these with its hybrid KV cache manager, which splits the model’s layers into multiple KV cache groups (one per attention behavior).
The LMCache multiprocess connector (LMCacheMPConnector) supports these
hybrid models: it stores and retrieves the KV cache for every group, so prefix
caching and KV reuse work the same way they do for plain models.
Validated hybrid models#
Recipe pages for the validated hybrid-attention architectures:
Model |
Attention layout |
Recipe |
|---|---|---|
Gemma 3 |
Sliding-window + full |
|
Gemma 4 |
Sliding-window + full |
|
gpt-oss |
Sliding-window + full |
|
Qwen3.5 / Qwen3.6 / Qwen3.8 series |
Mamba / GDN + full |
|
Qwen3.8-Flash-Next |
GDN linear-attention + QSA sparse full (scratch ring) |
|
Kimi-Linear |
KDA linear-attention + MLA full |
|
Kimi K3 |
KDA linear-attention + MLA full |
|
DeepSeek-V4-Flash |
Sparse-MLA (multiple KV groups) |
|
DeepSeek-V4.1-Flash |
Sparse-MLA + per-layer sliding window (9 KV groups, scratch ring) |
|
GLM 5.1/5.2 |
Dynamic Sparse Attention (multiple KV groups) |
|
GLM-5.3 |
Dynamic Sparse Attention (single KV group) |
|
MiniMax-M3 |
Sparse attention + lightning indexer (mixed KV formats in one group) |
What Works#
Models whose layers all use standard paged attention — including hybrids that mix sliding-window and full attention — are supported with no special configuration. Examples:
Model family |
Attention layout |
Status |
|---|---|---|
Gemma 2 / Gemma 3 |
Interleaved sliding-window + full |
Supported |
gpt-oss |
Interleaved sliding-window + full |
Supported |
Qwen3.5 (and other Gated-DeltaNet hybrids) |
Interleaved Mamba/GDN + full |
Supported (see below) |
Llama, Qwen2/Qwen3 (dense), Mistral, … |
Single attention type |
Supported |
Just point vLLM at the LMCache server as usual (see Quickstart); LMCache detects the model’s KV cache groups automatically at registration time.
Per-request scratch groups are excluded from caching, including on older vLLM
forks that mark them with participates_in_prefix_caching=False instead of
prefix_cacheable=False.
Note
Because LMCacheMPConnector advertises hybrid support to vLLM, vLLM keeps
its hybrid KV cache manager enabled for these models (it does not fall
back to a single unified group). You do not need
--no-disable-hybrid-kv-cache-manager or any related flag.
Object-group separation#
At KV-cache registration LMCache can bucket a hybrid model’s layers into
object groups — the unit it stores and retrieves as one object. With
--separate-object-groups each distinct cross-chunk attention window becomes
its own object group: full-attention layers form one group, and each
sliding-window size (mamba / GDN included) forms another. The default is
off — every layer shares a single full-attention object group.
# default: a single full-attention object group for all layers
lmcache server --chunk-size 256 --l1-size-gb 100
# one object group per attention window (required for mamba / GDN hybrids)
lmcache server --chunk-size 256 --l1-size-gb 100 --separate-object-groups
For a non-hybrid model (a single attention behavior) the setting makes no difference — every layer resolves to one object group either way. For a Mamba / linear-attention hybrid it is required (see below): it gives the linear-attention layers their own cache objects so their recurrent state is stored and loaded independently of the full-attention layers.
Mamba / Linear-Attention Hybrids#
Models that interleave Mamba / Gated-DeltaNet (GDN) linear-attention layers
with full attention — the Qwen3.5, Qwen3.6 and Qwen3.8 series
(Qwen/Qwen3.5-0.8B, Qwen/Qwen3.6-27B, Qwen/Qwen3.8-27B, …),
Qwen3-Next, Kimi-Linear
(moonshotai/Kimi-Linear-48B-A3B-Instruct), Kimi K3
(moonshotai/Kimi-K3), and other GDN hybrids — are supported.
Unlike a paged key/value cache, their linear-attention layers keep a recurrent
state cache (a convolution + SSM state). LMCache reinterprets that state as
an opaque page at registration time, so prefix caching and KV reuse work end to
end without any model-specific transfer code.
This section is the general procedure for any such model; the settings are identical across models.
Unified block size and chunk size#
For a Mamba / GDN hybrid, vLLM forces one block size N across all KV
cache groups, chosen large enough that an attention page is at least as big as
a Mamba state page. It depends on the model’s head dimensions and GDN state
size (and, when the recurrent state is sharded, on the tensor-parallel size),
so it is model-specific. vLLM prints it once at startup:
INFO ... interface.py:670] Setting attention block size to 784 tokens to
ensure that attention page size is >= mamba page size.
Nothing has to be derived from N by hand: the LMCache server’s
--chunk-size is a minimum, and the connector negotiates the smallest
multiple of N that is at least that value (N = 784 resolves the default
minimum of 256 to 784, while N = 544 resolves it to 544). See
Chunk-Size Negotiation for the full lifecycle and mixed-model
constraints. For reference, values grow with model size:
Model |
Unified block size |
GPUs |
|---|---|---|
|
784 |
1 |
|
784 |
1 |
|
544 |
1 |
|
944 |
2 |
|
768 |
8 |
Required flags#
LMCache server
--separate-object-groups— gives the linear-attention layers their own cache objects so their recurrent state is stored and loaded independently of the full-attention layers.--chunk-sizecan stay at its default; the connector negotiates it up to a multiple ofN:lmcache server --chunk-size 256 --separate-object-groups \ --l1-size-gb 100 --eviction-policy LRUvLLM
--mamba-cache-mode align --enable-prefix-caching—alignis mandatory (GDN backends do not support theallmode):vllm serve <model> \ --enable-prefix-caching --mamba-cache-mode align \ --kv-transfer-config \ '{"kv_connector":"LMCacheMPConnector", "kv_role":"kv_both"}'
No --no-disable-hybrid-kv-cache-manager or attention-backend flag is needed;
LMCacheMPConnector advertises hybrid support and vLLM auto-selects the GDN
backend.
Caveats#
Generation is not bit-exact between a cached and a fresh run: GDN backends do not support vLLM’s batch-invariant mode. Validate with a score-level comparison (see Verifying Correctness), not a token-level diff.
The cached pages are byte-opaque, so content-aware features (CacheGen compression, CacheBlend) do not apply, and cache entries must not be shared across engines with different attention backends or kernel block sizes.
Several of these models are vision-language (they load a vision tower). The validated, supported path is text KV caching; image/video KV caching is not validated.
vLLM’s Mamba prefix caching in
alignmode is marked experimental upstream.
See the Qwen3.5 / Qwen3.6 / Qwen3.8 recipe for the validated end-to-end commands and the per-model block sizes.
Verifying Correctness#
To convince yourself that a hybrid model’s KV is being cached and reused correctly, you can compare a cold run against a run served from LMCache:
Run an evaluation (e.g.
lm_evalongsm8k) against vLLM + LMCache. This computes the KV cache and stores it in LMCache.Reset only vLLM’s local prefix cache, leaving the LMCache-managed cache intact (requires launching vLLM with
VLLM_SERVER_DEV_MODE=1):curl -X POST http://localhost:8000/reset_prefix_cacheOmit the
reset_external=truequery parameter so the LMCache cache is preserved.Re-run the same evaluation. vLLM now misses in its local cache, so the prefix KV is retrieved from LMCache. The score should match the first run.
The project ships this as the hma_lm_eval continuous-integration test (see
.buildkite/k3_tests/multiprocess).
See Also#
Quickstart — launching the LMCache server and a vLLM client.
Design notes on how groups are detected and addressed:
docs/design/integration/vllm/hybrid-kv-cache-groups.mdin the source tree.