[CI/Build] Make Marlin Tests Green (#3753)

2024-03-30 21:18:34 -05:00 · 2024-03-30 21:18:34 -05:00 · 563c1d7ec5
commit 563c1d7ec5
parent 9c82a1bec3
1 changed files with 1 additions and 1 deletions
--- a/tests/models/test_marlin.py
+++ b/tests/models/test_marlin.py
@ -47,7 +47,7 @@ model_pairs = [
@pytest.mark.parametrize("model_pair", model_pairs)
@pytest.mark.parametrize("dtype", ["half"])
@pytest.mark.parametrize("max_tokens", [32])
-@pytest.mark.parametrize("num_logprobs", [3])
+@pytest.mark.parametrize("num_logprobs", [5])
 def test_models(
    vllm_runner,
    example_prompts,