diff --git a/litellm/tests/test_amazing_vertex_completion.py b/litellm/tests/test_amazing_vertex_completion.py index ac18a65f287..4e95425ed4f 100644 --- a/litellm/tests/test_amazing_vertex_completion.py +++ b/litellm/tests/test_amazing_vertex_completion.py @@ -50,6 +50,8 @@ VERTEX_MODELS_TO_NOT_TEST = [ "text-bison@001", "gemini-1.5-pro", "gemini-1.5-pro-preview-0215", + "gemini-pro-flash", + "gemini-pro-experimental", ]