[Misc] Update benchmark to support image_url file or http (#10287)

Signed-off-by: rbbang <anjaehyun87@gmail.com>
This commit is contained in:
Jaehyun An 2024-11-16 19:15:40 +09:00 committed by GitHub
parent 1d75472626
commit 8b6725b0cf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -251,6 +251,19 @@ def sample_hf_requests(
"url": f"data:image/jpeg;base64,{image_base64}"
},
}
elif "image" in data and isinstance(data["image"], str):
if (data["image"].startswith("http://") or \
data["image"].startswith("file://")):
image_url = data["image"]
else:
image_url = f"file://{data['image']}"
mm_content = {
"type": "image_url",
"image_url": {
"url": image_url
},
}
else:
mm_content = None