EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

Use Optional with None default (#2770)

Browse Source
This commit is contained in:
HAI
2025-01-07 01:35:08 -08:00
committed by GitHub
parent 380930a959
commit 6d08ce2aa9
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/sglang/srt/models/grok.py
View File

@@ -376,7 +376,7 @@ class Grok1ForCausalLM(nn.Module):
def load_weights( def load_weights(
self, self,
weights: Iterable[Tuple[str, torch.Tensor]], weights: Iterable[Tuple[str, torch.Tensor]],
use_presharded_weights: bool | None = None, use_presharded_weights: Optional[bool] = None,
): ):
if use_presharded_weights is None: if use_presharded_weights is None:
use_presharded_weights = self.use_presharded_weights use_presharded_weights = self.use_presharded_weights