diff --git a/litellm/llms/vertex_ai.py b/litellm/llms/vertex_ai.py
index 16f66c068..d9e961f09 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"