[CPU] Fix phi4-mm prompt issue in bench_serving (#9900)
This commit is contained in:
@@ -995,17 +995,25 @@ def sample_mmmu_requests(
|
|||||||
prompt = f"Question: {question}\n\nAnswer: "
|
prompt = f"Question: {question}\n\nAnswer: "
|
||||||
if apply_chat_template:
|
if apply_chat_template:
|
||||||
try:
|
try:
|
||||||
|
is_phi4_multimodal = (
|
||||||
|
"phi-4-multimodal" in tokenizer.name_or_path.lower()
|
||||||
|
)
|
||||||
|
if is_phi4_multimodal:
|
||||||
|
# <|endoftext10|> is the image token used in the phi-4-multimodal model.
|
||||||
|
content = prompt.replace("image 1", "<|endoftext10|>")
|
||||||
|
else:
|
||||||
|
content = [
|
||||||
|
{
|
||||||
|
"type": "image_url",
|
||||||
|
"image_url": {"url": image_data},
|
||||||
|
},
|
||||||
|
{"type": "text", "text": prompt},
|
||||||
|
]
|
||||||
prompt = tokenizer.apply_chat_template(
|
prompt = tokenizer.apply_chat_template(
|
||||||
[
|
[
|
||||||
{
|
{
|
||||||
"role": "user",
|
"role": "user",
|
||||||
"content": [
|
"content": content,
|
||||||
{
|
|
||||||
"type": "image_url",
|
|
||||||
"image_url": {"url": image_data},
|
|
||||||
},
|
|
||||||
{"type": "text", "text": prompt},
|
|
||||||
],
|
|
||||||
}
|
}
|
||||||
],
|
],
|
||||||
add_generation_prompt=True,
|
add_generation_prompt=True,
|
||||||
|
|||||||
Reference in New Issue
Block a user