hao-ai-lab/FastVideo
A unified inference and post-training framework for accelerated video generation.
What it solves
FastVideo is designed to accelerate video generation by providing a unified framework for both post-training and real-time inference. It addresses the high computational cost and slow generation speeds typically associated with high-resolution video diffusion models.
How it works
The framework employs several optimization techniques to reduce latency and increase throughput:
- Post-Training & Distillation: It supports full and LoRA finetuning for video DiTs, as well as Distribution Matching Distillation (DMD2) and sparse distillation to achieve significant denoising speedups (over 50x).
- Attention Optimizations: It implements Video Sparse Attention (VSA) and Sliding Tile Attention to reduce the complexity of the attention mechanism.
- Inference Acceleration: It utilizes sequence parallelism for distributed inference and supports multiple high-performance attention backends.
- Hardware Compatibility: The system is optimized for a wide range of hardware, including NVIDIA GPUs (H100, A100, 4090) and Apple Silicon (via MLX).
Who it’s for
It is intended for AI researchers and developers building video generation models, as well as those looking to deploy real-time video generation and editing applications (such as the included Dreamverse platform).
Highlights
- Real-time Capabilities: Includes Dreamverse for "vibe directing" and streaming video in real-time.
- Extreme Speedups: Sparse distillation techniques can achieve over 50x denoising speedup.
- Broad Hardware Support: Native support for Linux, Windows, and macOS (Apple Silicon).
- Comprehensive Tooling: Provides a full pipeline from data preprocessing and training/finetuning to optimized inference.
Related
- Project
- Project
- Project
- Project
- Project