shawnricecake / draft-attentionLinks
Code for Draft Attention
☆32Updated 2 weeks ago
Alternatives and similar repositories for draft-attention
Users that are interested in draft-attention are comparing it to the libraries listed below
Sorting:
- ☆13Updated 2 months ago
- The code repository of "MBQ: Modality-Balanced Quantization for Large Vision-Language Models"☆41Updated 2 months ago
- [CVPR 2025] Q-DiT: Accurate Post-Training Quantization for Diffusion Transformers☆50Updated 9 months ago
- QuEST: Efficient Finetuning for Low-bit Diffusion Models☆45Updated 4 months ago
- [ECCV24] MixDQ: Memory-Efficient Few-Step Text-to-Image Diffusion Models with Metric-Decoupled Mixed Precision Quantization☆42Updated 6 months ago
- [ICLR 2025] Mixture Compressor for Mixture-of-Experts LLMs Gains More☆45Updated 3 months ago
- [ICLR'25] ViDiT-Q: Efficient and Accurate Quantization of Diffusion Transformers for Image and Video Generation☆98Updated 2 months ago
- [ICLR 2024 Spotlight] This is the official PyTorch implementation of "EfficientDM: Efficient Quantization-Aware Fine-Tuning of Low-Bit Di…☆60Updated last year
- This is the official PyTorch implementation of "ZipAR: Accelerating Auto-regressive Image Generation through Spatial Locality"☆47Updated 2 months ago
- [ICLR 2024] This is the official PyTorch implementation of "QLLM: Accurate and Efficient Low-Bitwidth Quantization for Large Language Mod…☆38Updated last year
- Code implementation of GPTAQ (https://arxiv.org/abs/2504.02692)☆46Updated this week
- ☆167Updated 4 months ago
- torch_quantizer is a out-of-box quantization tool for PyTorch models on CUDA backend, specially optimized for Diffusion Models.☆22Updated last year
- [CVPR 2024 Highlight] This is the official PyTorch implementation of "TFMQ-DM: Temporal Feature Maintenance Quantization for Diffusion Mo…☆63Updated 10 months ago
- [NeurIPS 2024] Learning-to-Cache: Accelerating Diffusion Transformer via Layer Caching☆104Updated 10 months ago
- [NeurIPS 2024] The official implementation of ZipCache: Accurate and Efficient KV Cache Quantization with Salient Token Identification☆21Updated 2 months ago
- FastCache: Fast Caching for Diffusion Transformer Through Learnable Linear Approximation [Efficient ML Model]☆21Updated this week
- This is the official repo for the paper "Accelerating Parallel Sampling of Diffusion Models" Tang et al. ICML 2024 https://openreview.net…☆14Updated 10 months ago
- XAttention: Block Sparse Attention with Antidiagonal Scoring☆158Updated 3 weeks ago
- FORA introduces simple yet effective caching mechanism in Diffusion Transformer Architecture for faster inference sampling.☆46Updated 10 months ago
- Curated list of methods that focuses on improving the efficiency of diffusion models☆45Updated 10 months ago
- A WebUI for Side-by-Side Comparison of Media (Images/Videos) Across Multiple Folders☆24Updated 3 months ago
- The official implementation of PTQD: Accurate Post-Training Quantization for Diffusion Models☆98Updated last year
- Official PyTorch implementation of "Quantization Hurts Reasoning? An Empirical Study on Quantized Reasoning Models"☆34Updated last week
- Patch convolution to avoid large GPU memory usage of Conv2D☆87Updated 4 months ago
- An auxiliary project analysis of the characteristics of KV in DiT Attention.☆30Updated 6 months ago
- Triton implement of bi-directional (non-causal) linear attention☆48Updated 4 months ago
- PyTorch implementation of PTQ4DiT https://arxiv.org/abs/2405.16005☆30Updated 6 months ago
- Official PyTorch implementation of the paper "dLLM-Cache: Accelerating Diffusion Large Language Models with Adaptive Caching" (dLLM-Cache…☆72Updated this week
- [EMNLP 2024] RoLoRA: Fine-tuning Rotated Outlier-free LLMs for Effective Weight-Activation Quantization☆37Updated 8 months ago