aboutsummaryrefslogtreecommitdiffstats
path: root/modules/sd_models_types.py
diff options
context:
space:
mode:
authorRitesh Gangnani <riteshgangnani10>2023-11-05 19:13:49 +0530
committerRitesh Gangnani <riteshgangnani10>2023-11-05 19:13:49 +0530
commitff1609f91ea0e9a90ba7b6ecc6d794c39c1f8c8f (patch)
treefffc80140694c6a15e78366dcc8008ed4a6f0bc7 /modules/sd_models_types.py
parent4afaaf8a020c1df457bcf7250cb1c7f609699fa7 (diff)
downloadstable-diffusion-webui-gfx803-ff1609f91ea0e9a90ba7b6ecc6d794c39c1f8c8f.tar.gz
Add SSD-1B as a supported model
Diffstat (limited to 'modules/sd_models_types.py')
-rw-r--r--modules/sd_models_types.py5
1 files changed, 4 insertions, 1 deletions
diff --git a/modules/sd_models_types.py b/modules/sd_models_types.py
index 5ffd2f4f..1f28942a 100644
--- a/modules/sd_models_types.py
+++ b/modules/sd_models_types.py
@@ -22,7 +22,10 @@ class WebuiSdModel(LatentDiffusion):
"""structure with additional information about the file with model's weights"""
is_sdxl: bool
- """True if the model's architecture is SDXL"""
+ """True if the model's architecture is SDXL or SSD"""
+
+ is_ssd: bool
+ """True if the model is SSD"""
is_sd2: bool
"""True if the model's architecture is SD 2.x"""