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", }