Skip to content

Commit 7df9483

Browse files
committed
Update moondream chat format
1 parent 2fd41f9 commit 7df9483

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

llama_cpp/llama_chat_format.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2479,27 +2479,27 @@ def from_pretrained(
24792479

24802480
class MoondreamChatHanlder(Llava15ChatHandler):
24812481
# Chat Format:
2482-
# <image>\n\nQuestion: {prompt}\n\nAnswer:
2482+
# f"<image>\n\n{chat_history}Question: {question}\n\nAnswer:"
24832483
CHAT_FORMAT = (
24842484
"{% for message in messages %}"
24852485
"{% if message.role == 'user' %}"
24862486
"{% if message.content is iterable %}"
24872487
"{% for content in message.content %}"
24882488
"{% if content.type == 'image_url' %}"
2489-
"{{ content.image_url }}"
2489+
"{{ content.image_url }}\n\n"
24902490
"{% endif %}"
24912491
"{% if content.type == 'text' %}"
2492-
"Question: {{ content.text }}"
2492+
"Question: {{ content.text }}\n\n"
24932493
"{% endif %}"
24942494
"{% endfor %}"
24952495
"{% endif %}"
24962496
"{% endif %}"
24972497
"{% if message.role == 'assistant' %}"
2498-
"Answer: {{ message.content }}"
2498+
"Answer:{{ message.content }}"
24992499
"{% endif %}"
25002500
"{% endfor %}"
25012501
"{% if add_generation_prompt %}"
2502-
"Answer: "
2502+
"Answer:"
25032503
"{% endif %}"
25042504
)
25052505

0 commit comments

Comments
 (0)