Commit 69b8bd4b authored by wang.yuqi's avatar wang.yuqi Committed by khluu
Browse files

[CI Failure] pin colmodernvbert revision (#38612)


Signed-off-by: default avatarwang.yuqi <yuqi.wang@daocloud.io>
Signed-off-by: default avatarwang.yuqi <noooop@126.com>
Co-authored-by: default avatargemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
(cherry picked from commit 719735d6)
parent 12449f94
...@@ -15,6 +15,10 @@ from vllm.entrypoints.pooling.score.utils import compute_maxsim_score ...@@ -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): ...@@ -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): ...@@ -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): ...@@ -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): ...@@ -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,
......
...@@ -636,6 +636,7 @@ _LATE_INTERACTION_EXAMPLE_MODELS = { ...@@ -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(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment