From 00d1440d0dfbe03940c98b8b5562b495931e626f Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Tue, 30 Apr 2024 08:55:40 -0700 Subject: [PATCH] test(test_image_generation.py): change img model for test - bedrock EOL --- litellm/tests/test_image_generation.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/litellm/tests/test_image_generation.py b/litellm/tests/test_image_generation.py index 964b00582..31e7d3d90 100644 --- a/litellm/tests/test_image_generation.py +++ b/litellm/tests/test_image_generation.py @@ -136,7 +136,7 @@ def test_image_generation_bedrock(): litellm.set_verbose = True response = litellm.image_generation( 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", ) print(f"response: {response}") @@ -156,7 +156,7 @@ async def test_aimage_generation_bedrock_with_optional_params(): try: response = await litellm.aimage_generation( prompt="A cute baby sea otter", - model="bedrock/stability.stable-diffusion-xl-v0", + model="bedrock/stability.stable-diffusion-xl-v1", size="128x128", ) print(f"response: {response}")