diff --git a/python/sglang/srt/speculative/dflash_utils.py b/python/sglang/srt/speculative/dflash_utils.py index 284ae5661..f5aa6e158 100644 --- a/python/sglang/srt/speculative/dflash_utils.py +++ b/python/sglang/srt/speculative/dflash_utils.py @@ -53,8 +53,7 @@ elif is_hip(): from sglang.kernels.ops.sampling.renorm_triton import ( top_p_renorm_probs_triton as top_p_renorm_prob, ) - - _DFLASH_SAMPLING_VERIFY_AVAILABLE = True + tree_speculative_sampling_target_only = None else: top_k_renorm_prob = None top_p_renorm_prob = None