feat: Add Stability 3.5 large (#13)

childmindresearch · Dec 26, 2024 · 22452e8 · 22452e8
1 parent bedf47d
commit 22452e8
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/.gitignore b/.gitignore
@@ -4,4 +4,5 @@
 .open-webui
 local
 *.pyc
-__pycache__
+__pycache__
+.idea
diff --git a/proxy/app/routers/images/schemas.py b/proxy/app/routers/images/schemas.py
@@ -5,7 +5,7 @@
 
 import pydantic
 
-IMAGE_GENERATION_MODELS = Literal["aws/stability.sd3-large-v1:0"]
+IMAGE_GENERATION_MODELS = Literal["aws/stability.sd3-large-v1:0", "aws/stability.sd3-5-large-v1:0"]
 
 
 class PostImageGenerationRequest(pydantic.BaseModel):