From a789e0f2634563b9eb9b325a975c264b9fa97e58 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: 53cade69caed82b470fdb249274f03ee34af3100 --- 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="")