From 49c31ecee666e099951560acb41896f92100e3f0 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Tue, 12 Nov 2024 20:53:17 -0800 Subject: [PATCH] add stability.stable-image-ultra-v1:0 --- .../bedrock/image/amazon_stability3_transformation.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/litellm/llms/bedrock/image/amazon_stability3_transformation.py b/litellm/llms/bedrock/image/amazon_stability3_transformation.py index 784e86b04..2c90b3a12 100644 --- a/litellm/llms/bedrock/image/amazon_stability3_transformation.py +++ b/litellm/llms/bedrock/image/amazon_stability3_transformation.py @@ -53,9 +53,15 @@ class AmazonStability3Config: sd3-medium sd3.5-large sd3.5-large-turbo + + Stability ultra models + stable-image-ultra-v1 """ - if model and ("sd3" in model or "sd3.5" in model): - return True + if model: + if "sd3" in model or "sd3.5" in model: + return True + if "stable-image-ultra-v1" in model: + return True return False @classmethod