Skip to content

Commit

Permalink
[Misc] Remove deprecation warning for beam search (vllm-project#6659)
Browse files Browse the repository at this point in the history
Signed-off-by: Alvant <[email protected]>
  • Loading branch information
WoosukKwon authored and Alvant committed Oct 26, 2024
1 parent be9f306 commit d283474
Showing 1 changed file with 0 additions and 13 deletions.
13 changes: 0 additions & 13 deletions vllm/sampling_params.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,6 @@
from pydantic import Field
from typing_extensions import Annotated

import vllm.envs as envs
from vllm.logger import init_logger
from vllm.logits_process import LogitsProcessor, NoBadWordsLogitsProcessor

Expand Down Expand Up @@ -185,18 +184,6 @@ def __init__(

self._verify_args()
if self.use_beam_search:
# Lazy import to avoid circular imports.
from vllm.usage.usage_lib import set_runtime_usage_data
set_runtime_usage_data("use_beam_search", True)

if not envs.VLLM_NO_DEPRECATION_WARNING:
logger.warning(
"[IMPORTANT] We plan to discontinue the support for beam "
"search in the next major release. Please refer to "
"https://github.com/vllm-project/vllm/issues/6226 for "
"more information. Set VLLM_NO_DEPRECATION_WARNING=1 to "
"suppress this warning.")

self._verify_beam_search()
else:
self._verify_non_beam_search()
Expand Down

0 comments on commit d283474

Please sign in to comment.