From 69b8bd4b3363cd7f3ba99aed12712aee141e907a Mon Sep 17 00:00:00 2001 From: "wang.yuqi" Date: Tue, 31 Mar 2026 18:54:54 +0800 Subject: [PATCH] [CI Failure] pin colmodernvbert revision (#38612) Signed-off-by: wang.yuqi Signed-off-by: wang.yuqi Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com> (cherry picked from commit 719735d6c5f5dae14948c2150e6858351994b0b9) --- tests/models/multimodal/pooling/test_colmodernvbert.py | 8 ++++++++ tests/models/registry.py | 1 + 2 files changed, 9 insertions(+) diff --git a/tests/models/multimodal/pooling/test_colmodernvbert.py b/tests/models/multimodal/pooling/test_colmodernvbert.py index 01f3843c3..19551c60b 100644 --- a/tests/models/multimodal/pooling/test_colmodernvbert.py +++ b/tests/models/multimodal/pooling/test_colmodernvbert.py @@ -15,6 +15,10 @@ from vllm.entrypoints.pooling.score.utils import compute_maxsim_score MODEL_NAME = "ModernVBERT/colmodernvbert-merged" COLBERT_DIM = 128 DTYPE = "half" +# Fixme: +# Update colmodernvbert code to support the latest HF version +# and remove revision set. +REVISION = "4a0a9f3ac7a7992fec410bfa8e3d080ac9a5bcee" # ----------------------------------------------------------------------- @@ -26,6 +30,7 @@ def test_colmodernvbert_text_token_embed(vllm_runner): """Text query produces per-token embeddings with shape (seq_len, 128).""" with vllm_runner( MODEL_NAME, + revision=REVISION, runner="pooling", dtype=DTYPE, enforce_eager=True, @@ -49,6 +54,7 @@ def test_colmodernvbert_text_relevance_ordering(vllm_runner): with vllm_runner( MODEL_NAME, + revision=REVISION, runner="pooling", dtype=DTYPE, enforce_eager=True, @@ -66,6 +72,7 @@ def test_colmodernvbert_text_late_interaction(vllm_runner): with vllm_runner( MODEL_NAME, + revision=REVISION, runner="pooling", dtype=DTYPE, enforce_eager=True, @@ -92,6 +99,7 @@ def test_colmodernvbert_image_token_embed(vllm_runner, image_assets): """Image input produces per-token embeddings including vision tokens.""" with vllm_runner( MODEL_NAME, + revision=REVISION, runner="pooling", dtype=DTYPE, enforce_eager=True, diff --git a/tests/models/registry.py b/tests/models/registry.py index 4e13b49e4..3d4ecf8c2 100644 --- a/tests/models/registry.py +++ b/tests/models/registry.py @@ -636,6 +636,7 @@ _LATE_INTERACTION_EXAMPLE_MODELS = { # [Multimodal] "ColModernVBertForRetrieval": _HfExamplesInfo( "ModernVBERT/colmodernvbert-merged", + revision="4a0a9f3ac7a7992fec410bfa8e3d080ac9a5bcee", ), "ColPaliForRetrieval": _HfExamplesInfo("vidore/colpali-v1.3-hf"), "ColQwen3": _HfExamplesInfo(