We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 1b7624b commit 53ea6adCopy full SHA for 53ea6ad
tests/v1/sample/test_sampling_params_e2e.py
@@ -113,6 +113,10 @@ def test_allowed_token_ids(model):
113
PROMPT, SamplingParams(allowed_token_ids=allowed_token_ids))
114
assert output[0].outputs[0].token_ids[-1] == TOKEN_ID
115
116
+ # Reject empty allowed_token_ids.
117
+ with pytest.raises(ValueError):
118
+ _ = model.generate(PROMPT, SamplingParams(allowed_token_ids=[]))
119
+
120
# Reject negative token id.
121
with pytest.raises(ValueError):
122
_ = model.generate(PROMPT, SamplingParams(allowed_token_ids=[-1]))
0 commit comments