[diffusion] fix: fix accuracy bug caused by #14717 (#18296)

2026-02-05 20:36:18 +08:00
parent 8f8c1724ae
commit 4aa03d91fd
3 changed files with 6 additions and 6 deletions
--- a/python/sglang/multimodal_gen/runtime/layers/layernorm.py
+++ b/python/sglang/multimodal_gen/runtime/layers/layernorm.py
@@ -51,7 +51,7 @@ class RMSNorm(CustomOp):
        var_hidden_size: Optional[int] = None,
    ) -> None:
        super().__init__()
-        self.weight = nn.Parameter(torch.ones(hidden_size, dtype=dtype))
+        self.weight = nn.Parameter(torch.ones(hidden_size))
        self.variance_epsilon = eps
        self.hidden_size = hidden_size
        self.variance_size_override = (