diff --git a/tests/data/test_mm_plugin.py b/tests/data/test_mm_plugin.py index ea92f0df..9e8372a7 100644 --- a/tests/data/test_mm_plugin.py +++ b/tests/data/test_mm_plugin.py @@ -127,6 +127,7 @@ def test_base_plugin(): _check_plugin(**check_inputs) +@pytest.mark.skipif(not HF_TOKEN, reason="Gated model.") def test_cpm_o_plugin(): tokenizer_module = _load_tokenizer_module(model_name_or_path="openbmb/MiniCPM-V-2_6") cpm_o_plugin = get_mm_plugin(name="cpm_o", image_token="")