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

test(test_image_generation.py): change img model for test - bedrock EOL

Browse source
This commit is contained in:
Krrish Dholakia 2024-04-30 08:55:40 -07:00
parent d717fa2588
commit 00d1440d0d
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
litellm/tests/test_image_generation.py
View file

@ -136,7 +136,7 @@ def test_image_generation_bedrock():
litellm.set_verbose = True litellm.set_verbose = True
response = litellm.image_generation( response = litellm.image_generation(
prompt="A cute baby sea otter", prompt="A cute baby sea otter",
model="bedrock/stability.stable-diffusion-xl-v0", model="bedrock/stability.stable-diffusion-xl-v1",
aws_region_name="us-east-1", aws_region_name="us-east-1",
) )
print(f"response: {response}") print(f"response: {response}")
@ -156,7 +156,7 @@ async def test_aimage_generation_bedrock_with_optional_params():
try: try:
response = await litellm.aimage_generation( response = await litellm.aimage_generation(
prompt="A cute baby sea otter", prompt="A cute baby sea otter",
model="bedrock/stability.stable-diffusion-xl-v0", model="bedrock/stability.stable-diffusion-xl-v1",
size="128x128", size="128x128",
) )
print(f"response: {response}") print(f"response: {response}")