horseee / learning-to-cacheLinks
[NeurIPS 2024] Learning-to-Cache: Accelerating Diffusion Transformer via Layer Caching
☆111Updated last year
Alternatives and similar repositories for learning-to-cache
Users that are interested in learning-to-cache are comparing it to the libraries listed below
Sorting:
- FORA introduces simple yet effective caching mechanism in Diffusion Transformer Architecture for faster inference sampling.☆48Updated last year
- [ICML 2025] This is the official PyTorch implementation of "ZipAR: Accelerating Auto-regressive Image Generation through Spatial Locality…☆52Updated 5 months ago
- Adaptive Caching for Faster Video Generation with Diffusion Transformers☆156Updated 9 months ago
- ☆173Updated 7 months ago
- Locality-aware Parallel Decoding for Efficient Autoregressive Image Generation☆67Updated last month
- SpeeD: A Closer Look at Time Steps is Worthy of Triple Speed-Up for Diffusion Model Training☆183Updated 6 months ago
- [ICLR 2024 Spotlight] This is the official PyTorch implementation of "EfficientDM: Efficient Quantization-Aware Fine-Tuning of Low-Bit Di…☆64Updated last year
- Curated list of methods that focuses on improving the efficiency of diffusion models☆46Updated last year
- Code for Draft Attention☆90Updated 3 months ago
- [CVPR 2025] CoDe: Collaborative Decoding Makes Visual Auto-Regressive Modeling Efficient☆106Updated 4 months ago
- Official implementation of paper "VMoBA: Mixture-of-Block Attention for Video Diffusion Models"☆38Updated last month
- [CVPR 2024 Highlight & TPAMI 2025] This is the official PyTorch implementation of "TFMQ-DM: Temporal Feature Maintenance Quantization for…☆103Updated last month
- [CVPR 2025 Highlight] TinyFusion: Diffusion Transformers Learned Shallow☆134Updated 4 months ago
- [CVPR 2025] Q-DiT: Accurate Post-Training Quantization for Diffusion Transformers☆55Updated 11 months ago
- ☆84Updated 4 months ago
- Autoregressive Image Generation with Randomized Parallel Decoding☆71Updated 4 months ago
- ☆89Updated 3 months ago
- Inference-only implementation of "One-Step Diffusion Distillation through Score Implicit Matching" [NIPS 2024]☆82Updated 9 months ago
- TerDiT: Ternary Diffusion Models with Transformers☆71Updated last year
- ✈️ [ICCV 2025] Towards Stabilized and Efficient Diffusion Transformers through Long-Skip-Connections with Spectral Constraints☆72Updated last month
- [ICCV 2025] QuEST: Efficient Finetuning for Low-bit Diffusion Models☆52Updated last month
- [ICLR 2025] Official PyTorch implmentation of paper "T-Stitch: Accelerating Sampling in Pre-trained Diffusion Models with Trajectory Stit…☆103Updated last year
- An open-source implementation of Regional Adaptive Sampling (RAS), a novel diffusion model sampling strategy that introduces regional var…☆138Updated last month
- This is the official repo for the paper "Accelerating Parallel Sampling of Diffusion Models" Tang et al. ICML 2024 https://openreview.net…☆15Updated last year
- [ICML2025] Sparse VideoGen: Accelerating Video Diffusion Transformers with Spatial-Temporal Sparsity☆412Updated 2 months ago
- [ECCV 2024] Official pytorch implementation of "Switch Diffusion Transformer: Synergizing Denoising Tasks with Sparse Mixture-of-Experts"☆44Updated last year
- [NeurIPS 2023] Structural Pruning for Diffusion Models☆198Updated last year
- [CVPR 2025] Exploring the Deep Fusion of Large Language Models and Diffusion Transformers for Text-to-Image Synthesis☆112Updated 3 months ago
- [Preprint] GMem: A Modular Approach for Ultra-Efficient Generative Models☆39Updated 5 months ago
- The official implementation for "MonoFormer: One Transformer for Both Diffusion and Autoregression"☆86Updated 10 months ago