diff --git a/litellm/utils.py b/litellm/utils.py
index efd48e8ab6..bb16c9f0a5 100644
--- a/litellm/utils.py
+++ b/litellm/utils.py
@@ -4930,6 +4930,7 @@ def get_llm_provider(
             or model in litellm.vertex_language_models
             or model in litellm.vertex_embedding_models
             or model in litellm.vertex_vision_models
+            or model in litellm.vertex_ai_image_models
         ):
             custom_llm_provider = "vertex_ai"
         ## ai21