[fix]修复tp1 dpsk启动失败

b6a7135f · 王敏 · ad7c14d5 · b6a7135f
Commit b6a7135f authored Feb 12, 2026 by 王敏
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

vllm/model_executor/models/deepseek_v2.py vllm/model_executor/models/deepseek_v2.py +2 -1

No files found.
--- a/vllm/model_executor/models/deepseek_v2.py
+++ b/vllm/model_executor/models/deepseek_v2.py
@@ -484,6 +484,7 @@ class DeepseekV2MoE(nn.Module):
                final_hidden_states = (
                    self.experts.maybe_all_reduce_tensor_model_parallel(
                        final_hidden_states))
+
        return final_hidden_states.view(num_tokens, hidden_dim)