Deprecate SD2

README.md

@@ -93,9 +93,6 @@ To run SVD or SV3D on a streamlit server:
   ![tile](assets/sv3d.gif)
 
 
-**November 30, 2023**
-- Following the launch of SDXL-Turbo, we are releasing [SD-Turbo](https://huggingface.co/stabilityai/sd-turbo).
-
 **November 28, 2023**
 - We are releasing SDXL-Turbo, a lightning fast text-to image model.
   Alongside the model, we release a [technical report](https://stability.ai/research/adversarial-diffusion-distillation)
@@ -257,8 +254,6 @@ The following models are currently supported:
   ```
 - [SDXL-base-0.9](https://huggingface.co/stabilityai/stable-diffusion-xl-base-0.9)
 - [SDXL-refiner-0.9](https://huggingface.co/stabilityai/stable-diffusion-xl-refiner-0.9)
-- [SD-2.1-512](https://huggingface.co/stabilityai/stable-diffusion-2-1-base/blob/main/v2-1_512-ema-pruned.safetensors)
-- [SD-2.1-768](https://huggingface.co/stabilityai/stable-diffusion-2-1/blob/main/v2-1_768-ema-pruned.safetensors)
 
 **Weights for SDXL**:
 

scripts/demo/sampling.py

@@ -52,24 +52,6 @@
         "config": "configs/inference/sd_xl_base.yaml",
         "ckpt": "checkpoints/sd_xl_base_0.9.safetensors",
     },
-    "SD-2.1": {
-        "H": 512,
-        "W": 512,
-        "C": 4,
-        "f": 8,
-        "is_legacy": True,
-        "config": "configs/inference/sd_2_1.yaml",
-        "ckpt": "checkpoints/v2-1_512-ema-pruned.safetensors",
-    },
-    "SD-2.1-768": {
-        "H": 768,
-        "W": 768,
-        "C": 4,
-        "f": 8,
-        "is_legacy": True,
-        "config": "configs/inference/sd_2_1_768.yaml",
-        "ckpt": "checkpoints/v2-1_768-ema-pruned.safetensors",
-    },
     "SDXL-refiner-0.9": {
         "H": 1024,
         "W": 1024,

scripts/demo/turbo.py

@@ -13,15 +13,6 @@
         "config": "configs/inference/sd_xl_base.yaml",
         "ckpt": "checkpoints/sd_xl_turbo_1.0.safetensors",
     },
-    "SD-Turbo": {
-        "H": 512,
-        "W": 512,
-        "C": 4,
-        "f": 8,
-        "is_legacy": False,
-        "config": "configs/inference/sd_2_1.yaml",
-        "ckpt": "checkpoints/sd_turbo.safetensors",
-    },
 }
 
 

sgm/inference/api.py

@@ -17,8 +17,6 @@
 
 
 class ModelArchitecture(str, Enum):
-    SD_2_1 = "stable-diffusion-v2-1"
-    SD_2_1_768 = "stable-diffusion-v2-1-768"
     SDXL_V0_9_BASE = "stable-diffusion-xl-v0-9-base"
     SDXL_V0_9_REFINER = "stable-diffusion-xl-v0-9-refiner"
     SDXL_V1_BASE = "stable-diffusion-xl-v1-base"
@@ -89,26 +87,6 @@ class SamplingSpec:
 
 
 model_specs = {
-    ModelArchitecture.SD_2_1: SamplingSpec(
-        height=512,
-        width=512,
-        channels=4,
-        factor=8,
-        is_legacy=True,
-        config="sd_2_1.yaml",
-        ckpt="v2-1_512-ema-pruned.safetensors",
-        is_guided=True,
-    ),
-    ModelArchitecture.SD_2_1_768: SamplingSpec(
-        height=768,
-        width=768,
-        channels=4,
-        factor=8,
-        is_legacy=True,
-        config="sd_2_1_768.yaml",
-        ckpt="v2-1_768-ema-pruned.safetensors",
-        is_guided=True,
-    ),
     ModelArchitecture.SDXL_V0_9_BASE: SamplingSpec(
         height=1024,
         width=1024,