ai-workspace-services/litellm
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

(fix) use litellm.vertex_vision_models

Browse Source
This commit is contained in:
ishaan-jaff 2023-12-16 18:39:40 +05:30
parent f3ebfb0517
commit efe8b75200
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
litellm/llms/vertex_ai.py
View File

@ -205,7 +205,7 @@ def completion(
llm_model = GenerativeModel(model)
mode = ""
request_str += f"llm_model = GenerativeModel({model})\n"
elif model in ["gemini-pro-vision"]:
elif model in litellm.vertex_vision_models:
llm_model = GenerativeModel(model)
request_str += f"llm_model = GenerativeModel({model})\n"
mode = "vision"