Views
No views yet
undeadindustries/gemma-4-31b-it-dflash-drafter-nvfp4
for optional speculative-decoding speedup on vLLM.| Role | Repo | Size | Required? |
|---|---|---|---|
| Verifier (this repo) | gemma-4-31b-it-nvfp4-hybrid | ~30 GB | Yes — this is the 31B brain |
| DFlash drafter | gemma-4-31b-it-dflash-drafter-nvfp4 | ~8 GB | Optional — guesses tokens; verifier always has final say |
vllm serve model and point --speculative-config at the drafter. The drafter was trained on hidden states from this exact quant — a generic drafter trained on BF16 Gemma will not work well (acceptance collapses).google/gemma-4-31b-it for instruction following and long-document work on memory-constrained Blackwell boxes (e.g. 128 GB unified memory):q/k/v/o) — protects dynamic range for 256K-class reasoning and needle-in-haystack retrievalgate/up/down) — ~2× weight compression on the bulk of parametersGemma4ForCausalLM architecture-it instruction behavior, not maximum compression at any quality cost.| Component | Precision |
|---|---|
| Attention projections | BF16 (all 60 layers; v_proj on 50 sliding layers only — Gemma 4 global layers use shared K/V) |
MLP (gate/up/down) | NVFP4 W4A4, group_size=16 |
| KV cache (recommended) | BF16 — avoid FP8 KV default on Gemma 4 hybrid SWA |
scheme=NVFP4, exported as llm-compressor / nvfp4-pack-quantized for vLLM./v1/chat/completions (required for -it models):FlashInferCutlassNvFp4LinearKernel)1vllm serve undeadindustries/gemma-4-31b-it-nvfp4-hybrid \
2 --port 8000 --max-model-len 4096 \
3 --gpu-memory-utilization 0.55 --enforce-eager --trust-remote-code \
4 --generation-config vllm1vllm serve undeadindustries/gemma-4-31b-it-nvfp4-hybrid \
2 --port 8000 --max-model-len 4096 \
3 --gpu-memory-utilization 0.55 --enforce-eager --trust-remote-code \
4 --generation-config vllm \
5 --max-num-batched-tokens 8192 --max-num-seqs 64 \
6 --speculative-config '{"model":"undeadindustries/gemma-4-31b-it-dflash-drafter-nvfp4","method":"dflash","num_speculative_tokens":15}'vllm-node:latest). Combined weights: ~36 GB VRAM.google/gemma-4-31b-it (Apache 2.0).
Subject to Gemma Terms of Use.