[diffusion] fix: remove unnecessary norm_type argument from GLM-Image dits (#18382)

Co-authored-by: Hao Jin <Hao Jin>
2026-02-07 04:35:12 -08:00
parent eb4cf1dfc4
commit d792aa7618
1 changed files with 2 additions and 2 deletions
--- 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")