Skip to content

GitLab

Unverified Commit 222f5b08 authored by Cyrus Leung's avatar Cyrus Leung Committed by GitHub
Browse files

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

parent b031a455
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 0 deletions
tests/models/embedding/vision_language/test_llava_next.py
View file @ 222f5b08
...@@ -2,6 +2,7 @@ from typing import List, Type ...@@ -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( ...@@ -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(
......
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