From 22452e8570a94e4065971d5b7a492d4d1a514091 Mon Sep 17 00:00:00 2001
From: Reinder Vos de Wael <reinder.vosdewael@childmind.org>
Date: Thu, 26 Dec 2024 16:07:43 -0500
Subject: [PATCH] feat: Add Stability 3.5 large (#13)

---
 .gitignore                          | 3 ++-
 proxy/app/routers/images/schemas.py | 2 +-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/.gitignore b/.gitignore
index 059c3c5..f4ec548 100644
--- a/.gitignore
+++ b/.gitignore
@@ -4,4 +4,5 @@
 .open-webui
 local
 *.pyc
-__pycache__
\ No newline at end of file
+__pycache__
+.idea
\ No newline at end of file
diff --git a/proxy/app/routers/images/schemas.py b/proxy/app/routers/images/schemas.py
index bd51a48..1c0698f 100644
--- 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):