Unverified Commit 9f7b4ba8 authored by Kevin H. Luu's avatar Kevin H. Luu Committed by GitHub
Browse files

[ci/Build] Skip Chameleon for transformers 4.46.0 on broadcast test #9675 (#9676)

parent c91ed47c
import pytest import pytest
import transformers
from ....utils import multi_gpu_test from ....utils import multi_gpu_test
...@@ -23,6 +24,9 @@ def test_models(hf_runner, vllm_runner, image_assets, ...@@ -23,6 +24,9 @@ def test_models(hf_runner, vllm_runner, image_assets,
elif model.startswith("llava-hf/llava-v1.6"): elif model.startswith("llava-hf/llava-v1.6"):
from .test_llava_next import models, run_test # type: ignore[no-redef] from .test_llava_next import models, run_test # type: ignore[no-redef]
elif model.startswith("facebook/chameleon"): 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] from .test_chameleon import models, run_test # type: ignore[no-redef]
else: else:
raise NotImplementedError(f"Unsupported model: {model}") raise NotImplementedError(f"Unsupported model: {model}")
......
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