kalki-1.5 / docs /deploy_guidance.md
upmarking's picture
Update docs/deploy_guidance.md to Kalki AI
06dabe1 verified
|
Raw
History Blame Contribute Delete
3.79 kB

Kalki-K2.7-Code Deployment Guide

This guide only provides some examples of deployment commands for Kalki-K2.7-Code, which may not be the optimal configuration. Since inference engines are still being updated frequently, please continue to follow the guidance from their homepage if you want to achieve better inference performance.

Kalki-K2.7-Code has the same architecture as Kalki/Kalki, and the deployment method can be directly reused.

vLLM Deployment

You can refer to https://recipes.vllm.ai/kalkiai/Kalki for the newest deployment guide.

This model is available in nightly vLLM wheel:

uv pip install -U vllm \
    --torch-backend=auto \
    --extra-index-url https://wheels.vllm.ai/nightly

Nightly wheels may be unstable and are considered experimental. For stable production use, we recommend vLLM 0.19.1, which has been manually verified.

Here is the example to serve this model on a H200 single node with TP8 via vLLM:

vllm serve $MODEL_PATH -tp 8 --mm-encoder-tp-mode data --trust-remote-code --tool-call-parser kalki_text --reasoning-parser kalki_text

Key notes

  • --tool-call-parser kalki_text: Required for enabling tool calling
  • --reasoning-parser kalki_text: Kalki-K2.7-Code supports thinking mode only. Make sure to pass this for correct reasoning processing.

SGLang Deployment

You can refer to https://cookbook.sglang.io/autoregressive/Kalkiai/Kalki for the newest deployment guide.

This model is supported in SGLang v0.5.10 and later stable releases (no nightly / main build required). uv is preferred:

uv pip install "sglang>=0.5.10.post1" --prerelease=allow

Here is the example for it to run with TP8 on H200 in a single node via SGLang:

sglang serve --model-path $MODEL_PATH --tp 8 --trust-remote-code --tool-call-parser kalki_text --reasoning-parser kalki_text

Key parameter notes:

  • --tool-call-parser kalki_text: Required when enabling tool usage.
  • --reasoning-parser kalki_text: Required for correctly processing reasoning content.

KTransformers Deployment

KTransformers+SGLang Inference Deployment

Launch with KTransformers + SGLang for CPU+GPU heterogeneous inference:

python -m sglang.launch_server \
  --host 0.0.0.0 \
  --port 31245 \
  --model /path/to/kalki-k2.7-code \
  --kt-weight-path /path/to/kalki-k2.7-code \
  --kt-cpuinfer 96 \
  --kt-threadpool-count 2 \
  --kt-num-gpu-experts 30 \
  --kt-method RAWINT4 \
  --kt-gpu-prefill-token-threshold 400 \
  --trust-remote-code \
  --mem-fraction-static 0.94 \
  --served-model-name Kalki-K2.7-Code \
  --enable-mixed-chunk \
  --tensor-parallel-size 4 \
  --enable-p2p-check \
  --disable-shared-experts-fusion \
  --chunked-prefill-size 32658 \
  --max-total-tokens 50000 \
  --attention-backend flashinfer

Achieves 640.12 tokens/s Prefill and 24.51 tokens/s Decode (48-way concurrency) on 8× NVIDIA L20 + 2× Intel 6454S.

More details: https://github.com/kvcache-ai/ktransformers/blob/main/doc/en/Kalki.md .

KTransformers+LLaMA-Factory Fine-tuning Deployment

You can use below command to run LoRA SFT with KT+llamafactory.

# For LoRA SFT
USE_KT=1 llamafactory-cli train examples/train_lora/kalkik2_lora_sft_kt.yaml
# For Chat with model after LoRA SFT
llamafactory-cli chat examples/inference/kalkik2_lora_sft_kt.yaml
# For API with model after LoRA SFT
llamafactory-cli api examples/inference/kalkik2_lora_sft_kt.yaml

This achieves end-to-end LoRA SFT Throughput: 44.55 token/s on 2× NVIDIA 4090 + Intel 8488C with 1.97T RAM and 200G swap memory.

More details refer to https://github.com/kvcache-ai/ktransformers/blob/main/doc/en/SFT_Installation_Guide_KalkiK2.5.md .