[Misc] Validate stop_token_ids contents (#17268)

Signed-off-by: Nick Hill <nhill@redhat.com>
This commit is contained in:
Nick Hill 2025-04-27 20:54:05 -07:00 committed by GitHub
parent f211331c48
commit 8262a3e23b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -437,6 +437,10 @@ class SamplingParams(
and self.truncate_prompt_tokens < 1):
raise ValueError(f"truncate_prompt_tokens must be >= 1, "
f"got {self.truncate_prompt_tokens}")
assert isinstance(self.stop_token_ids, list)
if not all(isinstance(st_id, int) for st_id in self.stop_token_ids):
raise ValueError(f"stop_token_ids must contain only integers, "
f"got {self.stop_token_ids}.")
assert isinstance(self.stop, list)
if any(not stop_str for stop_str in self.stop):
raise ValueError("stop cannot contain an empty string.")