[diffusion] fix typo (#18790)

This commit is contained in:
triple-mu
2026-02-13 21:59:39 +08:00
committed by GitHub
parent 9a32f8ccb9
commit acc940d302

View File

@@ -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"