Skip to content

Commit 67d7ee4

Browse files
DarkLight1337yangw-dev
authored andcommitted
[Misc] Update chat utils tests (vllm-project#16520)
Signed-off-by: DarkLight1337 <[email protected]> Signed-off-by: Yang Wang <[email protected]>
1 parent 37bc125 commit 67d7ee4

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

tests/entrypoints/test_chat_utils.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -25,6 +25,7 @@
2525

2626
PHI3V_MODEL_ID = "microsoft/Phi-3.5-vision-instruct"
2727
ULTRAVOX_MODEL_ID = "fixie-ai/ultravox-v0_5-llama-3_2-1b"
28+
QWEN2AUDIO_MODEL_ID = "Qwen/Qwen2-Audio-7B-Instruct"
2829
QWEN2VL_MODEL_ID = "Qwen/Qwen2-VL-2B-Instruct"
2930
QWEN25VL_MODEL_ID = "Qwen/Qwen2.5-VL-3B-Instruct"
3031
MLLAMA_MODEL_ID = "meta-llama/Llama-3.2-11B-Vision-Instruct"
@@ -841,13 +842,16 @@ def test_resolve_hf_chat_template(sample_json_schema, model, use_tools):
841842
assert isinstance(chat_template, str)
842843

843844

845+
# NOTE: Qwen2-Audio default chat template is specially defined inside
846+
# processor class instead of using `tokenizer_config.json`
844847
# yapf: disable
845848
@pytest.mark.parametrize(
846849
("model", "expected_format"),
847850
[(PHI3V_MODEL_ID, "string"),
848851
(QWEN2VL_MODEL_ID, "openai"),
849852
(QWEN25VL_MODEL_ID, "openai"),
850853
(ULTRAVOX_MODEL_ID, "string"),
854+
(QWEN2AUDIO_MODEL_ID, "openai"),
851855
(MLLAMA_MODEL_ID, "openai"),
852856
(LLAMA_GUARD_MODEL_ID, "openai")],
853857
)
@@ -900,10 +904,13 @@ def test_resolve_content_format_hf_defined(model, expected_format):
900904
("template_chatglm2.jinja", "string"),
901905
("template_chatml.jinja", "string"),
902906
("template_deepseek_vl2.jinja", "string"),
907+
("template_dse_qwen2_vl.jinja", "openai"),
903908
("template_falcon_180b.jinja", "string"),
904909
("template_falcon.jinja", "string"),
910+
("template_florence2.jinja", "string"),
905911
("template_inkbot.jinja", "string"),
906912
("template_llava.jinja", "string"),
913+
("template_teleflm.jinja", "string"),
907914
("template_vlm2vec.jinja", "openai"),
908915
("tool_chat_template_granite_20b_fc.jinja", "string"),
909916
("tool_chat_template_hermes.jinja", "string"),

0 commit comments

Comments
 (0)