fix vertex ai import

This commit is contained in:
Ishaan Jaff
2024-05-20 13:13:19 -07:00
parent 11c9780ff0
commit 2c25bfa8df
-3
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 = {