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 "" in test_prompt: import re - image_match = re.search(r"(.*?)(.*)", test_prompt) + image_match = re.search(r"(.*?)(.*)", 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 "" in prompt: import re - image_match = re.search(r"(.*?)(.*)", prompt) + image_match = re.search(r"(.*?)(.*)", 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: