Rename image_url to image_file (#15)

5b27a1dc · Christopher Chou · GitHub · e71d4ab3 · 5b27a1dc
Unverified Commit 5b27a1dc authored Jan 16, 2024 by Christopher Chou Committed by GitHub Jan 16, 2024
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

python/sglang/srt/utils.py python/sglang/srt/utils.py +1 -1

No files found.
--- a/python/sglang/srt/utils.py
+++ b/python/sglang/srt/utils.py
@@ -209,7 +209,7 @@ def load_image(image_file):
    elif image_file.lower().endswith(("png", "jpg", "jpeg", "webp", "gif")):
        image = Image.open(image_file)
    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)))
    else:
        image = Image.open(BytesIO(base64.b64decode(image_file)))