biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

[CI/Build] Bump test transformers version (#10106)

Browse Source 
Signed-off-by: Isotr0py <2037008807@qq.com>
Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>
Co-authored-by: DarkLight1337 <tlleungac@connect.ust.hk>
This commit is contained in:
Isotr0py
2024-12-06 00:05:52 +08:00
committed by GitHub
parent 571da8fc43
commit 998eeafe58
5 changed files with 3 additions and 35 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/models/decoder_only/vision_language/test_pixtral.py
View File

@@ -228,7 +228,7 @@ def test_model_engine(vllm_runner, model: str, dtype: str) -> None:
name_1="output")
@large_gpu_test(min_gb=24)
@large_gpu_test(min_gb=48)
@pytest.mark.parametrize(
"prompt,expected_ranges",
[(_create_engine_inputs_hf(IMG_URLS[:1]), [{