Pre-requisites
pip install google-cloud-aiplatform- Authentication:
- run
gcloud auth application-default loginSee Google Cloud Docs - Alternatively you can set
application_default_credentials.json
- run
model flag:
Required Environment Variables
Set the following environment variables (click here to learn how) to use these models.| Environment Variable | Description | Where to Find |
|---|---|---|
VERTEXAI_PROJECT | The Google Cloud project ID. | Google Cloud Console |
VERTEXAI_LOCATION | The location of your Vertex AI resources. | Google Cloud Console |
Supported Models
- gemini-pro
- gemini-pro-vision
- chat-bison-32k
- chat-bison
- chat-bison@001
- codechat-bison
- codechat-bison-32k
- codechat-bison@001

