frozenleaves/vllm
1
0
Fork 0
mirror of https://github.com/vllm-project/vllm.git synced 2025-10-20 14:53:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Frontend] Clarify model_type error messages (#9345)

Browse Source
This commit is contained in:
Steve Grubb
2024-10-15 00:29:01 -04:00
committed by GitHub
parent 169b530607
commit 44eaa5a5d9
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
vllm/entrypoints/chat_utils.py
View File

@ -166,15 +166,15 @@ class BaseMultiModalItemTracker(ABC, Generic[_T]):
if model_type == "molmo":
return ""
raise TypeError(f"Unknown model type: {model_type}")
raise TypeError(f"Unknown {modality} model type: {model_type}")
elif modality == "audio":
if model_type == "ultravox":
return "<|reserved_special_token_0|>"
raise TypeError(f"Unknown model type: {model_type}")
raise TypeError(f"Unknown {modality} model type: {model_type}")
elif modality == "video":
if model_type == "qwen2_vl":
return "<|vision_start|><|video_pad|><|vision_end|>"
raise TypeError(f"Unknown model type: {model_type}")
raise TypeError(f"Unknown {modality} model type: {model_type}")
else:
raise TypeError(f"Unknown modality: {modality}")