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

vllm-project · Mar 31, 2024 · 563c1d7 · 563c1d7
1 parent 9c82a1b
commit 563c1d7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/models/test_marlin.py b/tests/models/test_marlin.py
@@ -47,7 +47,7 @@ class ModelPair:
 @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,