mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2025-12-10 06:25:01 +08:00
52c2a8d4ad
Signed-off-by: Woosuk Kwon <woosuk@thinkingmachines.ai> Signed-off-by: Woosuk Kwon <woosuk.kwon@berkeley.edu>
20 lines
574 B
Python
20 lines
574 B
Python
# SPDX-License-Identifier: Apache-2.0
|
|
# SPDX-FileCopyrightText: Copyright contributors to the vLLM project
|
|
|
|
import pytest
|
|
|
|
from vllm import LLM
|
|
|
|
|
|
def test_empty_prompt():
|
|
llm = LLM(model="openai-community/gpt2", enforce_eager=True)
|
|
with pytest.raises(ValueError, match='decoder prompt cannot be empty'):
|
|
llm.generate([""])
|
|
|
|
|
|
@pytest.mark.skip_v1
|
|
def test_out_of_vocab_token():
|
|
llm = LLM(model="openai-community/gpt2", enforce_eager=True)
|
|
with pytest.raises(ValueError, match='out of vocabulary'):
|
|
llm.generate({"prompt_token_ids": [999999]})
|