scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-07 08:26:10 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 36

🐛 fix: Ollama vision models call arguments (like : llava)

Browse source 
Signed-off-by: Lunik <lunik@tiwabbit.fr>
This commit is contained in:
Lunik 2024-02-26 17:52:42 +01:00
parent 4067d4f1e5
commit cee20695eb
No known key found for this signature in database
GPG key ID: AEC152C8AC2C85EE
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
litellm/llms/ollama.py
View file

@ -147,6 +147,7 @@ def get_ollama_response(
stream = optional_params.pop("stream", False)
format = optional_params.pop("format", None)
images = optional_params.pop("images", None)
data = {
"model": model,
"prompt": prompt,
@ -155,6 +156,8 @@ def get_ollama_response(
}
if format is not None:
data["format"] = format
if images is not None:
data["images"] = images
## LOGGING
logging_obj.pre_call(