vipshop / cache-ditLinks
π€A PyTorch-native Inference Engine with Hybrid Cache Acceleration and Parallelism for DiTs: Z-Image, FLUX2, Qwen-Image, etc.
β804Updated this week
Alternatives and similar repositories for cache-dit
Users that are interested in cache-dit are comparing it to the libraries listed below
Sorting:
- πA curated list of Awesome Diffusion Inference Papers with Codes: Sampling, Cache, Quantization, Parallelism, etc.πβ478Updated last month
- https://wavespeed.ai/ Context parallel attention that accelerates DiT model inference with dynamic cachingβ407Updated 5 months ago
- [ICML2025] SpargeAttention: A training-free sparse attention that accelerates any model inference.β852Updated last week
- Model Compression Toolbox for Large Language Models and Diffusion Modelsβ722Updated 4 months ago
- A Distributed Attention Towards Linear Scalability for Ultra-Long Context, Heterogeneous Data Trainingβ590Updated last week
- [ICML2025, NeurIPS2025 Spotlight] Sparse VideoGen 1 & 2: Accelerating Video Diffusion Transformers with Sparse Attentionβ605Updated 2 weeks ago
- β186Updated 11 months ago
- [ICCV2025] From Reusing to Forecasting: Accelerating Diffusion Models with TaylorSeersβ347Updated 4 months ago
- A parallelism VAE avoids OOM for high resolution image generationβ84Updated 4 months ago
- [CVPR 2024 Highlight] DistriFusion: Distributed Parallel Inference for High-Resolution Diffusion Modelsβ715Updated last year
- [NeurIPS 2025] Radial Attention: O(nlogn) Sparse Attention with Energy Decay for Long Video Generationβ567Updated last month
- Aiming to integrate most existing feature caching-based diffusion acceleration schemes into a unified framework.β82Updated 2 months ago
- High performance inference engine for diffusion modelsβ101Updated 3 months ago
- π Collection of awesome generation acceleration resources.β372Updated 5 months ago
- A sparse attention kernel supporting mix sparse patternsβ408Updated last week
- β330Updated this week
- Combining Teacache with xDiT to Accelerate Visual Generation Modelsβ32Updated 8 months ago
- [ICLR'25] ViDiT-Q: Efficient and Accurate Quantization of Diffusion Transformers for Image and Video Generationβ142Updated 9 months ago
- [ICLR2025] Accelerating Diffusion Transformers with Token-wise Feature Cachingβ197Updated 9 months ago
- β440Updated 4 months ago
- An out-of-the-box inference acceleration engine for Diffusion and DiT modelsβ60Updated 9 months ago
- SLA: Beyond Sparsity in Diffusion Transformers via Fine-Tunable SparseβLinear Attentionβ199Updated this week
- Code for Draft Attentionβ98Updated 7 months ago
- Light Video Generation Inference Frameworkβ1,336Updated this week
- Code for our ICCV 2025 paper "Adaptive Caching for Faster Video Generation with Diffusion Transformers"β163Updated last year
- End-to-end recipes for optimizing diffusion models with torchao and diffusers (inference and FP8 training).β388Updated 6 months ago
- USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inferenceβ618Updated this week
- An open-source implementation of Regional Adaptive Sampling (RAS), a novel diffusion model sampling strategy that introduces regional varβ¦β149Updated 6 months ago
- To pioneer training long-context multi-modal transformer modelsβ64Updated 4 months ago
- [ICLR 2025] FasterCache: Training-Free Video Diffusion Model Acceleration with High Qualityβ256Updated last year