From 446ea3327735e125e19d37b6a2c25aed7ead68f3 Mon Sep 17 00:00:00 2001 From: Li Bo Date: Thu, 19 Sep 2024 16:31:48 +0800 Subject: [PATCH] fix: creat new dict everytime for putting new frame (#1464) --- .../llava_onevision/http_llava_onevision_test.py | 15 ++++++--------- 1 file changed, 6 insertions(+), 9 deletions(-) diff --git a/examples/runtime/llava_onevision/http_llava_onevision_test.py b/examples/runtime/llava_onevision/http_llava_onevision_test.py index 2c7c2bd38..ce59f4061 100644 --- a/examples/runtime/llava_onevision/http_llava_onevision_test.py +++ b/examples/runtime/llava_onevision/http_llava_onevision_test.py @@ -217,17 +217,14 @@ def prepare_video_messages(video_path): base64_frames.append(base64_str) messages = [{"role": "user", "content": []}] - frame_format = { - "type": "image_url", - "image_url": {"url": "data:image/jpeg;base64,{}"}, - "modalities": "video", - } for base64_frame in base64_frames: - frame_format["image_url"]["url"] = "data:image/jpeg;base64,{}".format( - base64_frame - ) - messages[0]["content"].append(frame_format.copy()) + frame_format = { + "type": "image_url", + "image_url": {"url": f"data:image/jpeg;base64,{base64_frame}"}, + "modalities": "video", + } + messages[0]["content"].append(frame_format) prompt = {"type": "text", "text": "Please describe the video in detail."} messages[0]["content"].append(prompt)