diff --git a/litellm/utils.py b/litellm/utils.py index 4b80d203b..7302bea75 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -2406,7 +2406,7 @@ def get_optional_params( elif k == "hf_model_name" and custom_llm_provider != "sagemaker": continue elif ( - k.startswith("vertex_") and custom_llm_provider != "vertex_ai" + k.startswith("vertex_") and custom_llm_provider != "vertex_ai" and custom_llm_provider != "vertex_ai_beta" ): # allow dynamically setting vertex ai init logic continue passed_params[k] = v