diff --git a/litellm/llms/ollama.py b/litellm/llms/ollama.py index 860875a46b..670c565904 100644 --- a/litellm/llms/ollama.py +++ b/litellm/llms/ollama.py @@ -120,8 +120,8 @@ class OllamaConfig: and v is not None } -# ollama wants plain base64 jpeg files as images. strip any leading dataURI -# and convert to jpeg. +# ollama wants plain base64 jpeg/png files as images. strip any leading dataURI +# and convert to jpeg if necessary. def _convert_image(image): import base64, io try: