Skip to content

Commit a935d36

Browse files
chengduxiaowuchengduxiaowu
andauthored
[Misc] spec_decode: remove redundant warning introduced by debug logging probes (vllm-project#12501)
### What this PR does / why we need it? This pull request removes a warning log statement in `rejection_sample` that was triggered when `draft_probs` is `None` in the random sampling path. This warning is no longer necessary or was redundant. ### Does this PR introduce _any_ user-facing change? No. ### How was this patch tested? No tests were added as this is a minor cleanup of a warning log. Signed-off-by: chengduxiaowu <wuchengtai@huawei-partners.com> Co-authored-by: chengduxiaowu <wuchengtai@huawei-partners.com>
1 parent 74f669e commit a935d36

1 file changed

Lines changed: 0 additions & 4 deletions

File tree

vllm_ascend/sample/rejection_sampler.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -574,10 +574,6 @@ def rejection_sample(
574574
# For random sampling with selected logits
575575
# target_logits is [num_tokens, top_k*tp_size] with indices [num_tokens, top_k*tp_size]
576576
if target_indices is not None:
577-
if draft_probs is None:
578-
logger.warning(
579-
"[spec/dfx] rejection_sample: draft_probs is None in random sampling path; acceptance may be degraded.",
580-
)
581577
# Enable reduce_sampling: logits are [num_tokens, top_k*tp_size]
582578
# We need to handle rejection sampling with selected vocab
583579
selected_vocab_size = target_logits.shape[-1]

0 commit comments

Comments
 (0)