From 3167d8dabcb29139e4af8aee4dca627344868281 Mon Sep 17 00:00:00 2001 From: bing Date: Fri, 24 May 2024 18:38:01 +0800 Subject: [PATCH] fix test bug in srt_llava_next_test.py (#470) --- examples/usage/llava/srt_llava_next_test.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/examples/usage/llava/srt_llava_next_test.py b/examples/usage/llava/srt_llava_next_test.py index 8a7a6267d..d077fb2a6 100644 --- a/examples/usage/llava/srt_llava_next_test.py +++ b/examples/usage/llava/srt_llava_next_test.py @@ -17,14 +17,14 @@ def image_qa(s, image, question): def single(): image_url = "https://farm4.staticflickr.com/3175/2653711032_804ff86d81_z.jpg" - pil_image = load_image(image_url) + pil_image, _ = load_image(image_url) state = image_qa.run(image=pil_image, question="What is this?", max_new_tokens=512) print(state["answer"], "\n") def stream(): image_url = "https://farm4.staticflickr.com/3175/2653711032_804ff86d81_z.jpg" - pil_image = load_image(image_url) + pil_image, _ = load_image(image_url) state = image_qa.run( image=pil_image, question="Please generate short caption for this image.", @@ -40,7 +40,7 @@ def stream(): def batch(): image_url = "https://farm4.staticflickr.com/3175/2653711032_804ff86d81_z.jpg" - pil_image = load_image(image_url) + pil_image, _ = load_image(image_url) states = image_qa.run_batch( [ {"image": pil_image, "question": "What is this?"},