[Frontend] Clarify model_type error messages (#9345)

44eaa5a5 · Steve Grubb · GitHub · 169b5306 · 44eaa5a5
Unverified Commit 44eaa5a5 authored Oct 15, 2024 by Steve Grubb Committed by GitHub Oct 14, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

vllm/entrypoints/chat_utils.py vllm/entrypoints/chat_utils.py +3 -3

No files found.
--- a/vllm/entrypoints/chat_utils.py
+++ b/vllm/entrypoints/chat_utils.py
@@ -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}")