NVFP4 quantization

#19
by Duonglv - opened

Hello,
You did a amazing job. This model is very strong.
Could you guys release a NVFP4 quantized model with some evaluations between this one and FP16 model?
Thank so much.

@QBANIN That is GGUF version. I meant a NVFP4 version for running in vLLM. You can see the nvdia org, they have quantized many nvfp4 models with evaluations.

In an effort to make this all accessible for consumer cards. I can't speak for much but I work heavily with VLLM, and have only a single 5090 at my disposal, so here are two non-GGUF serve commands.

This one has limitation of 21gb to weights. Adjust cache or sequences to lower, as the below commands fill up to around 30gb maximum fully loaded, fitting perfectly.

INT4 (non blackwell)

(Click to Expand)

First run, use: vllm serve compute1/Agents-A1-GPTQ-INT4-Sym

CUDA_VISIBLE_DEVICES=0
vllm serve /path/to/.cache/huggingface/hub/models--compute1--Agents-A1-GPTQ-INT4-Sym/snapshots/somestring/
--served-model-name compute1/Agents-A1-GPTQ-INT4-Sym
--host 0.0.0.0
--port 8000
--tensor-parallel-size 1
--max-model-len 196608
--gpu-memory-utilization 0.87
--kv-cache-dtype fp8_e5m2
--max-num-seqs 2
--tool-call-parser qwen3_coder
--reasoning-parser qwen3
--enable-auto-tool-choice
--trust-remote-code

^^^ Great INT4 , performs as well as ~Q5/Q6 gguf, if not better.

NVFP4 (heavier, blackwell)

(Click to Expand)

First run, use: vllm serve cyankiwi/Agents-A1-AWQ-NVFP4

CUDA_VISIBLE_DEVICES=0
VLLM_USE_FLASHINFER_SAMPLER=0
vllm serve /path/to/.cache/huggingface/hub/models--cyankiwi--Agents-A1-AWQ-NVFP4/snapshots/somestring/
--served-model-name cyankiwi/Agents-A1-AWQ-NVFP4
--host 0.0.0.0
--port 8000
--moe-backend marlin
--tensor-parallel-size 1
--max-model-len 172032
--gpu-memory-utilization 0.88
--kv-cache-dtype fp8
--max-num-seqs 2
--tool-call-parser qwen3_coder
--reasoning-parser qwen3
--enable-auto-tool-choice
--trust-remote-code

^^^ Great NVFP4, performs as well as ~Q6 gguf, if not better.

For sm120 - i.e. 5090, consumer blackwell
Non-negotiable moe-backend and flashinfer sampler env var - discovered from (this model - link to highlight) - where they broke the vision heads 🤷

YMMV with system ram for profiling/warmup - Specs: 5090 + 64GB — vLLM runtime on CUDA host

Sign up or log in to comment