From e46123612dfe5516401ca87cef35ca2d43872e53 Mon Sep 17 00:00:00 2001
From: Acly <aclysia@gmail.com>
Date: Wed, 12 Jun 2024 16:56:25 +0200
Subject: [PATCH] Model info API: support SD3

---
 README.md | 2 +-
 api.py    | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index ccde202..d687f7a 100644
--- a/README.md
+++ b/README.md
@@ -58,7 +58,7 @@ Lists available models with additional classification info.
     ```
     {
         "checkpoint_file.safetensors": {
-            "base_model": "sd15"|"sd20"|"sd21"|"sdxl"|"ssd1b"|"svd"|"cascade-b"|"cascade-c",
+            "base_model": "sd15"|"sd20"|"sd21"|"sd3"|"sdxl"|"ssd1b"|"svd"|"cascade-b"|"cascade-c",
             "is_inpaint": true|false,
             "is_refiner": true|false
         },
diff --git a/api.py b/api.py
index 865acb1..80606e1 100644
--- a/api.py
+++ b/api.py
@@ -21,6 +21,7 @@ model_names = {
     "SVD_img2vid": "svd",
     "Stable_Cascade_B": "cascade-b",
     "Stable_Cascade_C": "cascade-c",
+    "SD3": "sd3",
 }