Plug-and-play sparse attention for video diffusion models
Project description
A one-line, plug-and-play sparse attention framework for accelerating video diffusion inference.
๐ฆ Installation
From PyPI
pip install sparsevideo --no-build-isolation
From source
git clone https://github.com/Mutual-Luo/SparseVideo.git
cd SparseVideo
MAX_JOBS=8 pip install -e . --no-build-isolation
๐ Quick Start
import torch
from diffusers import WanPipeline
import sparsevideo
pipe = WanPipeline.from_pretrained("Wan-AI/Wan2.1-T2V-14B-Diffusers", torch_dtype=torch.bfloat16)
pipe.to("cuda")
+ pipe = sparsevideo.replace_attention(pipe, method="svoo")
video = pipe("A cat playing piano", num_frames=81, num_inference_steps=50).frames[0]
โก Supported Sparse Attention Methods
| Method | Method | Method | |||
|---|---|---|---|---|---|
| โ | Dense (baseline) |
โ | SpargeAttn [paper] |
โ | Adacluster [paper] |
| โ | SVG1 [paper] |
โ | Radial [paper] |
โ | SVOO [paper] |
| โ | SVG2 [paper] |
โ | STA [paper] |
โ | FlashOmni [paper] |
| โ | SVG-EAR [paper] |
โ | DraftAttention [paper] |
๐งฉ Supported Frameworks
Works as a drop-in, one-line replacement for both Diffusers and DiffSynth-Studio pipelines. Just call sparsevideo.replace_attention(pipe, method=...), no model modifications required.
Diffusers
Supported 34 models:
| Model | Model | Model | |||
|---|---|---|---|---|---|
| โ | Wan 2.1 Text-to-Video 1.3B | โ | Wan 2.2 Text-to-Video A14B | โ | Video-as-Prompt Wan 2.1 14B |
| โ | Wan 2.1 Text-to-Video 14B | โ | Wan 2.2 Image-to-Video A14B | โ | HunyuanVideo Text-to-Video |
| โ | Wan 2.1 Image-to-Video 14B | โ | Wan 2.2 Speech-to-Video 14B | โ | HunyuanVideo Image-to-Video |
| โ | Wan 2.1 VACE 1.3B | โ | Wan 2.2 Animate 14B | โ | CogVideoX Text-to-Video |
| โ | Wan 2.1 VACE 14B | โ | Wan 2.2-Fun A14B Control | โ | CogVideoX Image-to-Video |
| โ | Wan 2.1-Fun 1.3B Control | โ | Wan 2.2-Fun A14B Control-Camera | โ | EasyAnimate V5 Text-to-Video 12B |
| โ | Wan 2.1-Fun 1.3B InP | โ | SkyReels-V2 Text-to-Video 14B | โ | LTX-Video Text-to-Video |
| โ | Wan 2.1-Fun V1.1 1.3B Control | โ | SkyReels-V2 Image-to-Video 14B | โ | LTX-Video Image-to-Video |
| โ | Wan 2.1-Fun V1.1 1.3B Control-Camera | โ | MoVA 720P | โ | LTX-2 |
| โ | Wan 2.1-Fun V1.1 14B Control | โ | LongCat-Video | โ | Mochi-1 |
| โ | Wan 2.1-Fun V1.1 14B Control-Camera | โ | Krea Realtime Video 14B | โ | Allegro |
| โ | Wan 2.1 Speed-Control 1.3B |
DiffSynth-Studio
Supported 30 models:
| Model | Model | Model | |||
|---|---|---|---|---|---|
| โ | Wan 2.1 Text-to-Video 1.3B | โ | Wan 2.1-Fun V1.1 1.3B Control | โ | Wan 2.2 Animate 14B |
| โ | Wan 2.1 Text-to-Video 14B | โ | Wan 2.1-Fun V1.1 1.3B Control-Camera | โ | Wan 2.2 Dancer 14B |
| โ | Wan 2.1 Image-to-Video 14B 480P | โ | Wan 2.1-Fun V1.1 14B Control | โ | Wan 2.2-Fun A14B Control |
| โ | Wan 2.1 Image-to-Video 14B 720P | โ | Wan 2.1-Fun V1.1 14B Control-Camera | โ | Wan 2.2-Fun A14B Control-Camera |
| โ | Wan 2.1 First-Last-Frame-to-Video 14B 720P | โ | Wan 2.1 VACE 1.3B | โ | LongCat-Video |
| โ | Wan 2.1 Speed-Control 1.3B | โ | Wan 2.1 VACE 14B | โ | Video-as-Prompt Wan 2.1 14B |
| โ | Wan 2.1-Fun 1.3B Control | โ | Wan 2.2 Text-to-Video A14B | โ | Krea Realtime Video 14B |
| โ | Wan 2.1-Fun 1.3B InP | โ | Wan 2.2 Image-to-Video A14B | โ | MoVA 720P |
| โ | Wan 2.1-Fun 14B Control | โ | Wan 2.2 Text/Image-to-Video 5B | โ | LTX-2 |
| โ | Wan 2.1-Fun 14B InP | โ | Wan 2.2 Speech-to-Video 14B | โ | LTX-2.3 |
๐ License
Apache-2.0
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
sparsevideo-0.1.0.tar.gz
(3.2 MB
view details)
File details
Details for the file sparsevideo-0.1.0.tar.gz.
File metadata
- Download URL: sparsevideo-0.1.0.tar.gz
- Upload date:
- Size: 3.2 MB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.2.0 CPython/3.12.9
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
0a55a81da2eecebdde412ea7c261b29d30c4f6dedee6fe1e7489ae4e801bec31
|
|
| MD5 |
5d9d3e622cf9786c74b5012a1f3ffa5f
|
|
| BLAKE2b-256 |
7dcf81a4d8f7a57fe50527cb6af48b436ebe8cd80620e968461270b329220c85
|