biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

[CI Failure] pin colmodernvbert revision (#38612)

Browse Source 
Signed-off-by: wang.yuqi <yuqi.wang@daocloud.io>
Signed-off-by: wang.yuqi <noooop@126.com>
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
(cherry picked from commit 719735d6c5)
This commit is contained in:
wang.yuqi
2026-03-31 18:54:54 +08:00
committed by khluu
parent 12449f9492
commit 69b8bd4b33
2 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
tests/models/multimodal/pooling/test_colmodernvbert.py
View File

@@ -15,6 +15,10 @@ from vllm.entrypoints.pooling.score.utils import compute_maxsim_score
MODEL_NAME = "ModernVBERT/colmodernvbert-merged" MODEL_NAME = "ModernVBERT/colmodernvbert-merged"
COLBERT_DIM = 128 COLBERT_DIM = 128
DTYPE = "half" 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).""" """Text query produces per-token embeddings with shape (seq_len, 128)."""
with vllm_runner( with vllm_runner(
MODEL_NAME, MODEL_NAME,
revision=REVISION,
runner="pooling", runner="pooling",
dtype=DTYPE, dtype=DTYPE,
enforce_eager=True, enforce_eager=True,
@@ -49,6 +54,7 @@ def test_colmodernvbert_text_relevance_ordering(vllm_runner):
with vllm_runner( with vllm_runner(
MODEL_NAME, MODEL_NAME,
revision=REVISION,
runner="pooling", runner="pooling",
dtype=DTYPE, dtype=DTYPE,
enforce_eager=True, enforce_eager=True,
@@ -66,6 +72,7 @@ def test_colmodernvbert_text_late_interaction(vllm_runner):
with vllm_runner( with vllm_runner(
MODEL_NAME, MODEL_NAME,
revision=REVISION,
runner="pooling", runner="pooling",
dtype=DTYPE, dtype=DTYPE,
enforce_eager=True, 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.""" """Image input produces per-token embeddings including vision tokens."""
with vllm_runner( with vllm_runner(
MODEL_NAME, MODEL_NAME,
revision=REVISION,
runner="pooling", runner="pooling",
dtype=DTYPE, dtype=DTYPE,
enforce_eager=True, enforce_eager=True,

1
tests/models/registry.py
View File

@@ -636,6 +636,7 @@ _LATE_INTERACTION_EXAMPLE_MODELS = {
# [Multimodal] # [Multimodal]
"ColModernVBertForRetrieval": _HfExamplesInfo( "ColModernVBertForRetrieval": _HfExamplesInfo(
"ModernVBERT/colmodernvbert-merged", "ModernVBERT/colmodernvbert-merged",
revision="4a0a9f3ac7a7992fec410bfa8e3d080ac9a5bcee",
), ),
"ColPaliForRetrieval": _HfExamplesInfo("vidore/colpali-v1.3-hf"), "ColPaliForRetrieval": _HfExamplesInfo("vidore/colpali-v1.3-hf"),
"ColQwen3": _HfExamplesInfo( "ColQwen3": _HfExamplesInfo(