From d792aa761873c92b867c311b6e4dd2c060103171 Mon Sep 17 00:00:00 2001 From: Hao Jin Date: Sat, 7 Feb 2026 04:35:12 -0800 Subject: [PATCH] [diffusion] fix: remove unnecessary norm_type argument from GLM-Image dits (#18382) Co-authored-by: Hao Jin --- python/sglang/multimodal_gen/runtime/models/dits/glm_image.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/sglang/multimodal_gen/runtime/models/dits/glm_image.py b/python/sglang/multimodal_gen/runtime/models/dits/glm_image.py index 68e6ebefa..1bae8c855 100644 --- a/python/sglang/multimodal_gen/runtime/models/dits/glm_image.py +++ b/python/sglang/multimodal_gen/runtime/models/dits/glm_image.py @@ -483,10 +483,10 @@ class GlmImageTransformerBlock(nn.Module): # 2. Feedforward self.norm2 = ScaleResidualLayerNormScaleShift( - dim, norm_type="layer", eps=1e-5, elementwise_affine=False + dim, eps=1e-5, elementwise_affine=False ) self.norm2_context = ScaleResidualLayerNormScaleShift( - dim, norm_type="layer", eps=1e-5, elementwise_affine=False + dim, eps=1e-5, elementwise_affine=False ) self.ff = FeedForward(dim=dim, dim_out=dim, activation_fn="gelu-approximate")