[CI][Bugfix] Skip chameleon for transformers 4.46.1 (#9808)

This commit is contained in:
Michael Goin 2024-10-29 14:12:43 -04:00 committed by GitHub
parent ac3d748dba
commit ab6f981671
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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]