研究人员开发了 MegaSlide-DiT,这是一个能够在一块高端 GPU 上适配大型视频扩散模型的系统。这是通过将模型权重和优化器状态保留在主机 RAM 中,并将仅必要的碎片流式传输到 GPU 来实现的。此外,该系统用线性复杂度的 3D 可变形滑动注意力(3D-DSA)算子取代了计算成本高昂的二次注意力,该算子能够自适应运动。 AI
影响 该系统有望使资源有限的研究人员更容易进行大规模视频扩散模型的适配。
排序理由 该集群包含一篇详细介绍适配大型 AI 模型新系统的研究论文。[lever_c_demoted from research: ic=1 ai=1.0]
- 3D Deformable Slide Attention
- 3D-DSA
- arXiv
- Diffusion Transformers
- H200 GPU
- Hugging Face
- MegaSlide-DiT
AI 生成摘要 · Google Gemini · 来自 1 个来源。 我们如何撰写摘要 →