From 69c92c07aa2e879477b508b4502edb269e0a5e3a Mon Sep 17 00:00:00 2001 From: wangcx18 Date: Fri, 18 Aug 2023 08:10:27 +0800 Subject: [PATCH] Fix typo in sampling_params.py --- vllm/sampling_params.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/vllm/sampling_params.py b/vllm/sampling_params.py index 6e51ad19e4e0..91f2cb1bf141 100644 --- a/vllm/sampling_params.py +++ b/vllm/sampling_params.py @@ -77,7 +77,7 @@ def __init__( self._verify_args() if self.use_beam_search: - self._verity_beam_search() + self._verify_beam_search() elif self.temperature < _SAMPLING_EPS: # Zero temperature means greedy sampling. self._verify_greedy_sampling() @@ -109,7 +109,7 @@ def _verify_args(self) -> None: raise ValueError( f"logprobs must be non-negative, got {self.logprobs}.") - def _verity_beam_search(self) -> None: + def _verify_beam_search(self) -> None: if self.best_of == 1: raise ValueError("best_of must be greater than 1 when using beam " f"search. Got {self.best_of}.")