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

Rename image_url to image_file (#15)

Browse Source
This commit is contained in:
Christopher Chou
2024-01-16 15:41:30 -08:00
committed by GitHub
parent e71d4ab3f9
commit 5b27a1dce4
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/sglang/srt/utils.py
View File

@@ -209,7 +209,7 @@ def load_image(image_file):
elif image_file.lower().endswith(("png", "jpg", "jpeg", "webp", "gif")): elif image_file.lower().endswith(("png", "jpg", "jpeg", "webp", "gif")):
image = Image.open(image_file) image = Image.open(image_file)
elif image_file.startswith("data:"): elif image_file.startswith("data:"):
image_file = image_url.split(",")[1] image_file = image_file.split(",")[1]
image = Image.open(BytesIO(base64.b64decode(image_file))) image = Image.open(BytesIO(base64.b64decode(image_file)))
else: else:
image = Image.open(BytesIO(base64.b64decode(image_file))) image = Image.open(BytesIO(base64.b64decode(image_file)))