From fb507b7b101ccd54dc27570e83d093979f08c962 Mon Sep 17 00:00:00 2001
From: Yikai Zhang <ykzhang@cs.wisc.edu>
Date: Sun, 1 Jun 2025 04:48:06 +0800
Subject: [PATCH] [FIX] mmmu bench serving result display error (#6525) (#6791)

---
 python/sglang/bench_serving.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/python/sglang/bench_serving.py b/python/sglang/bench_serving.py
index a02a3ec3a..f5578ef9f 100644
--- a/python/sglang/bench_serving.py
+++ b/python/sglang/bench_serving.py
@@ -1212,7 +1212,7 @@ async def benchmark(
     if "<image>" in test_prompt:
         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
         test_prompt = image_match.group(2) if image_match else test_prompt
     else:
@@ -1285,7 +1285,7 @@ async def benchmark(
         if "<image>" in prompt:
             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
             prompt = image_match.group(2) if image_match else prompt
         else: