[CI/Build][Misc] Update Pytest Marker for VLMs (#5623)

2024-06-18 06:10:04 -07:00
parent f0cc0e68e3
commit 4ad7b53e59
6 changed files with 8 additions and 8 deletions
--- a/tests/models/test_phi3v.py
+++ b/tests/models/test_phi3v.py
@@ -8,7 +8,7 @@ from vllm.utils import is_cpu

 from ..conftest import IMAGE_FILES

-pytestmark = pytest.mark.llava
+pytestmark = pytest.mark.vlm

 # The image token is placed before "user" on purpose so that the test can pass
 HF_IMAGE_PROMPTS = [