Skip to content

GitLab

Unverified Commit 563c1d7e authored by Robert Shaw's avatar Robert Shaw Committed by GitHub
Browse files

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

parent 9c82a1be
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/models/test_marlin.py
View file @ 563c1d7e
...@@ -47,7 +47,7 @@ model_pairs = [ ...@@ -47,7 +47,7 @@ model_pairs = [
@pytest.mark.parametrize("model_pair", model_pairs) @pytest.mark.parametrize("model_pair", model_pairs)
@pytest.mark.parametrize("dtype", ["half"]) @pytest.mark.parametrize("dtype", ["half"])
@pytest.mark.parametrize("max_tokens", [32]) @pytest.mark.parametrize("max_tokens", [32])
@pytest.mark.parametrize("num_logprobs", [3]) @pytest.mark.parametrize("num_logprobs", [5])
def test_models( def test_models(
vllm_runner, vllm_runner,
example_prompts, example_prompts,
......
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