EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

Fix spec_utils.py (#11247)

Browse Source
This commit is contained in:
sglang-bot
2025-10-05 19:01:11 -07:00
committed by GitHub
parent a20fc7b7dc
commit 8e66d87f0a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/sglang/srt/speculative/spec_utils.py
View File

@@ -15,7 +15,7 @@ from sglang.srt.managers.schedule_batch import Req
from sglang.srt.utils import is_cuda, is_hip from sglang.srt.utils import is_cuda, is_hip
if is_cuda(): if is_cuda():
from sgl_kernel import fast_topk from sgl_kernel import fast_topk, tree_speculative_sampling_target_only
elif is_hip(): elif is_hip():
from sgl_kernel import fast_topk from sgl_kernel import fast_topk