Skip to content

Commit f8f9c0b

Browse files
[Bugfix] Don't set an upper bound on repetition penalty (#16403)
Signed-off-by: Alex-Brooks <[email protected]> Co-authored-by: Nick Hill <[email protected]>
1 parent dda8110 commit f8f9c0b

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

vllm/sampling_params.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -385,9 +385,10 @@ def _verify_args(self) -> None:
385385
if not -2.0 <= self.frequency_penalty <= 2.0:
386386
raise ValueError("frequency_penalty must be in [-2, 2], got "
387387
f"{self.frequency_penalty}.")
388-
if not 0.0 < self.repetition_penalty <= 2.0:
389-
raise ValueError("repetition_penalty must be in (0, 2], got "
390-
f"{self.repetition_penalty}.")
388+
if self.repetition_penalty <= 0.0:
389+
raise ValueError(
390+
"repetition_penalty must be greater than zero, got "
391+
f"{self.repetition_penalty}.")
391392
if self.temperature < 0.0:
392393
raise ValueError(
393394
f"temperature must be non-negative, got {self.temperature}.")

0 commit comments

Comments
 (0)