mirror of
https://github.com/wassname/vllm.git
synced 2026-07-14 11:19:39 +08:00
10 lines
186 B
Python
10 lines
186 B
Python
import pytest
|
|
|
|
from vllm import LLM
|
|
|
|
|
|
def test_empty_prompt():
|
|
llm = LLM(model="gpt2")
|
|
with pytest.raises(ValueError, match='Prompt cannot be empty'):
|
|
llm.generate([""])
|