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

[CI/Build] Fix broken multimodal test (#10950)

Browse Source
This commit is contained in:
Cyrus Leung 2024-12-06 18:41:23 +08:00 committed by GitHub
parent b031a455a9
commit 222f5b082a
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -2,6 +2,7 @@ from typing import List, Type
import pytest import pytest
import torch.nn.functional as F import torch.nn.functional as F
import transformers
from transformers import AutoModelForVision2Seq from transformers import AutoModelForVision2Seq
from ....conftest import IMAGE_ASSETS, HfRunner, PromptImageInput, VllmRunner from ....conftest import IMAGE_ASSETS, HfRunner, PromptImageInput, VllmRunner
@ -85,6 +86,9 @@ def _run_test(
) )
@pytest.mark.skipif(transformers.__version__.startswith("4.46"),
reason="Model broken with changes in transformers 4.46")
@pytest.mark.core_model
@pytest.mark.parametrize("model", MODELS) @pytest.mark.parametrize("model", MODELS)
@pytest.mark.parametrize("dtype", ["half"]) @pytest.mark.parametrize("dtype", ["half"])
def test_models_text( def test_models_text(