Skip to content

GitLab

Unverified Commit 44eaa5a5 authored by Steve Grubb's avatar Steve Grubb Committed by GitHub
Browse files

[Frontend] Clarify model_type error messages (#9345)

parent 169b5306
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
vllm/entrypoints/chat_utils.py
View file @ 44eaa5a5
...@@ -166,15 +166,15 @@ class BaseMultiModalItemTracker(ABC, Generic[_T]): ...@@ -166,15 +166,15 @@ class BaseMultiModalItemTracker(ABC, Generic[_T]):
if model_type == "molmo": if model_type == "molmo":
return "" return ""
raise TypeError(f"Unknown model type: {model_type}") raise TypeError(f"Unknown {modality} model type: {model_type}")
elif modality == "audio": elif modality == "audio":
if model_type == "ultravox": if model_type == "ultravox":
return "<|reserved_special_token_0|>" 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": elif modality == "video":
if model_type == "qwen2_vl": if model_type == "qwen2_vl":
return "<|vision_start|><|video_pad|><|vision_end|>" 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: else:
raise TypeError(f"Unknown modality: {modality}") raise TypeError(f"Unknown modality: {modality}")
......
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