Model info API: support SD3

2024-06-12 16:56:25 +02:00
parent 6e7b2445db
commit e46123612d
2 changed files with 2 additions and 1 deletions
@@ -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
        },
@@ -21,6 +21,7 @@ model_names = {
    "SVD_img2vid": "svd",
    "Stable_Cascade_B": "cascade-b",
    "Stable_Cascade_C": "cascade-c",
+    "SD3": "sd3",
 }