Text Generation
Transformers
Safetensors
PyTorch
nemotron_h
nvidia
nemotron-3
latent-moe
mtp
conversational
custom_code
8-bit precision
modelopt

Spark not using NVFP4?

#18
by D-Lynch - opened

Using the provided docker and instructions in the README for DGX Spark:
docker pull vllm/vllm-openai:v0.17.1-cu130

Then

docker run --gpus all \
  --ipc=host \
  --entrypoint vllm \
  -v ~/models:/workspace/models \
  -p 8000:8000 \
  -e VLLM_NVFP4_GEMM_BACKEND=marlin \
  -e VLLM_ALLOW_LONG_MAX_MODEL_LEN=1 \
  -e VLLM_FLASHINFER_ALLREDUCE_BACKEND=trtllm \
  vllm/vllm-openai:v0.17.1-cu130 \
  serve /workspace/models/NVIDIA-Nemotron-3-Super-120B-A12B-NVFP4 \
  --served-model-name nemotron-3-super \
  --host 0.0.0.0 \
  --port 8000 \
  --async-scheduling \
  --dtype auto \
  --kv-cache-dtype fp8 \
  --tensor-parallel-size 1 \
  --pipeline-parallel-size 1 \
  --data-parallel-size 1 \
  --trust-remote-code \
  --gpu-memory-utilization 0.90 \
  --enable-chunked-prefill \
  --max-num-seqs 4 \
  --max-model-len 394000 \
  --attention-backend TRITON_ATTN \
  --mamba_ssm_cache_dtype float32 \
  --moe-backend marlin

This message is in the log:

(EngineCore_DP0 pid=99) WARNING 03-25 18:13:38 [marlin_utils_fp4.py:150] Your GPU does not have native support for FP4 computation but FP4 quantization is being used. Weight-only FP4 compression will be used leveraging the Marlin kernel. This may degrade performance for compute-heavy workloads.

Sign up or log in to comment