SolitaryThinker 4d04c1b01c [refactor] magi-human DiT: restore dtype-agnostic attention via orig_dtype
Wave 14b's bit-exact dtype boundary fix introduced two violations of
Wave 10's dtype-agnostic pattern in MagiAttention:

  1. q/k/v.to(torch.bfloat16) hardcoded before SDPA
  2. out.float() explicit upcast after attention to fp32 the gating

Both turn out to be unnecessary:

  1. orig_dtype = self.linear_qkv.weight.dtype already evaluates to
     bf16 in production (loader sets bf16) AND in the parity test
     (PackedExpertLinear's __init__ default is bf16, matching upstream
     BaseLinear at dit_module.py:330). So q.to(orig_dtype) gives the
     same bf16 cast as the hardcode without the dtype lock-in.

  2. PyTorch's float type-promotion rules already handle the
     bf16 -> fp32 transition implicitly: bf16_out * sigmoid(fp32_g)
     promotes to fp32, exactly matching upstream's intentional
     bf16 * fp32 -> fp32 boundary at dit_module.py:649. The explicit
     .float() was redundant.

Result: 11 insertions, 14 deletions, attention block reads more
cleanly without sacrificing any of the parity work:

  - All 7 magi-human local parity tests pass with bit-exact (0.0/0.0)
    or near-bit-exact (VAE 8e-4) outputs vs upstream
  - basic_magi_human.py E2E produces byte-identical mp4 (same md5
    hash) as the pre-refactor Wave 14b state, so production
    behavior is unchanged

The MagiAttention block stays custom (rather than reusing FV's
LocalAttention) because LocalAttention requires a forward_context
that the parity test does not establish; switching would force
either expanded test scaffolding or a parity-validation regression.
2026-05-05 10:55:49 -07:00
2026-05-05 10:55:49 -07:00
2026-02-23 15:14:42 -08:00
2026-04-02 16:07:24 -07:00
2026-04-02 16:07:24 -07:00
2026-04-02 16:07:24 -07:00
2026-02-10 01:11:17 -08:00
2025-02-20 11:33:58 -08:00
2026-03-17 14:26:07 -07:00
2025-11-27 00:37:03 -08:00

| Documentation | Quick Start | Weekly Dev Meeting | 🟣💬 Slack | 🟣💬 WeChat |

FastVideo is a unified post-training and real-time inference framework for accelerated video generation.

NEWS

More News

Key Features

FastVideo has the following features:

  • End-to-end post-training support for bidirectional and autoregressive models:
    • Support full finetuning and LoRA finetuning for state-of-the-art open video DiTs
    • Data preprocessing pipeline for video, image, and text data
    • Distribution Matching Distillation (DMD2) stepwise distillation.
    • Sparse attention with Video Sparse Attention
    • Sparse distillation to achieve >50x denoising speedup
    • Scalable training with FSDP2, sequence parallelism, and selective activation checkpointing.
    • Causal distillation through Self-Forcing
    • See this page for full list of supported models and recipes.
  • State-of-the-art performance optimizations for inference
    • Sequence Parallelism for distributed inference
    • Multiple state-of-the-art attention backends
    • User-friendly CLI and Python API
    • See this page for full list of supported optimizations.
  • Diverse hardware and OS support
    • Support H100, A100, 4090
    • Support Linux, Windows, MacOS
    • See this page for full list of supported models, hardware assumptions, and optimization compatibility.

Getting Started

We recommend using uv to create a clean environment. If you previously used Conda, switching to uv generally gives faster and more stable installs.

# Create and activate a new uv environment
uv venv --python 3.12 --seed
source .venv/bin/activate

# Install FastVideo
uv pip install fastvideo

Please see our docs for more detailed installation instructions.

Sparse Distillation

For our sparse distillation techniques, please see our distillation docs and check out our blog.

See below for recipes and datasets:

Model Sparse Distillation Dataset
FastWan2.1-T2V-1.3B Recipe FastVideo Synthetic Wan2.1 480P
FastWan2.2-TI2V-5B Recipe FastVideo Synthetic Wan2.2 720P

Inference

Generating Your First Video

Here's a minimal example to generate a video using the default settings. Make sure VSA kernels are installed. Create a file called example.py with the following code:

import os
from fastvideo import VideoGenerator

def main():
    os.environ["FASTVIDEO_ATTENTION_BACKEND"] = "VIDEO_SPARSE_ATTN"

    # Create a video generator with a pre-trained model
    generator = VideoGenerator.from_pretrained(
        "FastVideo/FastWan2.1-T2V-1.3B-Diffusers",
        num_gpus=1,  # Adjust based on your hardware
    )

    # Define a prompt for your video
    prompt = "A curious raccoon peers through a vibrant field of yellow sunflowers, its eyes wide with interest."

    # Generate the video
    video = generator.generate_video(
        prompt,
        output_path="my_videos/",  # Controls where videos are saved
        save_video=True
    )

if __name__ == '__main__':
    main()

Run the script with:

python example.py

For a more detailed guide, please see our inference quick start.

Awesome work using FastVideo or our research projects

  • SGLang: SGLang's diffusion inference functionality is based on a fork of FastVideo on Sept. 24, 2025.
  • DanceGRPO: A unified framework to adapt Group Relative Policy Optimization (GRPO) to visual generation paradigms. Code based on FastVideo.
  • SRPO: A method to directly align the full diffusion trajectory with fine-grained human preference. Code based on FastVideo.
  • DCM: Dual-expert consistency model for efficient and high-quality video generation. Code based on FastVideo.
  • HY-WorldPlay: An action-conditioned world model model trained using FastVideo framework.
  • Hunyuan Video 1.5: A leading lightweight video generation model, where they proposed SSTA based on Sliding Tile Attention.
  • Kandinsky-5.0: A family of diffusion models for video & image generation, where their NABLA attention includes a Sliding Tile Attention branch.
  • LongCat Video: A foundational video generation model with 13.6B parameters with block-sparse attention similar to Video Sparse Attention.

🤝 Contributing

We welcome all contributions. Please check out our guide here. See details in development roadmap.

Acknowledgement

We learned the design and reused code from the following projects: Wan-Video, ThunderKittens, DMD2, diffusers, xDiT, vLLM, SGLang. We thank MBZUAI, Anyscale, and GMI Cloud for their support throughout this project.

Citation

If you find FastVideo useful, please consider citing our research work:

@article{zhang2025vsa,
  title={Vsa: Faster video diffusion with trainable sparse attention},
  author={Zhang, Peiyuan and Chen, Yongqi and Huang, Haofeng 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}
}
S
Description
No description provided
Readme
442 MiB
Languages
Python 89.8%
TypeScript 5.5%
Cuda 1.9%
C++ 1%
Shell 0.6%
Other 1%