Skip to main content

⚡️🎉A PyTorch-native Inference Engine with Cache,
Parallelism, Quantization and CPU Offload for DiTs
Featured|HelloGitHub

**🤗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

©️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

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distributions

If you're not sure about the file name format, learn more about wheel file names.

cache_dit_cu13-1.5.1-cp314-cp314-manylinux_2_38_x86_64.whl (33.2 MB view details)

Uploaded CPython 3.14manylinux: glibc 2.38+ x86-64

cache_dit_cu13-1.5.1-cp313-cp313-manylinux_2_38_x86_64.whl (33.2 MB view details)

Uploaded CPython 3.13manylinux: glibc 2.38+ x86-64

cache_dit_cu13-1.5.1-cp312-cp312-manylinux_2_38_x86_64.whl (33.1 MB view details)

Uploaded CPython 3.12manylinux: glibc 2.38+ x86-64

cache_dit_cu13-1.5.1-cp311-cp311-manylinux_2_38_x86_64.whl (33.1 MB view details)

Uploaded CPython 3.11manylinux: glibc 2.38+ x86-64

cache_dit_cu13-1.5.1-cp310-cp310-manylinux_2_38_x86_64.whl (33.1 MB view details)

Uploaded CPython 3.10manylinux: glibc 2.38+ x86-64

File details

Details for the file cache_dit_cu13-1.5.1-cp314-cp314-manylinux_2_38_x86_64.whl.

File metadata

File hashes

Hashes for cache_dit_cu13-1.5.1-cp314-cp314-manylinux_2_38_x86_64.whl
Algorithm Hash digest
SHA256 0914a11afd60ca1583b3e67a08d3b8949b1971790567851853db0589505c551a
MD5 05bc43996f8610916219881ad4339388
BLAKE2b-256 4bf6bd2867f1f83c41b45f0d405067a781d4dbe32a3adca83c38e3b5fc7655f8

See more details on using hashes here.

File details

Details for the file cache_dit_cu13-1.5.1-cp313-cp313-manylinux_2_38_x86_64.whl.

File metadata

File hashes

Hashes for cache_dit_cu13-1.5.1-cp313-cp313-manylinux_2_38_x86_64.whl
Algorithm Hash digest
SHA256 d4975e3c8e4fa2095b27178330710f8effed42c7d70047b65463f4d73436200d
MD5 fd44f51c794158b92ef652aca2f50892
BLAKE2b-256 53f4c28e1b23a7ce318a360d1f82005bc17d5fd84f30e00e7aa9bd5263a571ba

See more details on using hashes here.

File details

Details for the file cache_dit_cu13-1.5.1-cp312-cp312-manylinux_2_38_x86_64.whl.

File metadata

File hashes

Hashes for cache_dit_cu13-1.5.1-cp312-cp312-manylinux_2_38_x86_64.whl
Algorithm Hash digest
SHA256 2aae9ba8ffa261f600b6ddd7683a6b2283982e7b5bb07c66e69fc9ff475103df
MD5 e8110bdfa04d6d902070ce7c9989adf7
BLAKE2b-256 46fe4a6a22be65a9cde47587244ea0bd753726647ca2d336f355919605596033

See more details on using hashes here.

File details

Details for the file cache_dit_cu13-1.5.1-cp311-cp311-manylinux_2_38_x86_64.whl.

File metadata

File hashes

Hashes for cache_dit_cu13-1.5.1-cp311-cp311-manylinux_2_38_x86_64.whl
Algorithm Hash digest
SHA256 a55032c8ae0bcf600775303cc64e4b928ff80b90059d665990bf42149632bd1b
MD5 b8d7f979171c6fdaa8ba9b591c63a384
BLAKE2b-256 4092937bc8d9cf7266256e0dea86ae67ca71f18ea703dfd4942bbb5f005b4f66

See more details on using hashes here.

File details

Details for the file cache_dit_cu13-1.5.1-cp310-cp310-manylinux_2_38_x86_64.whl.

File metadata

File hashes

Hashes for cache_dit_cu13-1.5.1-cp310-cp310-manylinux_2_38_x86_64.whl
Algorithm Hash digest
SHA256 3dc1a91cc26c8101ce0a205c6fbcd04cb67448ccdba52c07f1b5cb6138dd7d29
MD5 f38fae56490e7465e2cbdff11b386382
BLAKE2b-256 747a3006984d3a29c0cbfea1ef534cb4c508cace45f4b9c58d4986aa529b976b

See more details on using hashes here.

Release history Release notifications | RSS feed

This release

1.5.1 This release

5 files

1.5.0

10 files

1.3.12

10 files

1.3.11

10 files

1.3.10

5 files

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page