From acc940d30217ea6ff67a278a2d894d136466aca2 Mon Sep 17 00:00:00 2001 From: triple-mu Date: Fri, 13 Feb 2026 21:59:39 +0800 Subject: [PATCH] [diffusion] fix typo (#18790) --- .../diffusion/cutedsl/scale_residual_norm_scale_shift.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/sglang/jit_kernel/diffusion/cutedsl/scale_residual_norm_scale_shift.py b/python/sglang/jit_kernel/diffusion/cutedsl/scale_residual_norm_scale_shift.py index 495b1fce0..ba8782d3d 100644 --- a/python/sglang/jit_kernel/diffusion/cutedsl/scale_residual_norm_scale_shift.py +++ b/python/sglang/jit_kernel/diffusion/cutedsl/scale_residual_norm_scale_shift.py @@ -289,7 +289,7 @@ def fused_norm_scale_shift( where norm is either layernorm or rmsnorm. Expects: - - x: B, S, D] + - x: [B, S, D] - weight/bias: None, [D] - scale/shift: [1], [D], [1/B, D], [1/B, 1/S, D] or [B, F, 1, D] - norm_type: str, "layer" or "rms"