[Core] Factor out input preprocessing to a separate class (#7329)

This commit is contained in:
Cyrus Leung
2024-09-13 10:56:13 +08:00
committed by GitHub
parent 8f44a92d85
commit 5ec9c0fb3c
5 changed files with 590 additions and 540 deletions

View File

@@ -11,9 +11,10 @@ def test_skip_tokenizer_initialization(model: str):
# token ids.
llm = LLM(model=model, skip_tokenizer_init=True)
sampling_params = SamplingParams(prompt_logprobs=True, detokenize=True)
with pytest.raises(ValueError) as err:
with pytest.raises(ValueError, match="cannot pass text prompts when"):
llm.generate("abc", sampling_params)
assert "prompts must be None if" in str(err.value)
outputs = llm.generate({"prompt_token_ids": [1, 2, 3]},
sampling_params=sampling_params)
assert len(outputs) > 0