**🤗Why Cache-DiT❓❓**Cache-DiT is built on top of the 🤗Diffusers library and now supports nearly ALL DiTs from Diffusers. It provides hybrid cache acceleration (DBCache, TaylorSeer, SCM, etc.) and comprehensive parallelism optimizations, including Context Parallelism, Tensor Parallelism, hybrid 2D or 3D parallelism, and dedicated extra parallelism support for Text Encoder, VAE, and ControlNet.
Cache-DiT is compatible with compilation, CPU Offloading, and quantization, fully integrates with SGLang Diffusion, vLLM-Omni, TensorRT-LLM, ComfyUI, and runs natively on NVIDIA GPUs, Ascend NPUs and AMD GPUs. Cache-DiT is fast, easy to use, and flexible for various DiTs (online docs at 📘cache-dit.io, tech report at 📘paper).
⚡️9x speedup by Cache-DiT with Cache, Context Parallelism and Compilation
📋Supported DiT Models
Cache-DiT supports 40+ DiT pipeline families (120+ Variants) from 🤗Diffusers, covering the vast majority of DiT-based pipelines. For full support matrix and detailed usage, please refer to our documentation at 📘cache-dit.io.
| Modality | Pipeline Series | Transformer | Variants | C/P/Q/OF |
|---|---|---|---|---|
| Image | FLUX | FluxTransformer2DModel |
10+ | ✔️/✔️/✔️/✔️ |
| Image | FLUX.2 | Flux2Transformer2DModel |
1 | ✔️/✔️/✔️/✔️ |
| Image | FLUX.2 Klein | Flux2Transformer2DModel |
3 | ✔️/✔️/✔️/✔️ |
| Image | ERNIE-Image | ErnieImageTransformer2DModel |
1 | ✔️/✔️/✔️/✔️ |
| Image | Qwen-Image | QwenImageTransformer2DModel |
9+ | ✔️/✔️/✔️/✔️ |
| Image | Z-Image | ZImageTransformer2DModel |
6 | ✔️/✔️/✔️/✔️ |
| Image | Joy-Image | JoyImageEditTransformer3DModel |
1 | ✔️/✔️/✔️/✔️ |
| Image | Krea-2 | Krea2Transformer2DModel |
1 | ✔️/✔️/✔️/✔️ |
| Image | Bria-Fibo | BriaFiboTransformer2DModel |
2 | ✔️/✔️/✔️/✔️ |
| Image | GLM-Image | GlmImageTransformer2DModel |
1 | ✔️/✔️/✔️/✔️ |
| Image | LongCat-Image | LongCatImageTransformer2DModel |
2 | ✔️/✔️/✔️/✔️ |
| Image | Ovis-Image | OvisImageTransformer2DModel |
1 | ✔️/✔️/✔️/✔️ |
| Image | SD3 | SD3Transformer2DModel |
3 | ✔️/✖️/✔️/✔️ |
| Image | PixArt-Alpha | PixArtTransformer2DModel |
1 | ✔️/✔️/✔️/✔️ |
| Image | PixArt-Sigma | PixArtTransformer2DModel |
1 | ✔️/✔️/✔️/✔️ |
| Image | Sana (image) | SanaTransformer2DModel |
4 | ✔️/✖️/✔️/✔️ |
| Image | DiT | DiTTransformer2DModel |
1 | ✔️/✔️/✔️/✔️ |
| Image | HunyuanDiT | HunyuanDiT2DModel |
1 | ✔️/✔️/✔️/✔️ |
| Image | HunyuanDiT-PAG | HunyuanDiT2DModel |
1 | ✔️/✔️/✔️/✔️ |
| Image | AuraFlow | AuraFlowTransformer2DModel |
1 | ✔️/✖️/✔️/✔️ |
| Image | CogView4 | CogView4Transformer2DModel |
2 | ✔️/✔️/✔️/✔️ |
| Image | CogView3Plus | CogView3PlusTransformer2DModel |
1 | ✔️/✔️/✔️/✔️ |
| Image | Hunyuan-Image | HunyuanImageTransformer2DModel |
2 | ✔️/✔️/✔️/✔️ |
| Image | HiDream | HiDreamImageTransformer2DModel |
1 | ✔️/✖️/✔️/✔️ |
| Image | Bria | BriaTransformer2DModel |
1 | ✔️/✖️/✔️/✔️ |
| Image | Chroma | ChromaTransformer2DModel |
3 | ✔️/✔️/✔️/✔️ |
| Image | PRX | PRXTransformer2DModel |
2 | ✔️/✖️/✔️/✔️ |
| Image | OmniGen | OmniGenTransformer2DModel |
1 | ✔️/✖️/✔️/✔️ |
| Image | Lumina | LuminaNextDiT2DModel |
2 | ✔️/✖️/✔️/✔️ |
| Image | Lumina 2 | Lumina2Transformer2DModel |
2 | ✔️/✖️/✔️/✔️ |
| Image | VisualCloze | FluxTransformer2DModel |
2 | ✔️/✔️/✔️/✔️ |
| Image | Amused | UVit2DModel |
1 | ✔️/✖️/✔️/✔️ |
| Video | CogVideoX | CogVideoXTransformer3DModel |
4 | ✔️/✔️/✔️/✔️ |
| Video | Wan (T2V/I2V) | WanTransformer3DModel |
3 | ✔️/✔️/✔️/✔️ |
| Video | Any-Flow | AnyFlowTransformer3DModel |
1 | ✔️/✔️/✔️/✔️ |
| Video | Any-Flow-FAR | AnyFlowFARTransformer3DModel |
1 | ✖️/✖️/✔️/✔️ |
| Video | Wan-VACE | WanVACETransformer3DModel |
1 | ✔️/✔️/✔️/✔️ |
| Video | HunyuanVideo | HunyuanVideoTransformer3DModel |
4 | ✔️/✔️/✔️/✔️ |
| Video | HunyuanVideo 1.5 | HunyuanVideo15Transformer3DModel |
2 | ✔️/✖️/✔️/✔️ |
| Video | Mochi | MochiTransformer3DModel |
1 | ✔️/✔️/✔️/✔️ |
| Video | Allegro | AllegroTransformer3DModel |
1 | ✔️/✖️/✔️/✔️ |
| Video | EasyAnimate | EasyAnimateTransformer3DModel |
3 | ✔️/✖️/✔️/✔️ |
| Video | ConsisID | ConsisIDTransformer3DModel |
1 | ✔️/✔️/✔️/✔️ |
| Video | Cosmos | CosmosTransformer3DModel |
7+ | ✔️/✖️/✔️/✔️ |
| Video | Cosmos3 (Omni) | Cosmos3OmniTransformer |
2 | ✔️/✖️/✔️/✔️ |
| Video | LTX | LTXVideoTransformer3DModel |
5 | ✔️/✔️/✔️/✔️ |
| Video | LTX2 | LTX2VideoTransformer3DModel |
6 | ✔️/✔️/✔️/✔️ |
| Video | Helios | HeliosTransformer3DModel |
2 | ✔️/✔️/✔️/✔️ |
| Video | ChronoEdit | ChronoEditTransformer3DModel |
1 | ✔️/✔️/✔️/✔️ |
| Video | SkyReelsV2 | SkyReelsV2Transformer3DModel |
5 | ✔️/✔️/✔️/✔️ |
| Video | Kandinsky5 (T2V/I2V) | Kandinsky5Transformer3DModel |
4 | ✔️/✔️/✔️/✔️ |
| Audio | StableAudio | StableAudioDiTModel |
1 | ✔️/✖️/✔️/✔️ |
| 3D/Other | Shap-E | PriorTransformer |
1 | ✔️/✖️/✔️/✔️ |
| Other | LucyEdit | WanTransformer3DModel |
1 | ✔️/✔️/✔️/✔️ |
C: Hybrid Cache (DBCache + Calibrator: TaylorSeer/DMD/FoCa/SCM); P: Parallelism (Ulysses/Ring/USP/TP
TE-P/VAE-P/2D-P/3D-P); Q: Quantization (W8A8, W4A4); OF: Bucket-style Layerwise Offload (~0 overhead)
🤖Agentic Workflows
Cache-DiT provides a model-integration SKILL to help users integrate new DiT pipelines into Cache-DiT, including Cache, CP, TP, TE-P, VAE-P and carefully designed test cases. Users can use it with Coding Agents, e.g, GitHub Copilot, Claude Code, Open Code.
[!NOTE] Please note that quantization and layerwise offload in Cache-DiT are generally supported for nn.Module, thus no extra integration is needed for new DiT pipelines or transformers.
🌐Community Integration
- 🎉MindIE-SD x Cache-DiT
- 🎉xDiT x Cache-DiT
- 🎉ComfyUI x Cache-DiT
- 🎉(Intel) llm-scaler x Cache-DiT
- 🎉Diffusers x Cache-DiT
- 🎉TensorRT-LLM x Cache-DiT
- 🎉SGLang Omni x Cache-DiT
- 🎉SGLang Diffusion x Cache-DiT
- 🎉vLLM-Omni x Cache-DiT
- 🎉Nunchaku x Cache-DiT
- 🎉SD.Next x Cache-DiT
- 🎉stable-diffusion.cpp x Cache-DiT
- 🎉jetson-containers x Cache-DiT
- 🎉ComfyUI-RH-MiniMax-H3 x Cache-DiT
©️Acknowledgements
Special thanks to vipshop's Computer Vision AI Team for supporting testing and deployment of this project. We learned and reused codes from: Diffusers, SGLang, vLLM-Omni, Nunchaku, xDiT and TaylorSeer.
©️Citations
@misc{cache-dit@2025,
title={Cache-DiT: A PyTorch-native Inference Engine with Cache, Parallelism, Quantization and CPU Offload for DiTs.},
url={https://github.com/vipshop/cache-dit.git},
note={Open-source software available at https://github.com/vipshop/cache-dit.git},
author={DefTruth, vipshop.com, etc.},
year={2025}
}
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distributions
Built Distributions
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file cache_dit_cu13-1.5.1-cp314-cp314-manylinux_2_38_x86_64.whl.
File metadata
- Download URL: cache_dit_cu13-1.5.1-cp314-cp314-manylinux_2_38_x86_64.whl
- Upload date:
- Size: 33.2 MB
- Tags: CPython 3.14, manylinux: glibc 2.38+ x86-64
- Uploaded using Trusted Publishing? No
- Uploaded via:
twine/6.2.0 CPython/3.12.13
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
0914a11afd60ca1583b3e67a08d3b8949b1971790567851853db0589505c551a
|
|
| MD5 |
05bc43996f8610916219881ad4339388
|
|
| BLAKE2b-256 |
4bf6bd2867f1f83c41b45f0d405067a781d4dbe32a3adca83c38e3b5fc7655f8
|
File details
Details for the file cache_dit_cu13-1.5.1-cp313-cp313-manylinux_2_38_x86_64.whl.
File metadata
- Download URL: cache_dit_cu13-1.5.1-cp313-cp313-manylinux_2_38_x86_64.whl
- Upload date:
- Size: 33.2 MB
- Tags: CPython 3.13, manylinux: glibc 2.38+ x86-64
- Uploaded using Trusted Publishing? No
- Uploaded via:
twine/6.2.0 CPython/3.12.13
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
d4975e3c8e4fa2095b27178330710f8effed42c7d70047b65463f4d73436200d
|
|
| MD5 |
fd44f51c794158b92ef652aca2f50892
|
|
| BLAKE2b-256 |
53f4c28e1b23a7ce318a360d1f82005bc17d5fd84f30e00e7aa9bd5263a571ba
|
File details
Details for the file cache_dit_cu13-1.5.1-cp312-cp312-manylinux_2_38_x86_64.whl.
File metadata
- Download URL: cache_dit_cu13-1.5.1-cp312-cp312-manylinux_2_38_x86_64.whl
- Upload date:
- Size: 33.1 MB
- Tags: CPython 3.12, manylinux: glibc 2.38+ x86-64
- Uploaded using Trusted Publishing? No
- Uploaded via:
twine/6.2.0 CPython/3.12.13
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
2aae9ba8ffa261f600b6ddd7683a6b2283982e7b5bb07c66e69fc9ff475103df
|
|
| MD5 |
e8110bdfa04d6d902070ce7c9989adf7
|
|
| BLAKE2b-256 |
46fe4a6a22be65a9cde47587244ea0bd753726647ca2d336f355919605596033
|
File details
Details for the file cache_dit_cu13-1.5.1-cp311-cp311-manylinux_2_38_x86_64.whl.
File metadata
- Download URL: cache_dit_cu13-1.5.1-cp311-cp311-manylinux_2_38_x86_64.whl
- Upload date:
- Size: 33.1 MB
- Tags: CPython 3.11, manylinux: glibc 2.38+ x86-64
- Uploaded using Trusted Publishing? No
- Uploaded via:
twine/6.2.0 CPython/3.12.13
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
a55032c8ae0bcf600775303cc64e4b928ff80b90059d665990bf42149632bd1b
|
|
| MD5 |
b8d7f979171c6fdaa8ba9b591c63a384
|
|
| BLAKE2b-256 |
4092937bc8d9cf7266256e0dea86ae67ca71f18ea703dfd4942bbb5f005b4f66
|
File details
Details for the file cache_dit_cu13-1.5.1-cp310-cp310-manylinux_2_38_x86_64.whl.
File metadata
- Download URL: cache_dit_cu13-1.5.1-cp310-cp310-manylinux_2_38_x86_64.whl
- Upload date:
- Size: 33.1 MB
- Tags: CPython 3.10, manylinux: glibc 2.38+ x86-64
- Uploaded using Trusted Publishing? No
- Uploaded via:
twine/6.2.0 CPython/3.12.13
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
3dc1a91cc26c8101ce0a205c6fbcd04cb67448ccdba52c07f1b5cb6138dd7d29
|
|
| MD5 |
f38fae56490e7465e2cbdff11b386382
|
|
| BLAKE2b-256 |
747a3006984d3a29c0cbfea1ef534cb4c508cace45f4b9c58d4986aa529b976b
|