|
@@ -115,7 +115,7 @@ def remap_messages(messages: List[Message]) -> List[Message]:
|
|
|
|
|
|
def build_prompt(tokenizer, _messages: List[Message]):
|
|
def build_prompt(tokenizer, _messages: List[Message]):
|
|
messages = remap_messages(_messages)
|
|
messages = remap_messages(_messages)
|
|
- prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
|
|
|
|
|
|
+ prompt = tokenizer.apply_chat_template([m.to_dict() for m in messages], tokenize=False, add_generation_prompt=True)
|
|
image_str = None
|
|
image_str = None
|
|
for message in messages:
|
|
for message in messages:
|
|
if not isinstance(message.content, list):
|
|
if not isinstance(message.content, list):
|