EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

fix: fix unsupported palette mode of images in bench_serving for mmmu (#9206)

Browse Source
This commit is contained in:
Mick
2025-08-15 09:44:46 +08:00
committed by GitHub
parent 392de007cb
commit 584e1ab2d0
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
python/sglang/bench_serving.py
View File

@@ -864,11 +864,11 @@ def sample_mmmu_requests(
if image.mode == "RGBA": if image.mode == "RGBA":
image = image.convert("RGB") image = image.convert("RGB")
# Encode image to base64 # Encode image to base64 (save as PNG to support palette/alpha modes)
buffered = io.BytesIO() buffered = io.BytesIO()
image.save(buffered, format="JPEG") image.save(buffered, format="PNG")
img_str = pybase64.b64encode(buffered.getvalue()).decode("utf-8") img_str = pybase64.b64encode(buffered.getvalue()).decode("utf-8")
image_data = f"data:image/jpeg;base64,{img_str}" image_data = f"data:image/png;base64,{img_str}"
else: else:
continue continue