thu-ml/TurboDiffusion
TurboDiffusion: 100–200× Acceleration for Video Diffusion Models
What it solves
TurboDiffusion addresses the high computational cost and slow generation speeds of video diffusion models. It significantly reduces end-to-end generation latency—by 100 to 200 times on a single RTX 5090—while attempting to maintain the original video quality.
How it works
The framework achieves acceleration through a combination of three primary technologies:
- Attention Acceleration: It utilizes SageAttention and Sparse-Linear Attention (SLA) to speed up the attention mechanism.
- Timestep Distillation: It employs rCM for timestep distillation to reduce the number of sampling steps required.
- Quantization: It provides quantized checkpoints for linear layers to enable high-performance generation on consumer GPUs like the RTX 4090 and 5090.
Who it’s for
It is designed for developers and researchers working with video generation models (specifically the Wan series) who need near-instantaneous inference speeds on a single GPU.
Highlights
- Massive Speedup: Reduces generation time from 184 seconds to 1.9 seconds for certain models on an RTX 5090.
- Versatile Support: Supports both Text-to-Video (T2V) and Image-to-Video (I2V) workflows.
- Flexible Resolutions: Capable of generating videos at 480p and 720p resolutions.
- Multi-GPU Compatibility: Offers both quantized and unquantized checkpoints to suit different hardware (e.g., RTX series vs. H100).
Related
- Project
- Project
- Project
- Dispatch
- Project