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

fix(grok): remove duplicate replicate_lm_head configuration (#9549)

Browse Source
This commit is contained in:
Vincent Zhong
2025-08-23 22:49:24 -04:00
committed by GitHub
parent 80425e59bb
commit 327f7b7c87
1 changed files with 0 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -842,10 +842,6 @@ class Grok1ForCausalLM(nn.Module):
if self.is_weights_presharded: if self.is_weights_presharded:
setattr(DefaultModelLoader, "_prepare_weights", _prepare_presharded_weights) setattr(DefaultModelLoader, "_prepare_weights", _prepare_presharded_weights)
default_replicate_lm_head = False
self.replicate_lm_head = getattr(
config, "replicate_lm_head", default_replicate_lm_head
)
self.replicate_embedding = getattr(config, "replicate_embedding", False) self.replicate_embedding = getattr(config, "replicate_embedding", False)
self.model = Grok1Model( self.model = Grok1Model(