Skip to content

Commit 62efbd7

Browse files
Merge pull request #2201 from Lunik/fix/ollama-vision
🐛 fix: Ollama vision models call arguments (like : llava)
2 parents db797f1 + cee2069 commit 62efbd7

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

litellm/llms/ollama.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -147,6 +147,7 @@ def get_ollama_response(
147147

148148
stream = optional_params.pop("stream", False)
149149
format = optional_params.pop("format", None)
150+
images = optional_params.pop("images", None)
150151
data = {
151152
"model": model,
152153
"prompt": prompt,
@@ -155,6 +156,8 @@ def get_ollama_response(
155156
}
156157
if format is not None:
157158
data["format"] = format
159+
if images is not None:
160+
data["images"] = images
158161

159162
## LOGGING
160163
logging_obj.pre_call(

0 commit comments

Comments
 (0)