phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

docs - image generation vertex

Browse source
This commit is contained in:
Ishaan Jaff 2024-05-20 12:18:31 -07:00
parent 571d4cf569
commit aa0ed8238b
2 changed files with 28 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

16
docs/my-website/docs/image_generation.md
View file

@ -151,3 +151,19 @@ response = image_generation(
)
print(f"response: {response}")
```
## VertexAI - Image Generation Models
### Usage
Use this for image generation models on VertexAI
```python
response = litellm.image_generation(
prompt="An olympic size swimming pool",
model="vertex_ai/imagegeneration@006",
vertex_ai_project="adroit-crow-413218",
vertex_ai_location="us-central1",
)
print(f"response: {response}")
```

12
docs/my-website/docs/providers/vertex.md
View file

@ -508,6 +508,18 @@ All models listed [here](https://github.com/BerriAI/litellm/blob/57f37f743886a02
| text-embedding-preview-0409 | `embedding(model="vertex_ai/text-embedding-preview-0409", input)` |
| text-multilingual-embedding-preview-0409 | `embedding(model="vertex_ai/text-multilingual-embedding-preview-0409", input)` |
## Image Generation Models
Usage
```python
response = await litellm.aimage_generation(
prompt="An olympic size swimming pool",
model="vertex_ai/imagegeneration@006",
vertex_ai_project="adroit-crow-413218",
vertex_ai_location="us-central1",
)
```
## Extra