Skip to content

GitLab

Unverified Commit fb507b7b authored by Yikai Zhang's avatar Yikai Zhang Committed by GitHub
Browse files

[FIX] mmmu bench serving result display error (#6525) (#6791)

parent f90945c4
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
python/sglang/bench_serving.py
View file @ fb507b7b
...@@ -1212,7 +1212,7 @@ async def benchmark( ...@@ -1212,7 +1212,7 @@ async def benchmark(
if "<image>" in test_prompt: if "<image>" in test_prompt:
import re import re
image_match = re.search(r"<image>(.*?)</image>(.*)", test_prompt) image_match = re.search(r"<image>(.*?)</image>(.*)", test_prompt, re.DOTALL)
image_data = image_match.group(1) if image_match else None image_data = image_match.group(1) if image_match else None
test_prompt = image_match.group(2) if image_match else test_prompt test_prompt = image_match.group(2) if image_match else test_prompt
else: else:
...@@ -1285,7 +1285,7 @@ async def benchmark( ...@@ -1285,7 +1285,7 @@ async def benchmark(
if "<image>" in prompt: if "<image>" in prompt:
import re import re
image_match = re.search(r"<image>(.*?)</image>(.*)", prompt) image_match = re.search(r"<image>(.*?)</image>(.*)", prompt, re.DOTALL)
image_data = image_match.group(1) if image_match else None image_data = image_match.group(1) if image_match else None
prompt = image_match.group(2) if image_match else prompt prompt = image_match.group(2) if image_match else prompt
else: else:
......
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