fix vertex ai import

This commit is contained in:
Ishaan Jaff 2024-05-20 13:13:19 -07:00
parent 11c9780ff0
commit 2c25bfa8df

View file

@ -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 = {