From 8d562d6b0333f2ba5696ab81915a261997333178 Mon Sep 17 00:00:00 2001 From: Woosuk Kwon Date: Mon, 22 Jul 2024 22:45:13 +0000 Subject: [PATCH 1/2] [Misc] Remove deprecation warning for beam search --- vllm/sampling_params.py | 12 ------------ 1 file changed, 12 deletions(-) diff --git a/vllm/sampling_params.py b/vllm/sampling_params.py index ebe5e0fd34135..07d393c013a97 100644 --- a/vllm/sampling_params.py +++ b/vllm/sampling_params.py @@ -189,18 +189,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() From a3086d0d535204d39b1b17b889a5417f2c5f0991 Mon Sep 17 00:00:00 2001 From: Woosuk Kwon Date: Mon, 22 Jul 2024 22:53:03 +0000 Subject: [PATCH 2/2] ruff --- vllm/sampling_params.py | 1 - 1 file changed, 1 deletion(-) diff --git a/vllm/sampling_params.py b/vllm/sampling_params.py index 07d393c013a97..638c870c04371 100644 --- a/vllm/sampling_params.py +++ b/vllm/sampling_params.py @@ -8,7 +8,6 @@ from pydantic import Field from typing_extensions import Annotated -import vllm.envs as envs from vllm.logger import init_logger logger = init_logger(__name__)