Text-to-Video
Diffusers
Safetensors
WanDMDPipeline
BrianChen1129's picture
Update README.md
3b45704 verified
|
raw
history blame
3.01 kB
metadata
license: apache-2.0
datasets:
  - FastVideo/Wan-Syn_77x448x832_600k
base_model:
  - Wan-AI/Wan2.1-T2V-1.3B-Diffusers

FastVideo FastWan2.1-T2V-1.3B-Diffusers Model

Introduction

This model is jointly finetuned with DMD and VSA, based on Wan-AI/Wan2.1-T2V-1.3B-Diffusers. It supports efficient 3-step inference and generates high-quality videos at 61×448×832 resolution. We adopt the FastVideo 480P Synthetic Wan dataset, consisting of 600k synthetic latents.


Model Overview

  • 3-step inference is supported and achieves up to 20 FPS on a single H100 GPU.
  • Our model is trained on 61×448×832 resolution, but it supports generating videos with any resolution.(quality may degrade)
  • Finetuning and inference scripts are available in the FastVideo repository:
  • Try it out on FastVideo — we support a wide range of GPUs from H100 to 4090, and also support Mac users!

Training Infrastructure

Training was conducted on 4 nodes with 32 H200 GPUs in total, using a global batch size = 64.
We enable gradient checkpointing, set gradient_accumulation_steps=2, and use learning rate = 1e-5.
We set VSA attention sparsity to 0.8, and training runs for 4000 steps (~12 hours)
The detailed training example script is available here.

If you use the FastWan2.1-T2V-1.3B-Diffusers model for your research, please cite our paper:

@article{zhang2025vsa,
  title={VSA: Faster Video Diffusion with Trainable Sparse Attention},
  author={Zhang, Peiyuan and Huang, Haofeng and Chen, Yongqi and Lin, Will and Liu, Zhengzhong and Stoica, Ion and Xing, Eric and Zhang, Hao},
  journal={arXiv preprint arXiv:2505.13389},
  year={2025}
}
@article{zhang2025fast,
  title={Fast video generation with sliding tile attention},
  author={Zhang, Peiyuan and Chen, Yongqi and Su, Runlong and Ding, Hangliang and Stoica, Ion and Liu, Zhengzhong and Zhang, Hao},
  journal={arXiv preprint arXiv:2502.04507},
  year={2025}
}