tiennm99/litellm
1
0
Fork 0
mirror of https://github.com/tiennm99/litellm.git synced 2026-08-08 22:25:25 +00:00
Code Issues Packages Projects Releases Wiki Activity

test(test_image_generation.py): fix test

Browse Source
This commit is contained in:
Krrish Dholakia
2024-04-30 12:15:37 -07:00
parent 90cdfef1c1
commit fee488bd53
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
litellm/tests/test_image_generation.py
+2 -2
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: