ARTICLE · INTELLIGENCE

战地情报 · 详情页

来自尧图项目组的一线实战观察与深度解析

快速跑通 Wan2.2-TI2V-5B:单卡 720P 视频生成完整部署指南

快速跑通 Wan2.2-TI2V-5B:单卡 720P 视频生成完整部署指南

快速跑通 Wan2.2-TI2V-5B:单卡 720P 视频生成完整部署指南

【免费下载链接】Wan2.2-TI2V-5BWan2.2-TI2V-5B是一款开源的先进视频生成模型,基于创新的混合专家架构(MoE)设计,显著提升了视频生成的质量与效率。该模型支持文本生成视频和图像生成视频两种模项目地址: https://ai.gitcode.com/hf_mirrors/Wan-AI/Wan2.2-TI2V-5B

Wan2.2-TI2V-5B 是一款采用 MoE 混合专家架构的开源视频生成模型,一张24GB 显存(如 RTX 4090)就能跑出720P@24fps的 5 秒视频,同时支持文本生成视频和图像生成视频。本指南带你从下载权重到一条命令出结果,全程本地部署,无需外部服务。

RTX 4090 能跑吗:硬件与软件要求

先把门槛看清,能跑再动手。核心约束是显存 ≥ 24GB;内存和磁盘是软性建议。

项目最低要求建议
GPU 显存≥ 24GB(如 RTX 4090)≥ 80GB 可去掉省显存参数提速
内存(RAM)—≥ 32GB
磁盘空间预留≥ 20GB模型权重体积较大
Python≥ 3.8—
CUDA≥ 11.7—
PyTorch≥ 2.4.0与 CUDA 版本匹配

先克隆推理代码仓库(含generate.py与requirements.txt),再装依赖:

git clone https://gitcode.com/hf_mirrors/Wan-AI/Wan2.2-TI2V-5B Wan2.2 cd Wan2.2 pip install -r requirements.txt # 要求 torch >= 2.4.0

两种途径下载 Wan2.2-TI2V-5B 权重

权重和代码分开放:代码在上一步已克隆,这里把模型权重拉到本地./Wan2.2-TI2V-5B目录。两条途径任选其一。

途径一 · Hugging Face:

pip install "huggingface_hub[cli]" huggingface-cli download Wan-AI/Wan2.2-TI2V-5B --local-dir ./Wan2.2-TI2V-5B

途径二 · ModelScope:

pip install modelscope modelscope download Wan-AI/Wan2.2-TI2V-5B --local_dir ./Wan2.2-TI2V-5B

📌 落盘后目录主要包含:Wan2.2_VAE.pth(高压缩视频编码器)、models_t5_umt5-xxl-enc-bf16.pth(文本编码器)、diffusion_pytorch_model-*.safetensors(扩散模型权重分片)、config.json(模型配置,model_type: ti2v)以及google/umt5-xxl/分词器。

一条命令生成 720P 视频

装好代码和权重后,改一下--prompt就能出片。下面是文本到视频的最短命令:

python generate.py --task ti2v-5B --size 1280*704 --ckpt_dir ./Wan2.2-TI2V-5B \ --offload_model True --convert_model_dtype --t5_cpu \ --prompt "两只穿着舒适拳击装备的拟人化猫,在聚光灯下的舞台上激烈搏斗"

关键参数速查:

参数作用
--task ti2v-5B指定任务为文本/图像到视频
--size 1280*704720P分辨率;竖屏改为704*1280
--ckpt_dir权重本地路径
--offload_model True模型卸载,省显存
--convert_model_dtype转换参数精度,优化显存
--t5_cpu文本编码器放 CPU,进一步省显存
--prompt视频内容的文本描述

图像到视频只需多传一个--image,其余参数不变(见 示例输入图):

