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

test(test_image_generation.py): fix test

Browse source
This commit is contained in:
Krrish Dholakia 2024-04-30 12:14:29 -07:00
parent 398d503590
commit 285a3733a9
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

@ -137,7 +137,7 @@ def test_image_generation_bedrock():
response = litellm.image_generation(
prompt="A cute baby sea otter",
model="bedrock/stability.stable-diffusion-xl-v1",
aws_region_name="us-east-1",
aws_region_name="us-west-2",
)
print(f"response: {response}")
except litellm.RateLimitError as e:
@ -157,7 +157,7 @@ async def test_aimage_generation_bedrock_with_optional_params():
response = await litellm.aimage_generation(
prompt="A cute baby sea otter",
model="bedrock/stability.stable-diffusion-xl-v1",
size="128x128",
size="256x256",
)
print(f"response: {response}")
except litellm.RateLimitError as e: