[CI] Add Buildkite (#2355)

2024-01-14 12:37:58 -08:00
parent 9f659bf07f
commit 6e01e8c1c8
13 changed files with 192 additions and 37 deletions
--- a/tests/samplers/test_logprobs.py
+++ b/tests/samplers/test_logprobs.py
@@ -30,6 +30,7 @@ def test_get_prompt_logprobs(
                                          temperature=0.0)
    vllm_results = vllm_model.model.generate(
        example_prompts, sampling_params=vllm_sampling_params)
+    del vllm_model

    # Test whether logprobs are included in the results.
    for result in vllm_results: