nuwa-pytorch

用文本生成视频的PyTorch实现,体验女娲模型

开源 🏷️完全免费 🔧 仅开源项目,无在线服务,需自行部署 视频生成 🍴 复刻 55 🐛 待处理 8 👀 关注 19
GitHub ↗
GitHub 星标 ★ 547
维护状态 低维护
是否开源 开源
定价模式 🏷️完全免费 开源项目,MIT许可证,完全免费,需自行部署使用。

项目预览

nuwa-pytorch 项目界面预览图
图片来源:GitHub 官方仓库名片

项目数据

分类视频生成
开发团队lucidrains
定价模式🏷️完全免费
价格说明开源项目,MIT许可证,完全免费,需自行部署使用。
访问状态🔧 仅开源项目,无在线服务,需自行部署
是否开源是
开源协议MIT
主要语言Python
技术栈/模型artificial-intelligence,attention-mechanism,deep-learning,text-to-audio,text-to-video,transformers
GitHub 星标★ 547
上线时间2021-11-28 00:00:00
代码更新2023-01-17
仓库体积1.8 MB
本站数据更新2026-11-09
维护状态低维护
收录时间2026-08-09
浏览次数2

项目介绍

NUWA(女娲)是微软提出的多模态预训练模型,支持文本到视频、文本到图像及视频预测等任务。此项目为NUWA的PyTorch非官方实现,基于3D Transformer和3D Nearby Attention(3DNA)机制,通过VQVAE将视频和图像编码为离散token,再以自回归方式生成视频。它解决了文本到视频合成中时空一致性和计算效率问题,核心能力包括文本条件视频生成、图像动画化及视频预测。项目代码结构清晰,提供推理示例,适合研究者和开发者快速体验NUWA架构。

Star 增长趋势

本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整

数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 547

使用教程

ℹ 本教程由 AI 依据该项目 README 整理,个别步骤可能需要按你的环境微调。

难度:高级 约 60 分钟 部署方式:本地安装 5 步

环境要求

  • PyTorch
  • 单卡 16G 以上显存
  • 视频文本对数据

安装与启动步骤

  1. 1装包

    纯 PyTorch 实现,装完就能在代码里调。

    pip install nuwa-pytorch
  2. 2先训 VAE

    女娲是在离散码本上做自回归,VAE 不先训好后面无从谈起。

    python -c "from nuwa_pytorch import VQGanVAE; VQGanVAE(dim=512, num_layers=4, image_size=256)"
  3. 3再接文本到视频

    在训好的 VAE 之上叠因果 Transformer,才是完整的两阶段流程。

  4. 4备训练数据

    文本-视频成对数据最难搞,先用小规模公开集验证管线。

  5. 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···

★ 73 2026-08-09
GFT 开源

用树形词汇表让图模型学会跨数据集迁移,省去重复训练

GFT: Graph Foundation Model with Transferable Tree Vocabulary, NeurIPS 2024.

★ 61 2026-08-09
PocketDex 开源

用 SwiftUI 快速撸一个宝可梦图鉴,学并发和 MVVM

A simple Pokedex app written in SwiftUI that implements the PokeAPI, using Swift Con···

★ 138 2026-08-09
BayLing-Speech 开源

让开源模型像GPT-4o一样自然流畅地语音对话

LLaMA-Omni is a low-latency and high-quality end-to-end speech interaction model bui···

★ 3145 2026-08-09
InternLM 开源

开箱即用的中文大模型,从预训练到部署全流程搞定

Official release of InternLM series (InternLM, InternLM2, InternLM2.5, InternLM3).

★ 7277 2026-08-09
Awesome-LLM-Reasoning-with-NeSy 开源

一网打尽神经符号+大模型前沿,快速入门推理增强

✨✨Latest Advances on Neuro-Symbolic Learning in the era of Large Language Models

★ 334 2026-08-09
hello-claw 开源

从零搭建你的第一个AI助手,像领养龙虾一样简单有趣

哈喽!龙虾 ‍♀️ Adopt from scratch and build your first claw 来领养你的第一只龙虾···

★ 2216 2026-08-09
CoreML-Models 开源

下载即用的苹果端 AI 模型库,免去转换烦恼

Core ML model zoo for iOS/macOS — PyTorch models converted to ready-to-use .mlpacka···

★ 1894 2026-08-10