ThisisBillhe / torch_quantizer
torch_quantizer is a out-of-box quantization tool for PyTorch models on CUDA backend, specially optimized for Diffusion Models.
☆22Updated last year
Alternatives and similar repositories for torch_quantizer:
Users that are interested in torch_quantizer are comparing it to the libraries listed below
- [ICLR 2024] This is the official PyTorch implementation of "QLLM: Accurate and Efficient Low-Bitwidth Quantization for Large Language Mod…☆36Updated last year
- [CVPR 2025] Q-DiT: Accurate Post-Training Quantization for Diffusion Transformers☆47Updated 8 months ago
- The code repository of "MBQ: Modality-Balanced Quantization for Large Vision-Language Models"☆38Updated last month
- ☆13Updated last month
- [ICLR 2025] Mixture Compressor for Mixture-of-Experts LLMs Gains More☆39Updated 2 months ago
- This is the official PyTorch implementation of "ZipAR: Accelerating Auto-regressive Image Generation through Spatial Locality"☆46Updated last month
- Pytorch implementation of our paper accepted by NeurIPS 2022 -- Learning Best Combination for Efficient N:M Sparsity☆17Updated 2 years ago
- Learnable Semi-structured Sparsity for Vision Transformers and Diffusion Transformers☆11Updated 2 months ago
- [NeurIPS 2023] ShiftAddViT: Mixture of Multiplication Primitives Towards Efficient Vision Transformer☆32Updated last year
- [ECCV24] MixDQ: Memory-Efficient Few-Step Text-to-Image Diffusion Models with Metric-Decoupled Mixed Precision Quantization☆35Updated 5 months ago
- The official implementation of the ICML 2023 paper OFQ-ViT☆30Updated last year
- Official PyTorch implementation of "Quantization Hurts Reasoning? An Empirical Study on Quantized Reasoning Models"☆30Updated last week
- BitPack is a practical tool to efficiently save ultra-low precision/mixed-precision quantized models.☆52Updated 2 years ago
- [ICLR 2024 Spotlight] This is the official PyTorch implementation of "EfficientDM: Efficient Quantization-Aware Fine-Tuning of Low-Bit Di…☆59Updated 11 months ago
- Curated list of methods that focuses on improving the efficiency of diffusion models☆44Updated 9 months ago
- Paper survey of efficient computation for large scale models.☆32Updated 4 months ago
- Official Code For Dual Grained Quantization: Efficient Fine-Grained Quantization for LLM☆14Updated last year
- QuEST: Efficient Finetuning for Low-bit Diffusion Models☆44Updated 3 months ago
- This is the pytorch implementation for the paper: Generalizable Mixed-Precision Quantization via Attribution Rank Preservation, which is…☆25Updated 3 years ago
- [ICCV 23]An approach to enhance the efficiency of Vision Transformer (ViT) by concurrently employing token pruning and token merging tech…☆95Updated last year
- Official implementation of the EMNLP23 paper: Outlier Suppression+: Accurate quantization of large language models by equivalent and opti…☆46Updated last year
- ☆20Updated 6 months ago
- [ICML 2024] SPP: Sparsity-Preserved Parameter-Efficient Fine-Tuning for Large Language Models☆20Updated 11 months ago
- Code for ICML 2021 submission☆34Updated 4 years ago
- [TMLR] Official PyTorch implementation of paper "Efficient Quantization-aware Training with Adaptive Coreset Selection"☆30Updated 8 months ago
- ☆16Updated last year
- [NeurIPS 2024] The official implementation of ZipCache: Accurate and Efficient KV Cache Quantization with Salient Token Identification☆21Updated last month
- ACL 2023☆39Updated last year
- [EMNLP 2024] RoLoRA: Fine-tuning Rotated Outlier-free LLMs for Effective Weight-Activation Quantization☆36Updated 7 months ago
- [CVPR 2024 Highlight] This is the official PyTorch implementation of "TFMQ-DM: Temporal Feature Maintenance Quantization for Diffusion Mo…☆63Updated 9 months ago