add cpm_o test

Former-commit-id: 53cade69caed82b470fdb249274f03ee34af3100
2025-01-11 11:55:30 +08:00
parent f9ee00b6b6
commit a789e0f263
1 changed files with 1 additions and 0 deletions
--- 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="<image>")