Skip to content

GitLab

Unverified Commit 0c2af17c authored by Roger Wang's avatar Roger Wang Committed by GitHub
Browse files

[CI] Fix missing example model id in processor test (#14787) 


Signed-off-by: default avatarRoger Wang <ywang@roblox.com>
parent a6e0d096
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tests/models/multimodal/processing/test_common.py
View file @ 0c2af17c
......@@ -215,7 +215,7 @@ def test_processing_correctness(
# yapf: disable
@pytest.mark.parametrize("model_id", ["microsoft/Phi-3-vision-128k-instruct"])
@pytest.mark.parametrize("model_id", ["microsoft/Phi-3.5-vision-instruct"])
@pytest.mark.parametrize("hit_rate", [0.3, 0.5, 1.0])
@pytest.mark.parametrize("num_batches", [32])
@pytest.mark.parametrize("simplify_rate", [1.0])
......
tests/models/registry.py
View file @ 0c2af17c
......@@ -274,7 +274,7 @@ _MULTIMODAL_EXAMPLE_MODELS = {
trust_remote_code=True),
"PaliGemmaForConditionalGeneration": _HfExamplesInfo("google/paligemma-3b-mix-224", # noqa: E501
extras={"v2": "google/paligemma2-3b-ft-docci-448"}), # noqa: E501
"Phi3VForCausalLM": _HfExamplesInfo("microsoft/Phi-3-vision-128k-instruct",
"Phi3VForCausalLM": _HfExamplesInfo("microsoft/Phi-3.5-vision-instruct",
trust_remote_code=True),
"Phi4MMForCausalLM": _HfExamplesInfo("microsoft/Phi-4-multimodal-instruct",
trust_remote_code=True),
......
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