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

add stability.stable-image-ultra-v1:0

Browse source
This commit is contained in:
Ishaan Jaff 2024-11-12 20:53:17 -08:00
parent 70c8be59d7
commit 49c31ecee6
1 changed files with 8 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

10
litellm/llms/bedrock/image/amazon_stability3_transformation.py
View file

@ -53,9 +53,15 @@ class AmazonStability3Config:
sd3-medium sd3-medium
sd3.5-large sd3.5-large
sd3.5-large-turbo sd3.5-large-turbo
Stability ultra models
stable-image-ultra-v1
""" """
if model and ("sd3" in model or "sd3.5" in model): if model:
return True if "sd3" in model or "sd3.5" in model:
return True
if "stable-image-ultra-v1" in model:
return True
return False return False
@classmethod @classmethod