Fix spec_utils.py (#11247)

2025-10-05 19:01:11 -07:00
parent a20fc7b7dc
commit 8e66d87f0a
1 changed files with 1 additions and 1 deletions
--- 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