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 68e6ebefa316..1bae8c8559a2 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 @@ def __init__( # 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")