From c56a5efbaaf57208cd3d6b056af0d1a624b2cd99 Mon Sep 17 00:00:00 2001 From: Lianmin Zheng Date: Thu, 12 Feb 2026 18:41:41 -0800 Subject: [PATCH] [Auto Sync] Update grok.py (20260213) (#18765) Co-authored-by: github-actions[bot] Co-authored-by: Cheng Wan <54331508+ch-wan@users.noreply.github.com> --- python/sglang/srt/models/grok.py | 9 +-------- 1 file changed, 1 insertion(+), 8 deletions(-) diff --git a/python/sglang/srt/models/grok.py b/python/sglang/srt/models/grok.py index a089475b7..85155719a 100644 --- a/python/sglang/srt/models/grok.py +++ b/python/sglang/srt/models/grok.py @@ -690,19 +690,12 @@ class Grok1ForCausalLM(nn.Module): config, "load_presharded_embedding", False ) - self.is_weights_presharded = ( - self.load_presharded_mlp - or self.load_presharded_moe - or self.load_presharded_attn - or self.load_presharded_embedding - ) - default_replicate_lm_head = False self.replicate_lm_head = getattr( config, "replicate_lm_head", default_replicate_lm_head ) - if self.is_weights_presharded: + if get_tensor_model_parallel_world_size() > 1: setattr(DefaultModelLoader, "_prepare_weights", _prepare_presharded_weights) self.replicate_embedding = getattr(config, "replicate_embedding", False)