NVlabs/LongLive

Long Video Gen Infrastructure

它解决了什么问题

LongLive 2.0 是一种基础设施,旨在使长视频生成更加高效和可扩展。它解决了生成长序列视频帧所涉及的高计算和内存成本问题,支持实时交互式生成和高吞吐量推理。

它如何工作

该项目实现了一个并行基础设施,利用 NVFP4(一种 4 位浮点格式)和序列并行性来优化训练和推理。

在训练方面,它支持文本到视频(T2V)和图像到视频(I2V)模态的自回归(AR)教师强制训练,以及少步数蒸馏(DMD 蒸馏),以减少生成所需的步数。

在推理方面,它利用 NVFP4 权重和 KV 缓存来降低内存开销,采用序列并行推理和异步解码。它还结合了多帧注意力池等技术,以在长时间内保持稳定性。

适用于谁

该基础设施适用于从事长时程视频生成的研究人员和开发者,特别是那些使用 NVIDIA GPU(SM90/H100)以最大化吞吐量和利用硬件加速量化的人。

主要亮点

  • NVFP4 支持:实现 W4A4 量化和 NVFP4 KV 缓存,显著加快推理速度(最高达 45.7 FPS)。
  • 因果扩散:支持 AR 训练和 DMD 蒸馏,实现高效、少步数的视频生成。
  • 并行性:具备平衡的序列并行训练和序列并行推理能力,可处理长序列。
  • 多功能模态:支持文本到视频(T2V)和图像到视频(I2V)工作流。
  • FP8 PTQ:包含 TorchAO FP8 后训练量化,适用于 BF16 检查点。

相关

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