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

Handle grayscale images in expand2square (#97)

Browse Source
This commit is contained in:
isaac-vidas
2024-01-24 19:23:11 -05:00
committed by GitHub
parent d3fc86a43e
commit 0c457bae8f
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
python/sglang/srt/mm_utils.py
View File

@@ -163,7 +163,9 @@ def expand2square(pil_img, background_color):
width, height = pil_img.size width, height = pil_img.size
if width == height: if width == height:
return pil_img return pil_img
elif width > height: if pil_img.mode == "L":
pil_img = pil_img.convert("RGB")
if width > height:
result = Image.new(pil_img.mode, (width, width), background_color) result = Image.new(pil_img.mode, (width, width), background_color)
result.paste(pil_img, (0, (width - height) // 2)) result.paste(pil_img, (0, (width - height) // 2))
return result return result