Recommend public vLLM image for Kimi-K3
Browse filesSigned-off-by: Zhiyu Cheng <zhiyuc@nvidia.com>
README.md
CHANGED
|
@@ -150,13 +150,11 @@ This model was obtained from Kimi-K3 without calibration data. The source MXFP4
|
|
| 150 |
|
| 151 |
## Usage
|
| 152 |
|
| 153 |
-
This checkpoint was validated on 8 NVIDIA B300 GPUs with [`
|
| 154 |
|
| 155 |
Native vLLM support is being upstreamed to replace the compatibility patch. Generic mixed `FP8_PB_WO` dispatch and DeepGEMM preparation are tracked in [vLLM PR #50617](https://github.com/vllm-project/vllm/pull/50617), Kimi-K3 fused-projection support in [vLLM PR #52406](https://github.com/vllm-project/vllm/pull/52406), and the TRTLLM NVFP4 SiTU scale correction in [vLLM PR #52405](https://github.com/vllm-project/vllm/pull/52405). Continue using the bundled patch until those changes are included in a released vLLM build.
|
| 156 |
|
| 157 |
```sh
|
| 158 |
-
docker login nvcr.io
|
| 159 |
-
|
| 160 |
hf download nvidia/Kimi-K3-NVFP4 \
|
| 161 |
runtime_patches/sitecustomize.py \
|
| 162 |
--local-dir kimi-k3-runtime
|
|
@@ -173,7 +171,7 @@ docker run --rm --gpus all --ipc=host --network=host \
|
|
| 173 |
-e VLLM_USE_RUST_FRONTEND=0 \
|
| 174 |
-v "$HOME/.cache/huggingface:/root/.cache/huggingface" \
|
| 175 |
-v "$PWD/kimi-k3-runtime/runtime_patches:/k3-runtime-patches:ro" \
|
| 176 |
-
|
| 177 |
vllm serve nvidia/Kimi-K3-NVFP4 \
|
| 178 |
--served-model-name kimi-k3-nvfp4 \
|
| 179 |
--host 0.0.0.0 \
|
|
@@ -182,10 +180,16 @@ docker run --rm --gpus all --ipc=host --network=host \
|
|
| 182 |
--quantization modelopt_mixed \
|
| 183 |
--tensor-parallel-size 8 \
|
| 184 |
--moe-backend flashinfer_trtllm \
|
|
|
|
| 185 |
--max-model-len 196608 \
|
|
|
|
|
|
|
| 186 |
--kv-cache-dtype fp8 \
|
| 187 |
--attention-config '{"mla_prefill_backend":"TRTLLM_RAGGED","use_prefill_query_quantization":true}' \
|
| 188 |
--attention-backend FLASHINFER_MLA \
|
|
|
|
|
|
|
|
|
|
| 189 |
--enable-auto-tool-choice \
|
| 190 |
--tool-call-parser kimi_k3 \
|
| 191 |
--reasoning-parser kimi_k3
|
|
|
|
| 150 |
|
| 151 |
## Usage
|
| 152 |
|
| 153 |
+
This checkpoint was validated on 8 NVIDIA B300 GPUs with the public [`vllm/vllm-openai:kimi-k3`](https://hub.docker.com/r/vllm/vllm-openai/tags?name=kimi-k3) image. At validation time, the image included vLLM `0.1.dev19262+gb6bbf29dd.d20260727`. The compatibility patch included in this repository is required for this serving configuration.
|
| 154 |
|
| 155 |
Native vLLM support is being upstreamed to replace the compatibility patch. Generic mixed `FP8_PB_WO` dispatch and DeepGEMM preparation are tracked in [vLLM PR #50617](https://github.com/vllm-project/vllm/pull/50617), Kimi-K3 fused-projection support in [vLLM PR #52406](https://github.com/vllm-project/vllm/pull/52406), and the TRTLLM NVFP4 SiTU scale correction in [vLLM PR #52405](https://github.com/vllm-project/vllm/pull/52405). Continue using the bundled patch until those changes are included in a released vLLM build.
|
| 156 |
|
| 157 |
```sh
|
|
|
|
|
|
|
| 158 |
hf download nvidia/Kimi-K3-NVFP4 \
|
| 159 |
runtime_patches/sitecustomize.py \
|
| 160 |
--local-dir kimi-k3-runtime
|
|
|
|
| 171 |
-e VLLM_USE_RUST_FRONTEND=0 \
|
| 172 |
-v "$HOME/.cache/huggingface:/root/.cache/huggingface" \
|
| 173 |
-v "$PWD/kimi-k3-runtime/runtime_patches:/k3-runtime-patches:ro" \
|
| 174 |
+
vllm/vllm-openai:kimi-k3 \
|
| 175 |
vllm serve nvidia/Kimi-K3-NVFP4 \
|
| 176 |
--served-model-name kimi-k3-nvfp4 \
|
| 177 |
--host 0.0.0.0 \
|
|
|
|
| 180 |
--quantization modelopt_mixed \
|
| 181 |
--tensor-parallel-size 8 \
|
| 182 |
--moe-backend flashinfer_trtllm \
|
| 183 |
+
--gpu-memory-utilization 0.90 \
|
| 184 |
--max-model-len 196608 \
|
| 185 |
+
--max-num-batched-tokens 8192 \
|
| 186 |
+
--max-num-seqs 32 \
|
| 187 |
--kv-cache-dtype fp8 \
|
| 188 |
--attention-config '{"mla_prefill_backend":"TRTLLM_RAGGED","use_prefill_query_quantization":true}' \
|
| 189 |
--attention-backend FLASHINFER_MLA \
|
| 190 |
+
--enable-prefix-caching \
|
| 191 |
+
--enable-chunked-prefill \
|
| 192 |
+
--max-cudagraph-capture-size 32 \
|
| 193 |
--enable-auto-tool-choice \
|
| 194 |
--tool-call-parser kimi_k3 \
|
| 195 |
--reasoning-parser kimi_k3
|