diff --git a/tests/models/multimodal/pooling/test_jinavl_reranker.py b/tests/models/multimodal/pooling/test_jinavl_reranker.py index 05433e7269..dca2403e35 100644 --- a/tests/models/multimodal/pooling/test_jinavl_reranker.py +++ b/tests/models/multimodal/pooling/test_jinavl_reranker.py @@ -14,11 +14,6 @@ from vllm.entrypoints.pooling.scoring.typing import ScoreMultiModalParam from ....conftest import HfRunner, VllmRunner -pytestmark = pytest.mark.skip( - reason="jinaai/jina-reranker-m0 custom code is incompatible with " - "transformers v5 (missing all_tied_weights_keys)" -) - MODELS = ["jinaai/jina-reranker-m0"] MM_PROCESSOR_KWARGS = { @@ -29,8 +24,8 @@ MM_PROCESSOR_KWARGS = { LIMIT_MM_PER_PROMPT = {"image": 2} CHECKPOINT_TO_HF_MAPPER = { - "visual.": "model.visual.", - "model.": "model.language_model.", + r"^visual\.": "model.visual.", + r"^model\.(?!language_model\.|visual\.)": "model.language_model.", } HANDELSBLATT_IMAGE_URL = ( diff --git a/vllm/model_executor/models/config.py b/vllm/model_executor/models/config.py index 4073512d96..8843909300 100644 --- a/vllm/model_executor/models/config.py +++ b/vllm/model_executor/models/config.py @@ -546,6 +546,7 @@ class JinaVLForSequenceClassificationConfig(VerifyAndUpdateConfig): def verify_and_update_model_config(model_config: "ModelConfig") -> None: config = model_config.hf_config config.num_labels = 1 + config.get_text_config().num_labels = 1 pooler_config = model_config.pooler_config assert pooler_config is not None if pooler_config.logit_mean is None: