From daade9cc00bf8aa97a0d68b924fb7b08a1fabc74 Mon Sep 17 00:00:00 2001
From: Jimmy Shong <69131491+Jiminator@users.noreply.github.com>
Date: Fri, 15 May 2026 22:01:44 -0700
Subject: [PATCH] [Fix] Probe speculative draft config via sglang get_config
 (#25428)

---
 python/sglang/srt/server_args.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/python/sglang/srt/server_args.py b/python/sglang/srt/server_args.py
index 07c7962e8..379da02a6 100644
--- a/python/sglang/srt/server_args.py
+++ b/python/sglang/srt/server_args.py
@@ -324,9 +324,9 @@ def _resolve_speculative_algorithm_alias(
 
     is_gemma4_draft = False
     if speculative_draft_model_path:
-        from transformers import AutoConfig
+        from sglang.srt.utils.hf_transformers_utils import get_config
 
-        cfg = AutoConfig.from_pretrained(
+        cfg = get_config(
             speculative_draft_model_path, trust_remote_code=trust_remote_code
         )
         is_gemma4_draft = "Gemma4AssistantForCausalLM" in (