forked from phoenix/litellm-mirror
fix test_aimage_generation_bedrock_with_optional_params
This commit is contained in:
parent
501ced09ba
commit
84603763cc
1 changed files with 1 additions and 0 deletions
|
@ -181,6 +181,7 @@ def test_image_generation_azure_dall_e_3():
|
|||
@pytest.mark.asyncio
|
||||
async def test_aimage_generation_bedrock_with_optional_params():
|
||||
try:
|
||||
litellm.in_memory_llm_clients_cache = {}
|
||||
response = await litellm.aimage_generation(
|
||||
prompt="A cute baby sea otter",
|
||||
model="bedrock/stability.stable-diffusion-xl-v1",
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue