Fix spec_utils.py (#11247)

8e66d87f · sglang-bot · GitHub · a20fc7b7 · 8e66d87f
Unverified Commit 8e66d87f authored Oct 05, 2025 by sglang-bot Committed by GitHub Oct 05, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

python/sglang/srt/speculative/spec_utils.py python/sglang/srt/speculative/spec_utils.py +1 -1

No files found.
--- 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