MiaAI-Lab/DeepSeek-v4-Flash-DSpark-2x-DGX-Spark

DeepSeek-v4-Flash 0731 recipe for 2x DGX Sparks

解决的问题

本项目提供了一种在两台(或三台)NVIDIA DGX Spark 节点集群上运行 DeepSeek-V4-Flash-Vision-Exp 模型的部署方案和优化运行时。它特别解决了管理超大上下文窗口(最高达100万token)、通过推测解码优化吞吐量,以及在 Anemll 运行时中启用原生图像支持的挑战。

工作原理

系统使用 vLLM 配合张量并行(TP=2)和 DSpark 推测解码来加速推理。采用专用的 KV 缓存(nvfp4_ds_mla)以处理 100 万 token 的上限。部署通过 Docker 容器化,可配置为在每个节点上复制模型权重,或通过 ConnectX 链路使用 NFSv4 从主节点挂载权重,以节省存储空间。

适用对象

专为拥有 DGX Spark 硬件并希望以高吞吐量、长上下文能力及图像处理功能部署 DeepSeek 的 Flash Vision 模型的运维人员和研究人员设计。

主要亮点

  • 超大上下文: 支持高达 1,048,576 个 token 的上下文上限。
  • 推测解码: 使用 DSpark 提升生成速度。
  • 原生视觉支持: 包含 Anemll 运行时的启动热修复,支持通过 OpenAI 兼容的 image_url 和 <image> 标签处理图像。
  • 灵活的权重管理: 支持在工作节点本地缓存权重,也支持从主节点通过 NFS 挂载。
  • 消融支持: 可选的运行时拒绝方向投影,用于修改模型行为。
  • 优化吞吐量: 在多个并发流上可达到约 160–190 token/秒的聚合吞吐量。

相关

  • 项目
  • 项目
  • Dispatch
  • 项目
  • 项目