From 8e66d87f0a931c1b14a1f8c36fb82bd04e98b0a4 Mon Sep 17 00:00:00 2001 From: sglang-bot Date: Sun, 5 Oct 2025 19:01:11 -0700 Subject: [PATCH] Fix spec_utils.py (#11247) --- python/sglang/srt/speculative/spec_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/sglang/srt/speculative/spec_utils.py b/python/sglang/srt/speculative/spec_utils.py index 6640c077d..1bfa774bc 100644 --- a/python/sglang/srt/speculative/spec_utils.py +++ b/python/sglang/srt/speculative/spec_utils.py @@ -15,7 +15,7 @@ from sglang.srt.managers.schedule_batch import Req from sglang.srt.utils import is_cuda, is_hip if is_cuda(): - from sgl_kernel import fast_topk + from sgl_kernel import fast_topk, tree_speculative_sampling_target_only elif is_hip(): from sgl_kernel import fast_topk