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

[diffusion] fix typo (#18790)

Browse Source
This commit is contained in:
triple-mu
2026-02-13 21:59:39 +08:00
committed by GitHub
parent 9a32f8ccb9
commit acc940d302
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/sglang/jit_kernel/diffusion/cutedsl/scale_residual_norm_scale_shift.py
+1 -1
View File
@@ -289,7 +289,7 @@ def fused_norm_scale_shift(
where norm is either layernorm or rmsnorm. where norm is either layernorm or rmsnorm.
Expects: Expects:
- x: B, S, D] - x: [B, S, D]
- weight/bias: None, [D] - weight/bias: None, [D]
- scale/shift: [1], [D], [1/B, D], [1/B, 1/S, D] or [B, F, 1, D] - scale/shift: [1], [D], [1/B, D], [1/B, 1/S, D] or [B, F, 1, D]
- norm_type: str, "layer" or "rms" - norm_type: str, "layer" or "rms"