[Bugfix] fix qwen3vl hang when --mm-enable-dp-encoder is enable (#20759)

This commit is contained in:
Ling Zhang
2026-03-19 12:51:39 +08:00
committed by GitHub
parent 7f6f1a3ab1
commit f85455ab24

View File

@@ -329,8 +329,8 @@ class Qwen3VLMoeVisionModel(nn.Module, RotaryPosMixin):
self.num_position_embeddings,
self.hidden_size,
quant_config=quant_config,
use_attn_tp_group=is_dp_attention_enabled(),
enable_tp=False if use_data_parallel else True,
enable_tp=not use_data_parallel,
use_attn_tp_group=is_dp_attention_enabled() and not use_data_parallel,
prefix=add_prefix("pos_embed", prefix),
)
else: