ollama also accepts PNG

2024-04-08 03:35:02 +02:00 · 2024-04-08 03:35:02 +02:00 · 82a4232dce
commit 82a4232dce
parent 59ed4fb51e
1 changed files with 1 additions and 1 deletions
--- a/litellm/llms/ollama.py
+++ b/litellm/llms/ollama.py
@ -136,7 +136,7 @@ def _convert_image(image):
        image = image.split(",")[-1]
    try:
        image_data = Image.open(io.BytesIO(base64.b64decode(image)))
-        if image_data.format == "JPEG":
+        if image_data.format in ["JPEG", "PNG"]:
            return image
    except:
        return orig