[Structured Output][CI] Add test for outlines backend for structured output in CI (#2283)
### What this PR does / why we need it?
Add test for `outlines` backend for structured output in CI.
### Does this PR introduce _any_ user-facing change?
No.
### How was this patch tested?
Tests have all passed with:
```bash
pytest -sv tests/e2e/singlecard/test_guided_decoding.py
```
- vLLM version: v0.10.0
- vLLM main:
53415653ff
---------
Signed-off-by: shen-shanshan <467638484@qq.com>
This commit is contained in:
@@ -30,7 +30,7 @@ from tests.e2e.conftest import VllmRunner
|
|||||||
os.environ["PYTORCH_NPU_ALLOC_CONF"] = "max_split_size_mb:256"
|
os.environ["PYTORCH_NPU_ALLOC_CONF"] = "max_split_size_mb:256"
|
||||||
MODEL_NAME = "Qwen/Qwen2.5-0.5B-Instruct"
|
MODEL_NAME = "Qwen/Qwen2.5-0.5B-Instruct"
|
||||||
|
|
||||||
GuidedDecodingBackend = ["xgrammar", "guidance"]
|
GuidedDecodingBackend = ["xgrammar", "guidance", "outlines"]
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture(scope="module")
|
@pytest.fixture(scope="module")
|
||||||
@@ -120,6 +120,9 @@ def test_guided_json_completion(guided_decoding_backend: str,
|
|||||||
|
|
||||||
@pytest.mark.parametrize("guided_decoding_backend", GuidedDecodingBackend)
|
@pytest.mark.parametrize("guided_decoding_backend", GuidedDecodingBackend)
|
||||||
def test_guided_regex(guided_decoding_backend: str, sample_regex):
|
def test_guided_regex(guided_decoding_backend: str, sample_regex):
|
||||||
|
if guided_decoding_backend == "outlines":
|
||||||
|
pytest.skip("Outlines doesn't support regex-based guided decoding.")
|
||||||
|
|
||||||
sampling_params = SamplingParams(
|
sampling_params = SamplingParams(
|
||||||
temperature=0.8,
|
temperature=0.8,
|
||||||
top_p=0.95,
|
top_p=0.95,
|
||||||
|
|||||||
Reference in New Issue
Block a user