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

2026-03-18 21:51:39 -07:00
parent 7f6f1a3ab1
commit f85455ab24
1 changed files with 2 additions and 2 deletions
@@ -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=not use_data_parallel,
-                enable_tp=False if use_data_parallel else True,
+                use_attn_tp_group=is_dp_attention_enabled() and not use_data_parallel,
                prefix=add_prefix("pos_embed", prefix),
            )
        else: