From adfb35ac8245a382df443719164221a57b43316f Mon Sep 17 00:00:00 2001 From: Michael Goin Date: Tue, 29 Oct 2024 14:12:43 -0400 Subject: [PATCH] [CI][Bugfix] Skip chameleon for transformers 4.46.1 (#9808) --- tests/models/decoder_only/vision_language/test_broadcast.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/models/decoder_only/vision_language/test_broadcast.py b/tests/models/decoder_only/vision_language/test_broadcast.py index fd7af4a8b0b29..38c4a95de16f4 100644 --- a/tests/models/decoder_only/vision_language/test_broadcast.py +++ b/tests/models/decoder_only/vision_language/test_broadcast.py @@ -24,7 +24,7 @@ def test_models(hf_runner, vllm_runner, image_assets, elif model.startswith("llava-hf/llava-v1.6"): from .test_llava_next import models, run_test # type: ignore[no-redef] elif model.startswith("facebook/chameleon"): - if transformers.__version__.startswith("4.46.0"): + if transformers.__version__.startswith("4.46"): pytest.skip("Model broken in HF, " "see huggingface/transformers#34379") from .test_chameleon import models, run_test # type: ignore[no-redef]