minke.yu/sglang
1
0
Fork 0
Code Issues Pull Requests Actions 12 Packages Projects Releases Wiki Activity

[diffusion] comment: fix typo (#16257)

Browse Source 
Co-authored-by: Mick <mickjagger19@icloud.com>
This commit is contained in:
triple-mu
2026-01-03 11:42:04 +08:00
committed by GitHub
co-authored by Mick
parent bb23a8fe77
commit 888e126ac9
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/multimodal_gen/runtime/layers/usp.py
+1 -1
View File
@@ -54,7 +54,7 @@ def _usp_input_all_to_all(x: torch.Tensor, head_dim: int = 1) -> torch.Tensor:
[b, h, s_local, d] -> [b, h_local, s_global, d] [b, h, s_local, d] -> [b, h_local, s_global, d]
If heads are at dim=2 (input is [b, s_local, h, d]), set head_dim=2, and the If heads are at dim=2 (input is [b, s_local, h, d]), set head_dim=2, and the
function returns [b, s_global, h+local, d], preserving the original function returns [b, s_global, h_local, d], preserving the original
head/sequence dim ordering. head/sequence dim ordering.
Args: Args: