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

fix test_aimage_generation_bedrock_with_optional_params

Browse source
This commit is contained in:
Ishaan Jaff 2024-11-12 16:58:07 -08:00
parent 501ced09ba
commit 84603763cc
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
tests/image_gen_tests/test_image_generation.py
View file

@ -181,6 +181,7 @@ def test_image_generation_azure_dall_e_3():
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_aimage_generation_bedrock_with_optional_params(): async def test_aimage_generation_bedrock_with_optional_params():
try: try:
litellm.in_memory_llm_clients_cache = {}
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-v1", model="bedrock/stability.stable-diffusion-xl-v1",