xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-01-04 23:51:03 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Bugfix] Fix distributed bug in Qwen2.5-VL & Qwen2.5-Omni (#16907)

Browse Source
This commit is contained in:
Yang Fan 2025-04-21 18:25:21 +08:00 committed by GitHub
parent 4c41278b77
commit 26c0406555
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
vllm/model_executor/models/qwen2_5_vl.py
View File

@ -198,9 +198,8 @@ class Qwen2_5_VisionMLP(nn.Module):
def all_gather_interleave(local_tensor, hidden_size: int, tp_size: int):
"""All-gather the input tensor interleavely across model parallel group."""
import torch.distributed as dist
gathered_tensors = [torch.zeros_like(local_tensor) for _ in range(tp_size)]
dist.all_gather(gathered_tensors, local_tensor)
parallel_state.get_tp_group().all_gather(gathered_tensors, local_tensor)
gathered_tensors_split = [
torch.split(tensor, hidden_size // tp_size, -1)