LightCompress: Towards Accurate and Efficient AIGC Model Compression

 

[ English | 中文 ]

📢 Notice: This repository was formerly known as LLMC and has been renamed to LightCompress.

LightCompress is an off-the-shell tool designed for compressing aigc models(LLM, VLM, Diffusion ...), leveraging state-of-the-art compression algorithms to enhance efficiency and reduce model size without compromising performance. You can download the Docker image that can run LightCompress with the following command. Users in mainland China are recommended to use Alibaba Cloud Docker.

# docker hub: https://hub.docker.com/r/llmcompression/llmc
docker pull llmcompression/llmc:pure-latest

# aliyun docker: registry.cn-hangzhou.aliyuncs.com/yongyang/llmcompression:[tag]
docker pull registry.cn-hangzhou.aliyuncs.com/yongyang/llmcompression:pure-latest

Community: Discord Server, Tencent QQ Group.

Docs: English, Chinese.

🔥 Latest News

• May 12, 2025: 🔥 We now fully support quantization for the Wan2.1 series of video generation models and provide export of truly quantized INT8/FP8 weights, compatible with the lightx2v inference framework. For details, please refer to the lightx2v documentation.

• Feb 07, 2025: 🔥 We now fully support quantization of large-scale MOE models like DeepSeekv3, DeepSeek-R1, and DeepSeek-R1-zero with 671B parameters. You can now directly load FP8 weights without any extra conversion. AWQ and RTN quantization can run on a single 80GB GPU, and we also support the export of true quantized INT4/INT8 weights.

• Nov 20, 2024: 🔥 We now fully support the quantization of ✨DeepSeekv2(2.5) and other MOE models, as well as ✨Qwen2VL, Llama3.2, and other VLM models. Supported quantization methods include ✅integer quantization, ✅floating-point quantization, and advanced algorithms like ✅AWQ, ✅GPTQ, ✅SmoothQuant, and ✅Quarot.

• Nov 12, 2024: 🔥 We have added support for 💥static per-tensor activation quantization across various models and algorithms, covering ✅integer quantization and ✅floating-point quantization to further optimize performance and efficiency. Additionally, we now support exporting ✨real quantized models and using the VLLM and SGLang backends for inference acceleration. For more details, refer to the VLLM documentation and SGLang documentation.

• Sep 26, 2024: 🔥 We now support exporting 💥FP8 quantized(E4M3, E5M2) models from 🚀LLMC to advanced inference backends such as VLLM and SGLang. For detailed usage, please refer to the VLLM documentation and SGLang documentation.

Previous News

• Sep 24, 2024: 🔥 We have officially released ✅INT4 and ✅INT8 models of ✨Llama-3.1-405B, quantized using 🚀LLMC in save_lightllm mode. You can download the model parameters here.

• Sep 23, 2024: 🔥 We now support exporting ✨real quantized(INT4, INT8) models from 🚀LLMC to advanced inference backends such as VLLM, SGLang, AutoAWQ, and MLC-LLM for quantized inference deployment, enabling ✨reduced memory usage and ✨faster inference speeds. For detailed usage, please refer to the VLLM documentation, SGLang documentation, AutoAWQ documentation, and MLC-LLM documentation.

• Sep 09, 2024: 🔥 We provide some configs of our best practice towards superior performance (see Best Practice here).

• Sep 03, 2024: 🔥 We support opencompass 🤗 to eval 🚀LLMC model. Follow this doc and have a try!

• Aug 22, 2024: 🔥We support lots of small language models, including current SOTA SmolLM(see Supported Model List).

• Aug 22, 2024: 🔥 Additionally, we also support down stream task evaluation through our modified lm-evaluation-harness 🤗. Specifically, people can first employ save_trans mode(see save part in Configuration) to save a weight modified model. After obtaining the transformed model, they can directly evaluate the quantized model referring to run_lm_eval.sh. More details can be found in here.

• Jul 23, 2024: 🍺🍺🍺 We release a brand new version benchmark paper:

  LLMC: Benchmarking Large Language Model Quantization with a Versatile Compression Toolkit.

  Ruihao Gong*, Yang Yong*, Shiqiao Gu*, Yushi Huang*, Chengtao Lv, Yunchen Zhang, Xianglong Liu📧, Dacheng Tao

  (* denotes equal contribution, 📧 denotes corresponding author.)

• Jul 16, 2024: 🔥We support Wanda/Naive(Magnitude) for llm sparsification and layer-wise mix bits quantization now!

• Jul 14, 2024: 🔥We support rotation based quantization QuaRot now!

• May 17, 2024: 🚀 We support some advanced large models, e.g., LLaVA, Mixtral, LLaMA V3 and Qwen V2 now. Have a try!

• May 13, 2024: 🍺🍺🍺 We release our quantization benchmark paper:

  LLM-QBench: A Benchmark Towards the Best Practice for Post-training Quantization of Large Language Models.

  Ruihao Gong*, Yang Yong*, Shiqiao Gu*, Yushi Huang*, Yunchen Zhang, Xianglong Liu📧, Dacheng Tao

  (* denotes equal contribution, 📧 denotes corresponding author.)

  

  We modularly and fairly benchmark the quantization techniques considering calibration cost, inference efficiency, and quantized accuracy. Near 600 experiments on diverse models and datasets provide three insightful takeaways on the calibration data, algorithm pipeline, and quantization configuration selection. Based on the takeaways, a best practice for the LLM PTQ pipeline is designed, to achieve the best accuracy and efficiency performance balance under various scenarios.

