Skip to content

GitLab

Unverified Commit 3167d8da authored by bing's avatar bing Committed by GitHub
Browse files

fix test bug in srt_llava_next_test.py (#470)

parent 0fafc560
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
examples/usage/llava/srt_llava_next_test.py
View file @ 3167d8da
...@@ -17,14 +17,14 @@ def image_qa(s, image, question): ...@@ -17,14 +17,14 @@ def image_qa(s, image, question):
def single(): def single():
image_url = "https://farm4.staticflickr.com/3175/2653711032_804ff86d81_z.jpg" 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) state = image_qa.run(image=pil_image, question="What is this?", max_new_tokens=512)
print(state["answer"], "\n") print(state["answer"], "\n")
def stream(): def stream():
image_url = "https://farm4.staticflickr.com/3175/2653711032_804ff86d81_z.jpg" 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( state = image_qa.run(
image=pil_image, image=pil_image,
question="Please generate short caption for this image.", question="Please generate short caption for this image.",
...@@ -40,7 +40,7 @@ def stream(): ...@@ -40,7 +40,7 @@ def stream():
def batch(): def batch():
image_url = "https://farm4.staticflickr.com/3175/2653711032_804ff86d81_z.jpg" 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( states = image_qa.run_batch(
[ [
{"image": pil_image, "question": "What is this?"}, {"image": pil_image, "question": "What is this?"},
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment