fix vertex ai import

2024-05-20 13:13:19 -07:00 · 2024-05-20 13:13:19 -07:00 · 2c25bfa8df
commit 2c25bfa8df
parent 11c9780ff0
1 changed files with 0 additions and 3 deletions
--- a/litellm/llms/vertex_httpx.py
+++ b/litellm/llms/vertex_httpx.py
@ -157,9 +157,6 @@ class VertexLLM(BaseLLM):
        } \
        "https://us-central1-aiplatform.googleapis.com/v1/projects/PROJECT_ID/locations/us-central1/publishers/google/models/imagegeneration:predict"
        """
-
-        import vertexai
-
        auth_header = self._ensure_access_token()

        request_data = {