From 6d08ce2aa9f481f6704b8c04a9d2a5b138db7ebe Mon Sep 17 00:00:00 2001 From: HAI Date: Tue, 7 Jan 2025 01:35:08 -0800 Subject: [PATCH] Use Optional with None default (#2770) --- python/sglang/srt/models/grok.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/sglang/srt/models/grok.py b/python/sglang/srt/models/grok.py index e55a99465..0485b80fc 100644 --- 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