Use Optional with None default (#2770)

6d08ce2a · HAI · GitHub · 380930a9 · 6d08ce2a
Unverified Commit 6d08ce2a authored Jan 07, 2025 by HAI Committed by GitHub Jan 07, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

python/sglang/srt/models/grok.py python/sglang/srt/models/grok.py +1 -1

No files found.
--- 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