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

Signed-off-by: DarkLight1337 <tlleungac@connect.ust.hk>
2024-12-11 13:02:02 +08:00
parent e39400a4b6
commit 2e33fe4191
3 changed files with 3 additions and 12 deletions
--- a/tests/models/embedding/vision_language/test_llava_next.py
+++ b/tests/models/embedding/vision_language/test_llava_next.py
@@ -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)