Use Optional with None default (#2770)

2025-01-07 01:35:08 -08:00
parent 380930a959
commit 6d08ce2aa9
1 changed files with 1 additions and 1 deletions
--- a/python/sglang/srt/models/grok.py
+++ b/python/sglang/srt/models/grok.py
@@ -376,7 +376,7 @@ class Grok1ForCausalLM(nn.Module):
    def load_weights(
        self,
        weights: Iterable[Tuple[str, torch.Tensor]],
-        use_presharded_weights: bool | None = None,
+        use_presharded_weights: Optional[bool] = None,
    ):
        if use_presharded_weights is None:
            use_presharded_weights = self.use_presharded_weights