From 73fc87a74f5e6e2974403d95292dca9bb387b288 Mon Sep 17 00:00:00 2001 From: jianzhao-xu <978716854@qq.com> Date: Tue, 7 Apr 2026 10:04:41 +0800 Subject: [PATCH] fix(grok): adapt huihui-ai/grok-2 (#21522) Co-authored-by: Jianzhao Xu --- python/sglang/srt/models/grok.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/python/sglang/srt/models/grok.py b/python/sglang/srt/models/grok.py index aabc5c00d..408811d71 100644 --- a/python/sglang/srt/models/grok.py +++ b/python/sglang/srt/models/grok.py @@ -983,6 +983,9 @@ def _prepare_presharded_weights( for pattern in allow_patterns: hf_weights_files += glob.glob(os.path.join(hf_folder, pattern)) + if not hf_weights_files: + return old_prepare_weights(self, model_name_or_path, revision, fall_back_to_pt) + if hf_weights_files[0].endswith("safetensors"): use_safetensors = True else: