Skip to content

Commit

Permalink
[ci/Build] Skip Chameleon for transformers 4.46.0 on broadcast test v…
Browse files Browse the repository at this point in the history
  • Loading branch information
khluu authored Oct 25, 2024
1 parent c91ed47 commit 9f7b4ba
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions tests/models/decoder_only/vision_language/test_broadcast.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,5 @@
import pytest
import transformers

from ....utils import multi_gpu_test

Expand All @@ -23,6 +24,9 @@ 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"):
pytest.skip("Model broken in HF, "
"see huggingface/transformers#34379")
from .test_chameleon import models, run_test # type: ignore[no-redef]
else:
raise NotImplementedError(f"Unsupported model: {model}")
Expand Down

0 comments on commit 9f7b4ba

Please sign in to comment.