From 9c5000a01c9eb3cc7deb507daf83992548aed122 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 2 Sep 2024 17:49:36 -0700 Subject: [PATCH] fix get llm provider for imagen --- litellm/utils.py | 1 + 1 file changed, 1 insertion(+) 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