Skip to content

Commit 2cc3230

Browse files
committed
fix vllm comments
Signed-off-by: Hung-Yueh Chiang <hungyuehc@nvidia.com>
1 parent 0fede96 commit 2cc3230

4 files changed

Lines changed: 4 additions & 4 deletions

File tree

CHANGELOG.rst

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ Changelog
2727

2828
**New Features**
2929

30-
- Add NVFP4 W4A16 weight-only quantization (``w4a16_nvfp4``): FP4 weights with group_size=16, BF16 activations, no calibration forward pass required. Use ``mtq.W4A16_NVFP4_CFG`` or ``--qformat w4a16_nvfp4`` in ``hf_ptq.py``. Exported checkpoints can be served on vLLM after conversion to compressed-tensors format.
30+
- Add NVFP4 W4A16 weight-only quantization (``w4a16_nvfp4``): FP4 weights with group_size=16, BF16 activations, no calibration forward pass required. Use ``mtq.W4A16_NVFP4_CFG`` or ``--qformat w4a16_nvfp4`` in ``hf_ptq.py``. vLLM deployment support is in progress.
3131
- Support full Transformer Engine spec for Minitron pruning (``mcore_minitron``). Now we no longer need to use custom ModelOpt spec. Note that this does not affect the usage of the pruning workflow but makes pruning slightly faster and may result in slightly different pruned model because of different kernel and numerics.
3232
- Add end-to-end tutorial for Minitron pruning + distillation + quantization + evaluation + vLLM deployment for Nemotron-Nano-9B-v2 → Pruned 7B along with data blend preparation steps (and ablation study). See `examples/pruning/minitron/README.md <https://github.com/NVIDIA/Model-Optimizer/tree/main/examples/pruning/minitron/>`_ for details.
3333
- Add Puzzletron - a new algorithm for heterogeneous pruning of LLM and VLM models. See `examples/puzzletron/README.md <https://github.com/NVIDIA/Model-Optimizer/tree/main/examples/puzzletron>`_ for more details.

examples/llm_ptq/hf_ptq.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -789,7 +789,7 @@ def export_quantized(
789789
if args.qformat == "w4a16_nvfp4":
790790
warnings.warn(
791791
"TensorRT-LLM and SGLang do not support this format. "
792-
"To serve on vLLM, convert the NVFP4 W4A16 checkpoint to compressed-tensors format."
792+
"vLLM deployment support is in progress."
793793
)
794794

795795
# Restore default padding and export the tokenizer as well.

modelopt/torch/export/quant_utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -66,10 +66,10 @@
6666
QUANTIZATION_NVFP4,
6767
QUANTIZATION_NVFP4_AWQ,
6868
QUANTIZATION_NVFP4_SVDQUANT,
69-
QUANTIZATION_W4A16_NVFP4,
7069
QUANTIZATION_W4A8_AWQ,
7170
QUANTIZATION_W4A8_MXFP4_FP8,
7271
QUANTIZATION_W4A8_NVFP4_FP8,
72+
QUANTIZATION_W4A16_NVFP4,
7373
)
7474

7575
logger = logging.getLogger(__name__)

modelopt/torch/export/unified_export_hf.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -80,9 +80,9 @@
8080
QUANTIZATION_NVFP4,
8181
QUANTIZATION_NVFP4_AWQ,
8282
QUANTIZATION_NVFP4_SVDQUANT,
83-
QUANTIZATION_W4A16_NVFP4,
8483
QUANTIZATION_W4A8_AWQ,
8584
QUANTIZATION_W4A8_NVFP4_FP8,
85+
QUANTIZATION_W4A16_NVFP4,
8686
)
8787
from .model_utils import get_language_model_from_vl, is_multimodal_model
8888
from .moe_utils import _export_fused_experts

0 commit comments

Comments
 (0)