From 888e126ac971efc214c4a412dff0b5380e1335cc Mon Sep 17 00:00:00 2001 From: triple-mu Date: Sat, 3 Jan 2026 12:42:04 +0900 Subject: [PATCH] [diffusion] comment: fix typo (#16257) Co-authored-by: Mick --- python/sglang/multimodal_gen/runtime/layers/usp.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/sglang/multimodal_gen/runtime/layers/usp.py b/python/sglang/multimodal_gen/runtime/layers/usp.py index a197eb4af..04c020320 100644 --- a/python/sglang/multimodal_gen/runtime/layers/usp.py +++ b/python/sglang/multimodal_gen/runtime/layers/usp.py @@ -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] 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. Args: