xinyun/vllm
1
0
Fork 0
mirror of https://git.datalinker.icu/vllm-project/vllm.git synced 2025-12-16 12:05:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

[CI/Build] Check transformers v4.47 (#10991)

Browse Source 
Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>
This commit is contained in:
Cyrus Leung 2024-12-11 13:02:02 +08:00 committed by GitHub
parent e39400a4b6
commit 2e33fe4191
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 3 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
requirements-test.txt
View File

@ -506,7 +506,7 @@ tiktoken==0.7.0
# mistral-common
timm==1.0.11
# via -r requirements-test.in
tokenizers==0.20.3
tokenizers==0.21.0
# via transformers
torch==2.5.1
# via
@ -534,7 +534,7 @@ tqdm==4.66.6
# transformers
tqdm-multiprocess==0.0.11
# via lm-eval
transformers==4.46.3
transformers==4.47.0
# via
# lm-eval
# peft

9
tests/models/decoder_only/vision_language/mm_processor_kwargs/test_idefics3.py
View File

@ -3,7 +3,6 @@ from typing import Optional
import pytest
import torch
import transformers
from transformers import AutoImageProcessor, AutoTokenizer
from vllm.inputs import InputContext, token_inputs
@ -36,8 +35,6 @@ def get_max_idefics3_image_tokens():
return get_max_idefics3_image_tokens
@pytest.mark.skipif(transformers.__version__ < "4.46.0",
reason="Model introduced in HF >= 4.46.0")
@pytest.mark.parametrize("model", models)
@pytest.mark.parametrize("longest_edge", [None, 168, 336, 400, 2 * 336])
def test_input_mapper_override(model: str, image_assets: _ImageAssets,
@ -77,8 +74,6 @@ def test_input_mapper_override(model: str, image_assets: _ImageAssets,
assert torch.all(hf_result["pixel_values"] == vllm_result["pixel_values"])
@pytest.mark.skipif(transformers.__version__ < "4.46.0",
reason="Model introduced in HF >= 4.46.0")
@pytest.mark.parametrize("model", models)
@pytest.mark.parametrize("longest_edge, expected_max_tokens", [
(None, 2873),
@ -107,8 +102,6 @@ def test_max_tokens_override(get_max_idefics3_image_tokens, model: str,
assert expected_max_tokens == actual_max_tokens
@pytest.mark.skipif(transformers.__version__ < "4.46.0",
reason="Model introduced in HF >= 4.46.0")
@pytest.mark.parametrize("model", models)
@pytest.mark.parametrize("longest_edge, toks_per_img, num_imgs", [
(168, 169, 1),
@ -143,8 +136,6 @@ def test_dummy_data_override(dummy_data_for_idefics3, model: str,
assert img_tok_count == toks_per_img * num_imgs
@pytest.mark.skipif(transformers.__version__ < "4.46.0",
reason="Model introduced in HF >= 4.46.0")
@pytest.mark.parametrize("model", models)
@pytest.mark.parametrize("longest_edge,expected_toks_per_img,num_imgs", [
(336, 169 * (1**2 + 1), 1),

2
tests/models/embedding/vision_language/test_llava_next.py
View File

@ -86,7 +86,7 @@ def _run_test(
)
@pytest.mark.skipif(transformers.__version__.startswith("4.46"),
@pytest.mark.skipif(transformers.__version__ >= "4.46",
reason="Model broken with changes in transformers 4.46")
@pytest.mark.core_model
@pytest.mark.parametrize("model", MODELS)