From f002e9a8700c9970a1ac21711c8673f246bbaf1b Mon Sep 17 00:00:00 2001 From: Michael Goin Date: Wed, 23 Jul 2025 03:02:48 -0400 Subject: [PATCH] [Cleanup] Only log MoE DP setup warning if DP is enabled (#21315) Signed-off-by: mgoin --- vllm/model_executor/layers/fused_moe/config.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/vllm/model_executor/layers/fused_moe/config.py b/vllm/model_executor/layers/fused_moe/config.py index 51c421bd228f..f5ed2861b8fc 100644 --- a/vllm/model_executor/layers/fused_moe/config.py +++ b/vllm/model_executor/layers/fused_moe/config.py @@ -464,10 +464,11 @@ class FusedMoEConfig: ) else: _quant_config = FusedMoEQuantConfig() - logger.warning_once("MoE DP setup unable to determine " - "quantization scheme or unsupported " - "quantization type. This model will " - "not run with DP enabled.") + if moe_parallel_config.dp_size > 1: + logger.warning_once("MoE DP setup unable to determine " + "quantization scheme or unsupported " + "quantization type. This model will " + "not run with DP enabled.") else: _quant_config = quant_config