Skip to content

GitLab

Unverified Commit 91add85e authored by sroy745's avatar sroy745 Committed by GitHub
Browse files

Fix failing spec decode test (#9054)

parent 9aaf14c6
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tests/spec_decode/e2e/test_compatibility.py
View file @ 91add85e
...@@ -100,6 +100,7 @@ def test_spec_decode_xfail_spec_max_model_len(test_llm_generator): ...@@ -100,6 +100,7 @@ def test_spec_decode_xfail_spec_max_model_len(test_llm_generator):
"model": "JackFram/llama-68m", "model": "JackFram/llama-68m",
"speculative_model": "JackFram/llama-68m", "speculative_model": "JackFram/llama-68m",
"num_speculative_tokens": 5, "num_speculative_tokens": 5,
"use_v2_block_manager": False,
}]) }])
@pytest.mark.parametrize("per_test_common_llm_kwargs", [{}]) @pytest.mark.parametrize("per_test_common_llm_kwargs", [{}])
@pytest.mark.parametrize("test_llm_kwargs", [{}]) @pytest.mark.parametrize("test_llm_kwargs", [{}])
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment