Remove runwayml/stable-diffusion-v1-5 😢

README.md

@@ -15,19 +15,22 @@ header: mini
 license: apache-2.0
 models:
 - ai-forever/Real-ESRGAN
+- Comfy-Org/stable-diffusion-v1-5-archive
 - cyberdelia/CyberRealistic
 - fluently/Fluently-v4
 - h94/IP-Adapter
 - Linaqruf/anything-v3-1
 - Lykon/dreamshaper-8
 - prompthero/openjourney-v4
-- runwayml/stable-diffusion-v1-5
 - SG161222/Realistic_Vision_V5.1_noVAE
 - XpucT/Deliberate
 preload_from_hub:
 - >-
   ai-forever/Real-ESRGAN
   RealESRGAN_x2.pth,RealESRGAN_x4.pth
+- >-
+  Comfy-Org/stable-diffusion-v1-5-archive
+  v1-5-pruned-emaonly-fp16.safetensors
 - >-
   cyberdelia/CyberRealistic
   CyberRealistic_V5_FP16.safetensors
@@ -42,13 +45,10 @@ preload_from_hub:
   anything-v3-2.safetensors
 - >-
   Lykon/dreamshaper-8
-  text_encoder/model.fp16.safetensors,unet/diffusion_pytorch_model.fp16.safetensors,vae/diffusion_pytorch_model.fp16.safetensors
+  text_encoder/model.fp16.safetensors,unet/diffusion_pytorch_model.fp16.safetensors,vae/diffusion_pytorch_model.fp16.safetensors,model_index.json
 - >-
   prompthero/openjourney-v4
   openjourney-v4.ckpt
-- >-
-  runwayml/stable-diffusion-v1-5
-  text_encoder/model.fp16.safetensors,unet/diffusion_pytorch_model.fp16.safetensors,vae/diffusion_pytorch_model.fp16.safetensors
 - >-
   SG161222/Realistic_Vision_V5.1_noVAE
   Realistic_Vision_V5.1_fp16-no-ema.safetensors

lib/config.py

@@ -26,17 +26,18 @@ Config = SimpleNamespace(
     },
     MODEL="Lykon/dreamshaper-8",
     MODELS=[
+        "Comfy-Org/stable-diffusion-v1-5-archive",
         "cyberdelia/CyberRealistic",
         "fluently/Fluently-v4",
         "Linaqruf/anything-v3-1",
         "Lykon/dreamshaper-8",
         "prompthero/openjourney-v4",
-        "runwayml/stable-diffusion-v1-5",
         "SG161222/Realistic_Vision_V5.1_noVAE",
         "XpucT/Deliberate",
     ],
     MODEL_CHECKPOINTS={
         # keep keys lowercase
+        "comfy-org/stable-diffusion-v1-5-archive": "v1-5-pruned-emaonly-fp16.safetensors",
         "cyberdelia/cyberrealistic": "CyberRealistic_V5_FP16.safetensors",
         "fluently/fluently-v4": "Fluently-v4.safetensors",
         "linaqruf/anything-v3-1": "anything-v3-2.safetensors",

lib/inference.py

@@ -107,12 +107,12 @@ def generate(
     embeddings=[],
     style=None,
     seed=None,
-    model="runwayml/stable-diffusion-v1-5",
-    scheduler="PNDM",
+    model="Lykon/dreamshaper-8",
+    scheduler="DEIS 2M",
     width=512,
     height=512,
     guidance_scale=7.5,
-    inference_steps=50,
+    inference_steps=40,
     denoising_strength=0.8,
     deepcache=1,
     scale=1,

requirements.txt

@@ -3,11 +3,10 @@ accelerate
 einops==0.8.0
 compel==2.0.3
 deepcache==0.1.1
-diffusers==0.29.2
+diffusers==0.30.2
 hf-transfer
 gradio==4.41.0
 ruff==0.5.7
-scipy  # for LMS scheduler
 spaces
 torch==2.2.0
 torchvision==0.17.0

usage.md

@@ -30,12 +30,12 @@ Rescale up to 4x using [Real-ESRGAN](https://github.com/xinntao/Real-ESRGAN) fro
 
 Each model checkpoint has a different aesthetic:
 
+* [Comfy-Org/stable-diffusion-v1-5](https://huggingface.co/Comfy-Org/stable-diffusion-v1-5-archive): base
 * [cyberdelia/CyberRealistic_v5](https://huggingface.co/cyberdelia/CyberRealistic): photorealistic
 * [Lykon/dreamshaper-8](https://huggingface.co/Lykon/dreamshaper-8): general purpose (default)
 * [fluently/Fluently-v4](https://huggingface.co/fluently/Fluently-v4): general purpose
 * [Linaqruf/anything-v3-1](https://huggingface.co/Linaqruf/anything-v3-1): anime
 * [prompthero/openjourney-v4](https://huggingface.co/prompthero/openjourney-v4): Midjourney-like
-* [runwayml/stable-diffusion-v1-5](https://huggingface.co/runwayml/stable-diffusion-v1-5): base
 * [SG161222/Realistic_Vision_v5.1](https://huggingface.co/SG161222/Realistic_Vision_V5.1_noVAE): photorealistic
 * [XpucT/Deliberate_v6](https://huggingface.co/XpucT/Deliberate): general purpose