tonyd2wild/DeepSeek-v4-Flash-Vision-Exp-DSpark-1M-NVFP4-KV-2x-DGX-Spark
DeepSeek V4 Flash DSpark 1M NVFP4 KV recipe for 2x DGX Spark
解决的问题
此项目为 DeepSeek-V4-Flash-Vision-Exp 多模态模型在 DGX Spark 集群上提供生产就绪的部署配方。它特别解决了视觉变体缺乏生产级服务引擎的问题,提供必要的端口和补丁,以启用原生图像输入和使用 vLLM 的高吞吐量推理。
运作方式
此项目在 2 个 DGX Spark 节点上使用张量并行(TP=2)和 vLLM 实现服务堆栈。它利用自定义运行时镜像和一系列关键的 bind-mount 来应用补丁,而无需重建容器。关键技术组件包括:
- DSpark 推测解码:使用 drafter 加速 token 生成,并以特定的
k=5配置进行优化。 - NVFP4 KV 缓存:采用实验性的
nvfp4_ds_mlaKV 缓存来支持大型上下文窗口(最多 1M tokens)。 - 视觉端口:将 32 块 ViT 和 aligner 直接集成到 vLLM 中,无需 VLM 代理。
- 关键补丁:包含 Patch 3(调度器修复)、Patch 4(共享专家加载器修复,防止静默性能下降)和 Patch 5(在推理段期间抑制停止字符串)。
适用对象
在 DGX Spark 硬件上部署 DeepSeek-V4-Flash 模型的 ML 工程师和基础设施研究人员,需要高吞吐量、长上下文多模态功能。
亮点
- 原生多模态支持:将视觉功能直接集成到服务引擎中。
- 大型上下文:针对 1M-token 上下文窗口进行校准。
- 高吞吐量:在特定工作负载下可达 80.1 tok/s(使用 Patch 4)。
- 防止静默失败:提供特定脚本和补丁,以修复 draft 的共享专家加载未初始化状态的关键错误,否则会将解码速度减半而不影响输出质量。
相关
- 项目
- 项目
- 项目
- 项目