Skip to content

Commit 4794c8c

Browse files
committed
Added support when using the keyword 'audio' instead of 'audio_url'.
1 parent 3208891 commit 4794c8c

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

llama_cpp/llama_chat_format.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2996,13 +2996,13 @@ def _get_media_items(self, messages: List[llama_types.ChatCompletionRequestMessa
29962996
media_items.append({"url": url, "type": "image"})
29972997

29982998
# 2. Audio Processing
2999-
elif content_type in ["audio_url", "input_audio"]:
2999+
elif content_type in ["audio", "audio_url", "input_audio"]:
30003000
if not self.is_support_audio:
30013001
raise ValueError(f"{self.log_prefix}: This mmproj model instance does not support audio inputs.")
30023002

30033003
# Case A: Handle custom/forward-compatible audio_url format
3004-
if content_type == "audio_url":
3005-
audio_url = content["audio_url"]
3004+
if content_type == "audio_url" or content_type == "audio":
3005+
audio_url = content[content_type]
30063006
url = audio_url if isinstance(audio_url, str) else audio_url["url"]
30073007
media_items.append({"url": url, "type": "audio"})
30083008
# Case B: Handle OpenAI standard input_audio format

0 commit comments

Comments
 (0)