• Mar 07, 2024: 🚀 We release the quantization part of a powerful and efficient LLM compression tool. Notably, our benchmark paper is coming soon😊.

🚀 Highlight Feature

• 💥Comprehensive Algorithm Support: Provides a broad range of ✨SOTA compression algorithms, including ✅quantization, ✅mixed-precision quantization, and ✅sparsity, while maintaining accuracy consistent with the original repositories. ✨Quantization best practices (see 🚀Best Practices here) are also available to ensure optimal performance and efficiency.

• 💥Supported Formats: Supports both ✨quantization (integer and floating-point) and ✨sparsity, specifically including ✅weight-activation, ✅weight-only, ✅mixed-precision quantization, as well as ✅structured and ✅unstructured sparsity.

• 💥Wide Model Support: Offers support for a diverse array of ✨LLM models, including ✅LLama, ✅Mistral, ✅InternLM2, ✅Qwen2, among others, as well as ✅MOE(DeepSeekv2, Deepseek-R1) and ✅VLM(Llama3.2-vision, Qwen2-vl) models (see Supported Model List).

• 💥Multi-backend Compatibility: Seamlessly integrates with various backends for enhanced deployment flexibility. Multiple quantization settings and model formats are compatible with a wide range of backends and hardware platforms, such as ✅VLLM, ✅Sglang, ✅LightLLM, ✅MLC-LLM, and ✅AutoAWQ, making it highly versatile(see Section Backend here).

• 💥Performance Efficiency: Enables quantization of large LLMs, such as ✨Llama3.1-405B and ✨DeepSeek-R1-671B, with PPL evaluation on a single A100/H100/H800 GPU.

⚙️ Usage

Please refer to the 🚀Quick Start section in the documentation.

🤖 Supported Model List

• ✅ BLOOM
• ✅ LLaMA
• ✅ LLaMA V2
• ✅ StarCoder
• ✅ OPT

More Supported Models 

• ✅ Falcon
• ✅ InternLM2
• ✅ Mistral
• ✅ LLaMA V3
• ✅ Mixtral
• ✅ Qwen V2
• ✅ LLaVA
• ✅ InternLM2.5
• ✅ StableLM
• ✅ Gemma2
• ✅ Phi2
• ✅ Phi 1.5
• ✅ MiniCPM
• ✅ SmolLM
• ✅ DeepSeekv2.5
• ✅ LLaMA V3.2 Vision
• ✅ Qwen MOE
• ✅ Qwen2-VL
• ✅ InternVL2

You can add your own model type referring to files under llmc/models/*.py.

🚌 Supported Backend List

• ✅ VLLM
• ✅ LightLLM
• ✅ Sglang
• ✅ MLC-LLM
• ✅ AutoAWQ

💡 Supported Algorithm List

Quantization

• ✅ Naive
• ✅ AWQ
• ✅ GPTQ
• ✅ SmoothQuant
• ✅ OS+

More Supported Algorithms 

• ✅ OmniQuant
• ✅ NormTweaking
• ✅ AdaDim
• ✅ QUIK
• ✅ SpQR
• ✅ DGQ
• ✅ OWQ
• ✅ LLM.int8()
• ✅ HQQ
• ✅ QuaRot
• ✅ SpinQuant (See this branch)
• ✅ TesseraQ

Pruning

• ✅ Naive(Magnitude)
• ✅ Wanda
• ✅ ShortGPT

🤝 Acknowledgments

We develop our code referring to the following repos:

• mit-han-lab/llm-awq
• mit-han-lab/smoothquant
• OpenGVLab/OmniQuant
• IST-DASLab/gptq
• ModelTC/Outlier_Suppression_Plus

More Related Implementations 

• IST-DASLab/QUIK
• Vahe1994/SpQR
• ilur98/DGQ
• xvyaward/owq
• TimDettmers/bitsandbytes
• mobiusml/hqq
• spcl/QuaRot
• locuslab/wanda
• EleutherAI/lm-evaluation-harness
• facebookresearch/SpinQuant
• Intelligent-Computing-Lab-Yale/TesseraQ

🌟 Star History

✏️ Citation

If you find our toolkit or research paper useful or relevant to your research, please kindly cite our work:

@inproceedings{DBLP:conf/emnlp/GongYGHLZT024,
  author={Ruihao Gong and Yang Yong and Shiqiao Gu and Yushi Huang and Chengtao Lv and Yunchen Zhang and Dacheng Tao and Xianglong Liu},
  title={LLMC: Benchmarking Large Language Model Quantization with a Versatile Compression Toolkit},
  year={2024},
  cdate={1704067200000},
  pages={132-152},
  url={https://aclanthology.org/2024.emnlp-industry.12},
  booktitle={EMNLP (Industry Track)},
  crossref={conf/emnlp/2024i}
}