Skip to content

GitLab

"git@developer.sourcefind.cn:change/sglang.git" did not exist on "880221bd3b3e56a4bc2268fe9a9f77f426accf6c"
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):
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?"},
......
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