【免费下载链接】Kimi-K2-Thinking-NVFP4
Kimi-K2-Thinking-NVFP4 是 NVIDIA 发布的万亿参数推理模型 Kimi-K2-Thinking 的NVFP4 4位量化版本。本文带你用 SGLang 部署大模型推理服务:3 条命令即可启动 OpenAI 兼容的 API 接口,并附完整的关键参数清单与模型文件说明,新手照着做就能跑起来。
为什么选 NVFP4 量化版:1T 参数轻松省内存
在动手之前,先弄清楚这个仓库里装的是什么:
- 基座模型:Moonshot AI 的 Kimi-K2-Thinking,采用 DeepSeek V3 架构(MLA 注意力 + MoE 专家混合),总参数约 1T
- 量化方式:使用 NVIDIA TensorRT Model Optimizer(modelopt)将 MoE 中线性层的权重与激活从 BF16 转为NVFP4(4 位浮点,group_size=16),KV Cache 使用FP8
- 目标硬件:NVIDIA Blackwell 架构 GPU(如 B200,官方实测硬件),仅支持 Linux 环境
- 支持引擎:vLLM / SGLang / TensorRT-LLM,本文以 SGLang 为例
💡 相比 BF16 原版,NVFP4 权重大约只占原来的 1/4,显存和带宽压力大幅下降,这是它能跑得起 1T 级 MoE 模型的关键。
详细背景可查阅仓库中的 README.md。
SGLang 部署三步走:拉镜像 → 启动服务 → 验证接口
第 1 条命令:拉取 SGLang 官方镜像
docker pull lmsysorg/sglang:latest第 2 条命令:启动 SGLang 推理服务(核心命令)
docker run --gpus all -it --rm \ -p 30000:30000 \ -v /path/to/Kimi-K2-Thinking-NVFP4:/data/Kimi-K2-Thinking-NVFP4 \ lmsysorg/sglang:latest \ python3 -m sglang.launch_server \ --model-path /data/Kimi-K2-Thinking-NVFP4 \ --tp 4 --quantization modelopt_fp4 --trust-remote-code如果模型已在本地 Hugging Face 缓存目录中,也可以直接--model-path nvidia/Kimi-K2-Thinking-NVFP4让引擎自动拉取。日志出现The server is fired up and ready to roll!即表示加载完成。
第 3 条命令:用 curl 验证服务
curl http://127.0.0.1:30000/v1/chat/completions \ -H "Content-Type: application/json" \ -d '{"model":"Kimi-K2-Thinking-NVFP4","messages":[{"role":"user","content":"用一句话介绍一下你自己"}]}'返回 JSON 中能看到choices字段,说明 SGLang 部署成功 ✅,任何支持 OpenAI 协议的客户端(Python SDK、LangChain 等)都可以直接接入。
关键参数完整清单:启动时每个参数都是干什么的
| 参数 | 是否必加 | 默认值 | 作用说明 |
|---|---|---|---|
--model-path | ✅ 必加 | - | 模型目录路径或 HF 模型 ID,指定加载哪个 checkpoint |
--tp 4 | ✅ 强烈建议 | 1 | Tensor 并行度。1T 参数模型单卡放不下,官方示例用 4 卡,可按显存改为 8 |
--quantization modelopt_fp4 | ✅ 必加 | auto | 告诉引擎按 NVFP4 格式解析权重,漏掉会导致加载慢或精度异常 |
--trust-remote-code | ✅ 必加 | False | 本仓库自带推理代码(见下文文件清单),必须信任远程代码才能加载 |
--port | 可选 | 30000 | 服务监听端口,多实例部署时避免冲突 |
--host | 可选 | 127.0.0.1 | 需要跨机器访问时设为0.0.0.0 |
--mem-fraction-static | 可选 | 0.85 | 静态显存(权重+KV Cache)占比。显存紧张(OOM)时调低,如 0.8 |
--context-length | 可选 | 模型上限 | 限制最大上下文长度,模型理论上限 256K |
--max-running-requests | 可选 | 自动 | 最大并发请求数,显存吃紧时调小可保稳定 |
⚠️ 新手最常踩的三个坑:① 忘记
--quantization modelopt_fp4;②--tp设置过低导致显存不足;③ 漏掉--trust-remote-code报Unrecognized model类错误。
模型文件清单:仓库里每个文件的作用
| 文件 | 作用 |
|---|---|
| model-00001-of-00119.safetensors ~ model-00119-of-00119.safetensors | NVFP4 量化后的模型权重,共 119 个分片 |
| model.safetensors.index.json | 张量 → 分片文件的映射索引,引擎据此加载权重 |
| config.json | 模型结构配置:层数、专家数、注意力参数等 |
| hf_quant_config.json | 量化配置:NVFP4权重 +FP8KV Cache、group_size=16、不参与量化的模块列表 |
| modeling_deepseek.py | DeepSeek V3 模型推理代码(--trust-remote-code会加载它) |
| configuration_deepseek.py | 模型 Config 类的自定义实现 |
| chat_template.jinja | 对话模板:支持 system/user/assistant 多轮、思维链(thinking)与工具调用 |
| tokenizer_config.json、tiktoken.model、tokenization_kimi.py | 分词器配置(tiktoken 方案),词表大小 163840 |
| generation_config.json | 生成默认值:max_length: 262144,eos_token_id 163586 |
| LICENSE | NVIDIA Open Model License,可商用/非商用 |
模型核心规格速览
| 规格项 | 数值 | 说明 |
|---|---|---|
| 总参数量 | ~1T | MoE 架构,每 token 只激活部分专家 |
| 网络架构 | DeepSeek V3 | MLA 多头潜在注意力 + MoE |
| 层数 | 61 | 见 config.json 中num_hidden_layers |
| 专家配置 | 384 个路由专家 + 1 个共享专家 | 每 token 激活 top-8 专家 |
| 上下文长度 | 262144(256K) | max_position_embeddings,YARN RoPE 缩放 |
| 权重精度 | NVFP4(4 位,组大小 16) | 注意力层与共享专家保持高精度未量化 |
| KV Cache 精度 | FP8 | 进一步压缩长上下文显存占用 |
| 测试硬件 | B200 | NVIDIA Blackwell 架构 |
常见问题排查清单
- CUDA out of memory:调大
--tp(4→8)、调低--mem-fraction-static,或减少--max-running-requests - 加载权重极慢 / 显存爆掉:检查是否漏加
--quantization modelopt_fp4,导致按非量化路径解析 - 报
Unrecognized model architecture:补上--trust-remote-code - 非 Blackwell 显卡跑不起来:NVFP4 依赖 Blackwell(如 B200)硬件特性,A100/H100 请改用原版或其他量化格式
- 服务存活检查:请求
GET /health返回 200 即正常,适合接入监控脚本
小结
用 SGLang 部署 Kimi-K2-Thinking-NVFP4 的核心就是记住这 3 条命令:docker pull拉镜像 →sglang.launch_server加--tp 4 --quantization modelopt_fp4 --trust-remote-code启动 →curl /v1/chat/completions验证。参数方面,--model-path、--tp、--quantization、--trust-remote-code四件套一个都不能少。掌握这些,你就可以把这台万亿参数推理大模型接入自己的业务了 🚀
【免费下载链接】Kimi-K2-Thinking-NVFP4
相关推荐
Kimi K2部署实战:vLLM、SGLang、TensorRT-LLM完整配置指南
Kimi K2部署实战:vLLM、SGLang、TensorRT LLM完整配置指南 🚀 终极指南:快速掌握Kimi K2大语言模型的高性能部署方案 Kimi
大模型人工智能SGLang部署Kimi-K2-Thinking-Pruning实战:单节点8卡运行256K上下文思考模型
SGLang部署Kimi K2 Thinking Pruning实战:单节点8卡运行256K上下文思考模型 Kimi K2 Thinking Pruning 是
大模型推理模型模型量化一文读懂Kimi-K2-Thinking-NVFP4:1T参数MoE思考大模型的NVFP4量化完全指南
一文读懂Kimi K2 Thinking NVFP4:1T参数MoE思考大模型的NVFP4量化完全指南 Kimi K2 Thinking NVFP4 是 NVI
创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考