From d3c67acbef906ccbd119633d2f8ebf0cfd3bf645 Mon Sep 17 00:00:00 2001
From: Jake Chai <61836864+jakechai@users.noreply.github.com>
Date: Wed, 23 Oct 2024 09:29:58 +0800
Subject: [PATCH] Add SD3.5.

---
 stability_api.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/stability_api.py b/stability_api.py
index ce727d1..a9a4c04 100644
--- a/stability_api.py
+++ b/stability_api.py
@@ -304,7 +304,7 @@ class StabilitySD3(StabilityBase):
     ACCEPT = "image/*"
     INPUT_SPEC = {
         "required": {
-            "model": (["sd3", "sd3-turbo"],),
+            "model": (["sd3.5-large", "sd3.5-large-turbo", "sd3-large", "sd3-large-turbo", "sd3-medium"],),
             "prompt": ("STRING", {"multiline": True}),
         },
         "optional": {