Open-Sora-Plan

Open-Sora-Plan 项目介绍

Open-Sora-Plan 是一个由北京大学-兔展AIGC联合实验室发起的开源项目，旨在通过开源社区的力量复现OpenAI的Sora模型。该项目致力于创建一个简单且可扩展的代码库，以实现类似Sora的视频生成能力。

项目背景与目标

项目的主要目标是复现Sora模型的功能，实现高质量的文本到视频生成。

虽然当前版本与最终目标还有一定差距，但项目团队正在持续完善和快速迭代。

Open-Sora-Plan欢迎开源社区的贡献，鼓励开发者提交Pull Request。

技术特点

支持国产AI计算系统：

完整支持使用华为昇腾AI计算系统进行训练和推理。

基于昇腾训练的模型可以输出与业界水平相当的视频质量。

高性能的CausalVideoVAE：

具有出色的压缩比，能够将视频压缩256倍（4×8×8）。

采用因果卷积，支持图像和视频的同时推理，只需1个节点即可训练。

基于3D注意力的视频扩散模型：

使用新的稀疏注意力架构，取代2+1D模型。

3D注意力能更好地捕捉空间和时间特征的联合表示。

项目进展

Open-Sora-Plan已经发布了多个版本，每个版本都带来了显著的改进：

v1.3.0版本引入了WFVAE、提示词精炼器、数据过滤策略、稀疏注意力和桶训练策略等新特性。

v1.2.0版本支持图像到视频生成和过渡生成，采用了真正的3D视频扩散模型。

v1.1.0版本大幅提升了视频质量和长度。

v1.0.0版本显著提高了视频生成质量和文本控制能力。

资源与工具

项目提供了多个版本的模型权重、数据集和工具：

不同版本的扩散模型，支持多种分辨率和帧数。

CausalVideoVAE模型，支持任意尺寸。

数据集和标注。

提示词精炼器。

使用方法

Open-Sora-Plan提供了多种使用方式：

Gradio Web UI：推荐使用的网页演示界面。

训练与推理：项目提供了详细的文档，包括CausalVideoVAE、文本到视频、图像到视频等模块的训练和推理指南。

社区与贡献

Open-Sora-Plan是一个开放的社区项目，欢迎开发者以多种方式参与贡献：

提交代码改进和新功能。

报告问题和提出建议。

改进文档和示例。

项目遵循MIT许可证，鼓励开放合作和创新。

通过这个项目，开发者可以深入了解和参与最前沿的视频生成技术研究，为开源AI社区做出贡献。