python generate.py --task ti2v-5B --size 1280*704 --ckpt_dir ./Wan2.2-TI2V-5B \ --offload_model True --convert_model_dtype --t5_cpu \ --image examples/i2v_input.JPG \ --prompt "夏季海滩度假风格,一只戴太阳镜的白猫坐在冲浪板上..."

⚠️ 图生视频时--size代表生成区域的面积,宽高比自动跟随输入图;不传--image则默认为文生视频。

它凭什么又快又强:MoE 架构与高压缩 VAE

跑通之后,再回头看它为什么能又快又强——核心是两项设计。

MoE 双专家架构:Wan2.2 把 MoE 引入视频扩散模型,按去噪阶段拆成两个专家——高噪声专家负责早期、搭整体布局,低噪声专家负责后期、细化细节。切换点由信噪比(SNR)决定。每个专家约14B参数,全系列共27B参数但每步只激活14B,推理算力与显存开销几乎不变。

高压缩 VAE:TI2V-5B 本身是5B 稠密模型,靠 Wan2.2-VAE 实现4×16×16时空压缩,总体压缩率达64;叠加额外的分块层后,总压缩比提升到4×32×32。正因如此,它不加任何优化就能在单张消费级 GPU 上9 分钟内生成 5 秒 720P 视频,属于当前720P@24fps里速度领先的开源模型,并在 Wan-Bench 2.0 上多项指标优于主流闭源商业模型。

进阶玩法:多卡推理与显存压榨

多卡加速:用 FSDP + DeepSpeed Ulysses 做分布式推理,把大模型摊到多卡上:

torchrun --nproc_per_node=8 generate.py --task ti2v-5B --size 1280*704 \ --ckpt_dir ./Wan2.2-TI2V-5B --dit_fsdp --t5_fsdp --ulysses_size 8 \ --image examples/i2v_input.JPG \ --prompt "夏季海滩度假风格,一只戴太阳镜的白猫坐在冲浪板上..."

单卡省显存:显存吃紧时保留--offload_model True、--convert_model_dtype、--t5_cpu三件套即可稳定跑在 24GB 上。

单卡提速:若你的 GPU显存 ≥ 80GB,把上面三个参数全部去掉,直接吃满算力、显著提速。

分辨率选择:

  • 横屏 720P:--size 1280*704
  • 竖屏 720P:--size 704*1280
  • 图生视频:--size只决定面积,比例跟随输入图,别强行改比例

Prompt 技巧:写得越具体出片越好——把场景、动作、情感、光照、构图、对比度这些美学标签都塞进去,模型可控性会明显提升。

踩坑速查

  • 遇到 显存不足(OOM)→ 加--offload_model True和--t5_cpu;还不行就换更大显存的卡。
  • 遇到 出片质量不达标→ 重写--prompt,补上场景、动作、光照、构图等细节描述。
  • 遇到 生成太慢→ 在 ≥80GB 显存的卡上去掉--offload_model、--convert_model_dtype、--t5_cpu。
  • 遇到 想要竖屏→--size改成704*1280;图生视频则别动比例,让它跟随输入图。
  • 遇到 单卡装不下大任务→ 上torchrun多卡 +--dit_fsdp --t5_fsdp --ulysses_size分布式推理。

Wan2.2-TI2V-5B 以 Apache 2.0 开源,让你能免费获得工业级的 720P 视频生成能力,直接用于学术研究与生产落地。跑通后想深挖参数或接入 ComfyUI / Diffusers,回到仓库 config.json 与 examples/ 目录翻一遍就行。

【免费下载链接】Wan2.2-TI2V-5BWan2.2-TI2V-5B是一款开源的先进视频生成模型,基于创新的混合专家架构(MoE)设计,显著提升了视频生成的质量与效率。该模型支持文本生成视频和图像生成视频两种模项目地址: https://ai.gitcode.com/hf_mirrors/Wan-AI/Wan2.2-TI2V-5B

创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考

RELATED READING

延伸阅读

更多一线实战笔记与深度复盘,助您持续精进