xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2026-06-09 11:02:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Robert Shaw 2024-03-30 21:18:34 -05:00 committed by GitHub
parent 9c82a1bec3
commit 563c1d7ec5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/test_marlin.py
View File

@ -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,