From 9fcab392a4a536c7d433b8c929057ac0e9ecf719 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 efd48e8ab63..bb16c9f0a55 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