Skip to content

Repository files navigation

UnStep

Training-Free Acceleration of Causal Video Diffusion

50 FPS on a single H100   ·   64 FPS on GB200
Default two-step wrapper   ·   832 × 480   ·   No retraining
One step: 60 FPS on H100 and 77 FPS on GB200. Speeds rounded as in the paper.

UnStep accelerates existing Self Forcing and Causal Forcing checkpoints at inference. It combines fewer denoising steps and a smaller KV attention window with clean-cache refinement, truncated SVD, and an optimized DiT/VAE runtime stack.

Method

UnStep overview: runtime optimization, compute reduction, and quality recovery.

Start here

Each folder contains its own instructions and relevant files. Start with installation, then generation and evaluation.

Task Where to go
Install dependencies and download models and prompts installation/
Generate videos generation/
Evaluate with VBench evaluation/
Measure speed and run ablations benchmarks/
Run correctness checks tests/

Samples

Selected frames from five-second videos generated by Self Forcing + UnStep, using the default two-step wrapper.

A corgi in a cyberpunk park

A corgi moving through a neon-lit park at sunset.

A rabbit exploring a fantasy landscape

A rabbit in a purple robe walking through a colorful fantasy landscape.

A raccoon playing guitar in a boat

A raccoon strumming a guitar in a boat on the ocean.

Acknowledgments

UnStep builds on Self Forcing, including its Wan2.1 implementation, and uses VBench for evaluation. We thank their authors for releasing their code. Code attribution and licenses are listed in THIRD_PARTY_NOTICES.md.

Contributing

See CONTRIBUTING.md and our Code of Conduct.

License

CC BY-NC 4.0. Third-party code and model weights retain their own licenses.

About

Official implementation of the paper UnStep Training-Free Acceleration of Causal Video Diffusion with Fewer Steps Than Distillation

Resources

Code of conduct

Contributing

Security policy

Stars

2 stars

Watchers

0 watching

Forks

Releases

Packages

Used by

Contributors

Languages