训练后量化是在不重训的前提下降低大型扩散 Transformer 显存与延迟成本的主要手段。对文生图而言,工程上要回答的问题很具体:能回收多少显存、输出保真度是否保持,以及在何种并发下低精度内核的算术收益开始压过其逐次调用的反量化开销。本文针对通过 vLLM-Omni 服务、运行在两块 NVIDIA B300(Blackwell Ultra,计算能力 10.3 / SM103,张量并行为 2)上的扩散模型 Qwen-Image-2512 给出实证回答。
评估两类量化。离线量化加载预量化的 ModelOpt checkpoint,由 vLLM-Omni 自动识别,测试了全 FP8 与混合 FP8/NVFP4 两个 checkpoint;在线量化则在加载时通过 --quantization <method> 对 BF16 checkpoint 施加量化。全部测量使用固定负载:1024x1024, 20 inference steps, seed 42, prompt=‘A ceramic teapot on a wooden table’。
实验设置与样例输出
预量化 checkpoint 无需 --quantization 标志,由 checkpoint 配置选择路径,且已在启动时核验内核选择、无静默回退到 BF16。
# BF16 基线
vllm serve Qwen/Qwen-Image-2512 --omni --tensor-parallel-size 2
# 离线 ModelOpt FP8(预量化 checkpoint,自动识别)
vllm serve feizhai123/qwen-image-2512-modelopt-fp8-dynamic-all --omni \
--tensor-parallel-size 2 --linear-backend cutlass --force-cutlass-fp8 --trust-remote-code
# 离线 混合 FP8/NVFP4
vllm serve feizhai123/qwen-image-2512-modelopt-mixed-fp8-sensitive-nvfp4-heavy --omni \
--tensor-parallel-size 2 --linear-backend cutlass --force-cutlass-fp8 --trust-remote-code
# 在线 FP8(加载时对 BF16 权重做动态量化)
vllm serve Qwen/Qwen-Image-2512 --omni --tensor-parallel-size 2 --quantization fp8
三个量化输出与基线、彼此之间的校验和均不同,证实是真实的不同计算,而非缓存或静默回退。各配置的视觉保真度均得到保持,差异仅限于细小的高频细节,且在混合 FP8/NVFP4 配置上最明显——与其更激进的权重压缩一致。




单请求:延迟、显存与保真度
下表给出各配置的单请求延迟、双卡峰值显存、磁盘 checkpoint 体积,以及相对 BF16 基线的重建保真度(以 SSIM 与 PSNR 量化,均在匹配的提示词、种子与步数下计算)。
| 配置 | 延迟 (s) | 峰值显存 (MiB) | Checkpoint | SSIM | PSNR (dB) |
|---|---|---|---|---|---|
| BF16 | 1.160 | 82,804 | 54 GB | 基线 | 基线 |
| 在线 FP8 | 1.157 | 70,504 | 复用 BF16 | 0.9984 | 39.34 |
| 离线 ModelOpt FP8 | 1.203 | 69,896 | 35 GB | 0.983 | 29.1 |
| 混合 FP8/NVFP4 | 1.772 | 68,844 | 31 GB | 0.968 | 26.3 |
峰值显存降低 15.6%(离线 FP8)与 16.9%(混合);checkpoint 缩减 35% 与 43%。在线 FP8 取得最高保真度(PSNR 39.3 dB,SSIM 0.9984),因为它在加载时对 BF16 权重做动态量化,而非依赖单独标定的产物。
并发下的吞吐
下图给出并发 1、4、8 个在途请求时的持续吞吐(images/s),采用批处理服务(--step-execution --max-num-seqs 8,每点 16 个请求)。可观察到明显的交叉:低并发下 BF16 最快,但随批次填满,量化格式的权重带宽节省压过其逐次开销。
该交叉是核心的运行结论:对延迟敏感的单流服务,本硬件上 BF16 仍更优;对吞吐导向的批处理服务,量化配置全面占优。这一“随机制依赖”与上游 Qwen3-Omni W4A4 工作(PR #4025)一致——量化优势同样在并发下、而非单流延迟中显现。
结论
对于通过 vLLM-Omni 服务、运行在 NVIDIA B300 上的 Qwen-Image-2512,FP8 与混合 FP8/NVFP4 量化在保持输出保真度的同时回收约 16% 运行显存、缩减 35%–43% checkpoint 体积,并在并发饱和内核后超过 BF16 吞吐。单流延迟敏感场景下 BF16 仍更优;追求吞吐时优先量化,其中混合 FP8/NVFP4 在并发下最快、峰值显存也最低。后续我们会继续展开在线量化、内核细节与更多模型上的实践。