horseee / learning-to-cacheLinks
[NeurIPS 2024] Learning-to-Cache: Accelerating Diffusion Transformer via Layer Caching
☆116Updated 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.☆51Updated last year
- [ICML 2025] This is the official PyTorch implementation of "ZipAR: Accelerating Auto-regressive Image Generation through Spatial Locality…☆53Updated 8 months ago
- Code for our ICCV 2025 paper "Adaptive Caching for Faster Video Generation with Diffusion Transformers"☆161Updated last year
- ☆187Updated 10 months ago
- Official implementation of paper "VMoBA: Mixture-of-Block Attention for Video Diffusion Models"☆55Updated 5 months ago
- [ICLR 2024 Spotlight] This is the official PyTorch implementation of "EfficientDM: Efficient Quantization-Aware Fine-Tuning of Low-Bit Di…☆66Updated last year
- Locality-aware Parallel Decoding for Efficient Autoregressive Image Generation☆80Updated 4 months ago
- ☆89Updated 8 months ago
- [CVPR 2025] CoDe: Collaborative Decoding Makes Visual Auto-Regressive Modeling Efficient☆107Updated 2 months ago
- SpeeD: A Closer Look at Time Steps is Worthy of Triple Speed-Up for Diffusion Model Training☆186Updated 10 months ago
- [CVPR 2024 Highlight & TPAMI 2025] This is the official PyTorch implementation of "TFMQ-DM: Temporal Feature Maintenance Quantization for…☆109Updated 2 months ago
- Code for Draft Attention☆94Updated 6 months ago
- [CVPR 2025] Q-DiT: Accurate Post-Training Quantization for Diffusion Transformers☆73Updated last year
- Autoregressive Image Generation with Randomized Parallel Decoding☆81Updated last month
- [NeurIPS'25] dKV-Cache: The Cache for Diffusion Language Models☆121Updated 6 months ago
- [CVPR 2025 Highlight] TinyFusion: Diffusion Transformers Learned Shallow☆147Updated 7 months ago
- The official implementation of "Sparse-vDiT: Unleashing the Power of Sparse Attention to Accelerate Video Diffusion Transformers" (arXiv …☆49Updated 5 months ago
- Curated list of methods that focuses on improving the efficiency of diffusion models☆44Updated last year
- This is the official repo for the paper "Accelerating Parallel Sampling of Diffusion Models" Tang et al. ICML 2024 https://openreview.net…☆16Updated last year
- ✈️ [ICCV 2025] Towards Stabilized and Efficient Diffusion Transformers through Long-Skip-Connections with Spectral Constraints☆77Updated 4 months ago
- [CVPR 2025] Exploring the Deep Fusion of Large Language Models and Diffusion Transformers for Text-to-Image Synthesis☆127Updated 6 months ago
- The official implementation of "Neighboring Autoregressive Modeling for Efficient Visual Generation"☆58Updated 7 months ago
- [ICCV 2025] QuEST: Efficient Finetuning for Low-bit Diffusion Models☆55Updated 5 months ago
- [ICLR 2025] Distilled Decoding 1: One-step Sampling of Image Auto-regressive Models with Flow Matching☆52Updated 7 months ago
- [ICLR2025] Accelerating Diffusion Transformers with Token-wise Feature Caching☆195Updated 8 months ago
- Inference-only implementation of "One-Step Diffusion Distillation through Score Implicit Matching" [NIPS 2024]☆84Updated last year
- [ICLR 2025] Official PyTorch implmentation of paper "T-Stitch: Accelerating Sampling in Pre-trained Diffusion Models with Trajectory Stit…☆103Updated last year
- [Preprint] GMem: A Modular Approach for Ultra-Efficient Generative Models☆40Updated 8 months ago
- The official implementation for "MonoFormer: One Transformer for Both Diffusion and Autoregression"☆87Updated last year
- [ICLR 2025] Implementation of Accelerating Auto-regressive Text-to-Image Generation with Training-free Speculative Jacobi Decoding☆47Updated 7 months ago