A high-throughput and memory-efficient inference and serving engine for LLMs
Mirrored from https://github.com/tduncan-lum/Nvidia-Optimization-Stack---VLLM.git
Synced Nov 15, 19:45 UTC
[CI] Fix missing example model id in processor test (#14787)
Signed-off-by: Roger Wang <ywang@roblox.com>
Roger Wang committed
Mar 13, 2025 at 22:52 UTC
0c2af17c766f1e9c3e6827c0a04a07efe9e3ae9b
2 files changed
+2
-2
tests/models/multimodal/processing/test_common.py
+1
-1
@@ -215,7 +215,7 @@ def test_processing_correctness(
215
216
217
# yapf: disable
218
-@pytest.mark.parametrize("model_id", ["microsoft/Phi-3-vision-128k-instruct"])
218
+@pytest.mark.parametrize("model_id", ["microsoft/Phi-3.5-vision-instruct"])
219
@pytest.mark.parametrize("hit_rate", [0.3, 0.5, 1.0])
220
@pytest.mark.parametrize("num_batches", [32])
221
@pytest.mark.parametrize("simplify_rate", [1.0])
tests/models/registry.py
+1
-1
@@ -274,7 +274,7 @@ _MULTIMODAL_EXAMPLE_MODELS = {
274
trust_remote_code=True),
275
"PaliGemmaForConditionalGeneration": _HfExamplesInfo("google/paligemma-3b-mix-224", # noqa: E501
276
extras={"v2": "google/paligemma2-3b-ft-docci-448"}), # noqa: E501
277
- "Phi3VForCausalLM": _HfExamplesInfo("microsoft/Phi-3-vision-128k-instruct",
277
+ "Phi3VForCausalLM": _HfExamplesInfo("microsoft/Phi-3.5-vision-instruct",
278
trust_remote_code=True),
279
"Phi4MMForCausalLM": _HfExamplesInfo("microsoft/Phi-4-multimodal-instruct",
280
trust_remote_code=True),