From 81653d968842d2ec51b2642b6b5d83786271f9af Mon Sep 17 00:00:00 2001 From: Simon Mo Date: Wed, 13 Mar 2024 17:02:21 -0700 Subject: [PATCH] [Hotfix] [Debug] test_openai_server.py::test_guided_regex_completion (#3383) --- .buildkite/test-pipeline.yaml | 2 +- requirements.txt | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/.buildkite/test-pipeline.yaml b/.buildkite/test-pipeline.yaml index 42a1eacb..6a130f6f 100644 --- a/.buildkite/test-pipeline.yaml +++ b/.buildkite/test-pipeline.yaml @@ -13,7 +13,7 @@ steps: - label: Basic Correctness Test command: pytest -v -s --forked basic_correctness - + - label: Core Test command: pytest -v -s core diff --git a/requirements.txt b/requirements.txt index 05ec2e80..d6c33ad8 100644 --- a/requirements.txt +++ b/requirements.txt @@ -12,5 +12,5 @@ pydantic >= 2.0 # Required for OpenAI server. prometheus_client >= 0.18.0 pynvml == 11.5.0 triton >= 2.1.0 -outlines >= 0.0.27 +outlines == 0.0.34 cupy-cuda12x == 12.1.0 # Required for CUDA graphs. CUDA 11.8 users should install cupy-cuda11x instead.