zhiyucheng commited on
Commit
3f7e105
·
unverified ·
1 Parent(s): 1ec10fa

Recommend public vLLM image for Kimi-K3

Browse files

Signed-off-by: Zhiyu Cheng <zhiyuc@nvidia.com>

Files changed (1) hide show
  1. README.md +8 -4
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 [`nvcr.io/nvidia/ai-dynamo/vllm-runtime:1.4.0-kimi-k3-dev.1`](https://catalog.ngc.nvidia.com/orgs/nvidia/ai-dynamo/containers/vllm-runtime/1.4.0-kimi-k3-dev.1), which includes vLLM `0.1.dev19251+g13c59a3da.d20260726`. The image requires NGC authentication. 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
- 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
- nvcr.io/nvidia/ai-dynamo/vllm-runtime:1.4.0-kimi-k3-dev.1 \
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