[Example] Add Mixed W2A16 and W4A16 MoE Example - #2940
Conversation
|
👋 Hi! Thank you for contributing to llm-compressor. Please add the ready label when the PR is ready for review. Note: This is required to complete the testing suite, please only add the label once the PR is code complete and local testing has been performed. |
|
Important Review skippedAuto reviews are disabled on this repository. Please check the settings in the CodeRabbit UI or the ⚙️ Run configurationConfiguration used: Path: .coderabbit.yaml Review profile: CHILL Plan: Pro Plus Run ID: You can disable this status message by setting the Use the checkbox below for a quick retry:
✨ Finishing Touches🧪 Generate unit tests (beta)
Thanks for using CodeRabbit! It's free for OSS, and your support helps us grow. If you like it, consider giving us a shout-out. Comment |
There was a problem hiding this comment.
Code Review
This pull request introduces a new example script, qwen3_moe_example_mixed_w2a16_w4a16.py, demonstrating mixed-precision quantization (W2A16 and W4A16) using AutoRound on a Qwen MoE model. The feedback suggests three key improvements: replacing the non-public model ID with a publicly available one, loading the model with torch_dtype="auto" to prevent potential out-of-memory errors, and increasing the number of calibration iterations from 2 to a standard value like 200 to ensure high-quality quantization.
Merge Protections🟢 All 2 merge protections satisfied — ready to merge. Show 2 satisfied protections🟢 Require one maintainer reviewAll PRs must have at least one approving review from a maintainer before merging.
🟢 Require two reviewsPRs labelled "two-reviews" must have at least two approving reviews before merging.
|
|
The quality checks have failed. Please run |
Signed-off-by: yiliu30 <yi4.liu@intel.com>
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com> Signed-off-by: Yi Liu <yi4.liu@intel.com>
…w2a16_w4a16.py Co-authored-by: Kyle Sayers <kylesayrs@gmail.com> Signed-off-by: Yi Liu <yi4.liu@intel.com>
Signed-off-by: yiliu30 yi4.liu@intel.com
SUMMARY:
"please provide a brief summary"
The evluation is depends on vllm's PR vllm-project/vllm#48918