
nuwa-pytorch
用文本生成视频的PyTorch实现,体验女娲模型
项目预览
项目数据
项目介绍
Star 增长趋势
本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整
数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 547
使用教程
ℹ 本教程由 AI 依据该项目 README 整理,个别步骤可能需要按你的环境微调。
环境要求
- PyTorch
- 单卡 16G 以上显存
- 视频文本对数据
安装与启动步骤
-
1装包
纯 PyTorch 实现,装完就能在代码里调。
pip install nuwa-pytorch -
2先训 VAE
女娲是在离散码本上做自回归,VAE 不先训好后面无从谈起。
python -c "from nuwa_pytorch import VQGanVAE; VQGanVAE(dim=512, num_layers=4, image_size=256)" -
3再接文本到视频
在训好的 VAE 之上叠因果 Transformer,才是完整的两阶段流程。
-
4备训练数据
文本-视频成对数据最难搞,先用小规模公开集验证管线。
-
5读作者的状态说明
他会直接写明这条路线的现状,看完再决定要不要投入。
如何确认成功
VAE 能在小批样本上跑通前向,重建出的帧与输入尺寸一致且能看出原始内容
注意事项
- 这是社区复现不是官方代码,作者本人也说明了扩散模型已取代该路线的 SOTA 地位。
- 仓库定位是研究复现与思路验证,不要期待开箱就能生成可用视频。
- 两阶段流程里 VAE 的训练质量决定上限,跳过这步直接训上层基本白费。
- 视频和音频双解码器的扩展是它的额外尝试,与主流程可以分开看。
核心亮点
- 完整实现NUWA核心架构,含3D注意力机制和VQVAE
- 提供文本到视频、图像动画、视频预测多任务示例
- 代码模块化,易于扩展和二次开发
不足之处
- 非官方实现,可能未完全对齐原论文细节
- 依赖预训练权重,但未提供下载链接
适用场景
- 学术研究:复现NUWA模型用于多模态生成实验
- 创意内容:根据文本描述快速生成短视频素材
- 教学演示:学习3D Transformer在视频生成中的应用
替代项目
CogVideo、VideoGPT、TATS
上一篇:AI-Ecosystem
下一篇:ContextDiff
同类项目推荐
RET-CLIP
开源
用临床报告训练视网膜AI,零样本识别眼科疾病
RET-CLIP: A Retinal Image Foundation Model Pre-trained with Clinical Diagnostic Repo···
GFT
开源
用树形词汇表让图模型学会跨数据集迁移,省去重复训练
GFT: Graph Foundation Model with Transferable Tree Vocabulary, NeurIPS 2024.
PocketDex
开源
用 SwiftUI 快速撸一个宝可梦图鉴,学并发和 MVVM
A simple Pokedex app written in SwiftUI that implements the PokeAPI, using Swift Con···
BayLing-Speech
开源
让开源模型像GPT-4o一样自然流畅地语音对话
LLaMA-Omni is a low-latency and high-quality end-to-end speech interaction model bui···
InternLM
开源
开箱即用的中文大模型,从预训练到部署全流程搞定
Official release of InternLM series (InternLM, InternLM2, InternLM2.5, InternLM3).
Awesome-LLM-Reasoning-with-NeSy
开源
一网打尽神经符号+大模型前沿,快速入门推理增强
✨✨Latest Advances on Neuro-Symbolic Learning in the era of Large Language Models
hello-claw
开源
从零搭建你的第一个AI助手,像领养龙虾一样简单有趣
哈喽!龙虾 ♀️ Adopt from scratch and build your first claw 来领养你的第一只龙虾···
CoreML-Models
开源
下载即用的苹果端 AI 模型库,免去转换烦恼
Core ML model zoo for iOS/macOS — PyTorch models converted to ready-to-use .mlpacka···
Awesome-MLLM-LLM-Colab
开源
一键运行主流大模型,Colab里轻松做实验
Happy experimenting with MLLM and LLM models!