SD 1.5! new default. ChangeLog. update MODEL_IDS (waifu too)

gadicc · gadicc · commit f99ccdf53955 · 2022-10-21T11:30:27.000+01:00
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -2,18 +2,24 @@
 
 * **2022-10-21**
 
-  * Upgrade to **Diffusers 0.6.0**
-  
-  * Biggest news of the above release is support for
+  * **Stable Diffusion 1.5 released!!!**
+
+    Accept the license at:
+    ["runwayml/stable-diffusion-v1-5"](https://huggingface.co/runwayml/stable-diffusion-v1-5)
+
+    It's the new default model.
+
+  * **Official Stable Diffusion inpainting model**
+
+    Accept the license at:
     ["runwayml/stable-diffusion-inpainting"](https://huggingface.co/runwayml/stable-diffusion-inpainting),
-    the first official stable diffusion checkpoint fine-tuned on **inpainting**.
 
-    You need to use the above link to *accept the license terms* on the
-    model card before you'll be able to download it.  Also, unfortunately,
-    this means we're back to having a separate container for inpainting,
-    as it's a different model.
+    Unfortunately, as it's a different model, it means we're back to having a
+    separate container for inpainting.
+
+    NB: `model_input` takes `image` now, and not `init_image` like the legacy model.
 
-    Note also: you need to send the image as `image` now, and not `init_image`.
+  * Upgrade to **Diffusers v0.7.0.dev0**
 
   * **Flash attention** now disabled by default.  1) Because it's built on
     an older version of diffusers, but also because 2) I didn't succeed in
diff --git a/Dockerfile b/Dockerfile
@@ -35,8 +35,9 @@ ARG HF_AUTH_TOKEN
 ENV HF_AUTH_TOKEN=${HF_AUTH_TOKEN}
 
 # "CompVis/stable-diffusion-v1-4", "hakurei/waifu-diffusion",
-# "runwayml/stable-diffusion-inpainting", etc. (huggingface)
-ARG MODEL_ID="CompVis/stable-diffusion-v1-4"
+# "runwayml/stable-diffusion-inpainting",
+# "runwayml/stable-diffusion-v1-5" etc. (huggingface)
+ARG MODEL_ID="runwayml/stable-diffusion-v1-5"
 ENV MODEL_ID=${MODEL_ID}
 
 # If set, it will be downloaded and converted to diffusers format, and
diff --git a/loadModel.py b/loadModel.py
@@ -3,7 +3,13 @@
 from diffusers import StableDiffusionPipeline
 
 HF_AUTH_TOKEN = os.getenv("HF_AUTH_TOKEN")
-MODEL_IDS = ["CompVis/stable-diffusion-v1-4", "hakurei/waifu-diffusion"]
+MODEL_IDS = [
+    "CompVis/stable-diffusion-v1-4",
+    "hakurei/waifu-diffusion",
+    "hakurei/waifu-diffusion-v1-3",
+    "runwayml/stable-diffusion-inpainting",
+    "runwayml/stable-diffusion-v1-5",
+]
 
 
 def loadModel(model_id: str, load=True):
