forked from phoenix/litellm-mirror
(fix) use litellm.vertex_vision_models
This commit is contained in:
parent
f3ebfb0517
commit
efe8b75200
1 changed files with 1 additions and 1 deletions
|
@ -205,7 +205,7 @@ def completion(
|
||||||
llm_model = GenerativeModel(model)
|
llm_model = GenerativeModel(model)
|
||||||
mode = ""
|
mode = ""
|
||||||
request_str += f"llm_model = GenerativeModel({model})\n"
|
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)
|
llm_model = GenerativeModel(model)
|
||||||
request_str += f"llm_model = GenerativeModel({model})\n"
|
request_str += f"llm_model = GenerativeModel({model})\n"
|
||||||
mode = "vision"
|
mode = "vision"
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue