phoenix/litellm-mirror
0
0
Fork 1
mirror of https://github.com/BerriAI/litellm.git synced 2025-04-26 11:14:04 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix get llm provider for imagen

Browse source
This commit is contained in:
Ishaan Jaff 2024-09-02 17:49:36 -07:00
parent ce4fb10b04
commit 9c5000a01c
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
litellm/utils.py
View file

@ -4930,6 +4930,7 @@ def get_llm_provider(
or model in litellm.vertex_language_models or model in litellm.vertex_language_models
or model in litellm.vertex_embedding_models or model in litellm.vertex_embedding_models
or model in litellm.vertex_vision_models or model in litellm.vertex_vision_models
or model in litellm.vertex_ai_image_models
): ):
custom_llm_provider = "vertex_ai" custom_llm_provider = "vertex_ai"
## ai21 ## ai21