Alternatives and similar repositories for draft-attention

Users that are interested in draft-attention are comparing it to the libraries listed below

• thu-nics / DiTFastAttn
  ☆189Updated last year
• AdaCache-DiT / AdaCache
  Code for our ICCV 2025 paper "Adaptive Caching for Faster Video Generation with Diffusion Transformers"
  ☆164Updated last year
• thu-ml / SLA
  SLA: Beyond Sparsity in Diffusion Transformers via Fine-Tunable Sparse–Linear Attention
  ☆243Updated 3 weeks ago
• Peyton-Chen / Sparse-vDiT
  The official implementation of "Sparse-vDiT: Unleashing the Power of Sparse Attention to Accelerate Video Diffusion Transformers" (arXiv …
  ☆50Updated 7 months ago
• KlingTeam / VMoBA
  Official implementation of paper "VMoBA: Mixture-of-Block Attention for Video Diffusion Models"
  ☆58Updated 6 months ago
• prathebaselva / FORA
  FORA introduces simple yet effective caching mechanism in Diffusion Transformer Architecture for faster inference sampling.
  ☆52Updated last year
• horseee / learning-to-cache
  [NeurIPS 2024] Learning-to-Cache: Accelerating Diffusion Transformer via Layer Caching
  ☆116Updated last year
• TZW1998 / ParaTAA-Diffusion
  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
• xdit-project / DistVAE
  A parallelism VAE avoids OOM for high resolution image generation
  ☆84Updated 5 months ago
• svg-project / Sparse-VideoGen
  [ICML2025, NeurIPS2025 Spotlight] Sparse VideoGen 1 & 2: Accelerating Video Diffusion Transformers with Sparse Attention
  ☆617Updated last month
• Tele-AI / TeleTron
  To pioneer training long-context multi-modal transformer models
  ☆67Updated 5 months ago
• Bujiazi / DiCache
  Official implementation of DiCache: Let Diffusion Model Determine Its Own Cache
  ☆54Updated 3 months ago
• Vchitect / LiteGen
  A light-weight and high-efficient training framework for accelerating diffusion tasks.
  ☆51Updated last year
• ThisisBillhe / ZipAR
  [ICML 2025] This is the official PyTorch implementation of "ZipAR: Accelerating Auto-regressive Image Generation through Spatial Locality…
  ☆53Updated 9 months ago
• thu-nics / ViDiT-Q
  [ICLR'25] ViDiT-Q: Efficient and Accurate Quantization of Diffusion Transformers for Image and Video Generation
  ☆146Updated 9 months ago
• NoakLiu / FastCache-xDiT
  FastCache: Fast Caching for Diffusion Transformer Through Learnable Linear Approximation [Efficient ML Model]
  ☆46Updated last month
• mit-han-lab / Block-Sparse-Attention
  A sparse attention kernel supporting mix sparse patterns
  ☆436Updated last week
• thu-nics / MixDQ
  [ECCV24] MixDQ: Memory-Efficient Few-Step Text-to-Image Diffusion Models with Metric-Decoupled Mixed Precision Quantization
  ☆49Updated last year
• mit-han-lab / lpd
  Locality-aware Parallel Decoding for Efficient Autoregressive Image Generation
  ☆81Updated 6 months ago
• mit-han-lab / patch_conv
  Patch convolution to avoid large GPU memory usage of Conv2D
  ☆95Updated 11 months ago
• Shenyi-Z / ToCa
  [ICLR2025] Accelerating Diffusion Transformers with Token-wise Feature Caching
  ☆206Updated 10 months ago
• ModelTC / TFMQ-DM
  [CVPR 2024 Highlight & TPAMI 2025] This is the official PyTorch implementation of "TFMQ-DM: Temporal Feature Maintenance Quantization for…
  ☆108Updated 3 months ago
• microsoft / RAS
  An open-source implementation of Regional Adaptive Sampling (RAS), a novel diffusion model sampling strategy that introduces regional var…
  ☆150Updated 6 months ago
• Shenyi-Z / Cache4Diffusion
  Aiming to integrate most existing feature caching-based diffusion acceleration schemes into a unified framework.
  ☆83Updated 2 months ago
• ThisisBillhe / NAR
  [ICCV 2025] The official implementation of "Neighboring Autoregressive Modeling for Efficient Visual Generation"
  ☆58Updated 9 months ago
• Juanerx / Q-DiT
  [CVPR 2025] Q-DiT: Accurate Post-Training Quantization for Diffusion Transformers
  ☆73Updated last year
• mit-han-lab / VisCompare
  A WebUI for Side-by-Side Comparison of Media (Images/Videos) Across Multiple Folders
  ☆24Updated 10 months ago
• ziplab / BLADE
  This is the official PyTorch implementation of "BLADE: Block-Sparse Attention Meets Step Distillation for Efficient Video Generation."
  ☆38Updated 3 months ago
• Shenyi-Z / DuCa
  (ToCa-v2) A New version of ToCa，with faster speed and better acceleration!
  ☆38Updated 10 months ago
• Shenyi-Z / TaylorSeer
  [ICCV2025] From Reusing to Forecasting: Accelerating Diffusion Models with TaylorSeers
  ☆358Updated 5 months ago