From 693dff5149334d0d928d4f7a9622f7ce86ce6822 Mon Sep 17 00:00:00 2001 From: KuntaiDu Date: Thu, 3 Oct 2024 21:52:34 +0000 Subject: [PATCH] Test CI fix: set block manager to v1 when running compatibility test --- tests/spec_decode/e2e/test_compatibility.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/spec_decode/e2e/test_compatibility.py b/tests/spec_decode/e2e/test_compatibility.py index 81f91c5e10b0..2fbdfaefbe9c 100644 --- a/tests/spec_decode/e2e/test_compatibility.py +++ b/tests/spec_decode/e2e/test_compatibility.py @@ -102,7 +102,7 @@ def test_spec_decode_xfail_spec_max_model_len(test_llm_generator): "num_speculative_tokens": 5, }]) @pytest.mark.parametrize("per_test_common_llm_kwargs", [{}]) -@pytest.mark.parametrize("test_llm_kwargs", [{}]) +@pytest.mark.parametrize("test_llm_kwargs", [{"use_v2_block_manager": False}]) @pytest.mark.parametrize("seed", [1]) def test_spec_decode_xfail_block_manager_v1(test_llm_generator): """Verify that speculative decoding with block manager v1 fails.