From 8c79fe6a5aa6aab460f5196021f5df293005114e Mon Sep 17 00:00:00 2001 From: fzc8578 <1428195643@qq.com> Date: Sat, 11 Jan 2025 11:55:30 +0800 Subject: [PATCH] add cpm_o test Former-commit-id: 10073319b4215be900744a28a61bd442e70143cc --- tests/data/test_mm_plugin.py | 1 + 1 file changed, 1 insertion(+) 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="")