From efe8b75200dcedd04cc47608571093d82373e5c6 Mon Sep 17 00:00:00 2001 From: ishaan-jaff Date: Sat, 16 Dec 2023 18:39:40 +0530 Subject: [PATCH] (fix) use litellm.vertex_vision_models --- litellm/llms/vertex_ai.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/litellm/llms/vertex_ai.py b/litellm/llms/vertex_ai.py index 16f66c0684e..d9e961f09c5 100644 --- a/litellm/llms/vertex_ai.py +++ b/litellm/llms/vertex_ai.py @@ -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"