Added additional context to README

README.md

@@ -20,7 +20,7 @@ models:
 
 ## Stable Diffusion WebUI + ControlNet
 
-Uses Stable Diffusion 2.1 models. Comes with several popular extensions to [AUTOMATIC1111's WebUI]([https://github.com/AUTOMATIC1111/stable-diffusion-webui]), including the [ControlNet WebUI extension](https://github.com/Mikubill/sd-webui-controlnet).
+Comes both with Stable Diffusion 2.1 models and Stable Diffusion 1.5 models. Comes with several popular extensions to [AUTOMATIC1111's WebUI]([https://github.com/AUTOMATIC1111/stable-diffusion-webui]), including the [ControlNet WebUI extension](https://github.com/Mikubill/sd-webui-controlnet). ControlNet models primarily works best with the SD 1.5 models at the time of writing.
 
 🐳 🤗 Builds a Docker image to be run as a [Hugging Face Space](https://huggingface.co/) using A10G or T4 hardware.
 
@@ -28,7 +28,7 @@ Uses Stable Diffusion 2.1 models. Comes with several popular extensions to [AUTO
 
 1. Duplicate this space to your Hugging Face account or clone this repo to your account.
 2. Under the *"Settings"* tab of your space you can choose which hardware for your space, that you will also be billed for.
-3. The [`on_start.sh`](./on_start.sh) file will be run when the container is started, right before the webui is initiated. This is where you can install any additional extensions or models you may need.
+3. The [`on_start.sh`](./on_start.sh) file will be run when the container is started, right before the WebUI is initiated. This is where you can install any additional extensions or models you may need.
 
 ---
 
@@ -48,12 +48,40 @@ Uses Stable Diffusion 2.1 models. Comes with several popular extensions to [AUTO
 * Trained models: [https://github.com/lllyasviel/ControlNet](https://github.com/lllyasviel/ControlNet)
 * Pre-extracted models: [https://huggingface.co/webui/ControlNet-modules-safetensors/tree/main](https://huggingface.co/webui/ControlNet-modules-safetensors/tree/main)
 
-#### Stable Diffusion Models License + ControlNet Models License
+#### Licenses for using Stable Diffusion models and ControlNet models
 
 * [https://huggingface.co/stabilityai/stable-diffusion-2/blob/main/LICENSE-MODEL](https://huggingface.co/stabilityai/stable-diffusion-2/blob/main/LICENSE-MODEL)
 * [https://huggingface.co/spaces/CompVis/stable-diffusion-license](https://huggingface.co/spaces/CompVis/stable-diffusion-license)
 * [https://github.com/lllyasviel/ControlNet/blob/main/LICENSE](https://github.com/lllyasviel/ControlNet/blob/main/LICENSE)
 
-#### Additional (optional) models
+### Enable additional models (checkpoints, LoRA, VAE, etc.)
 
-* [Dreamlike Diffusion 1.0](https://huggingface.co/dreamlike-art/dreamlike-diffusion-1.0) ([license](https://huggingface.co/dreamlike-art/dreamlike-diffusion-1.0/blob/main/LICENSE.md))
+Enable the models you want to use on the bottom of the [`on_start.sh`](./on_start.sh) file. This is also the place to add any additional models you may want to install when starting your space.
+
+```bash
+## Checkpoint · Example:
+download-model --checkpoint "FILENAME" "URL"
+
+## LORA (low-rank adaptation) · Example:
+download-model --lora "FILENAME" "URL"
+
+## VAE (variational autoencoder) · Example:
+download-model --vae "FILENAME" "URL"
+```
+
+#### Some examples of additional (optional) models
+
+Some models such as additional checkpoints, VAE, LoRA, etc. may already be present in the [`on_start.sh`](./on_start.sh) file. You can enable them by removing the `#` in front of their respective line or disable them by removing the line or adding a leading `#` before `download-model`.
+
+* [Checkpoint · Dreamlike Diffusion 1.0](https://huggingface.co/dreamlike-art/dreamlike-diffusion-1.0) ([license](https://huggingface.co/dreamlike-art/dreamlike-diffusion-1.0/blob/main/LICENSE.md))
+* [Checkpoint · Dreamshaper 3.31](https://huggingface.co/Lykon/DreamShaper)
+* [Checkpoint · The Ally's Mix III: Revolutions](https://civitai.com/models/10752/the-allys-mix-iii-revolutions)
+* [Checkpoint · Deliberate v2](https://civitai.com/models/4823/deliberate)
+* [Checkpoint · dalcefo_painting](https://civitai.com/models/5396/dalcefopainting)
+* [Checkpoint · RPG v4](https://huggingface.co/Anashel/rpg)
+* [Checkpoint · A to Zovya RPG Artist's Tools (1.5 & 2.1)](https://civitai.com/models/8124/a-to-zovya-rpg-artists-tools-15-and-21)
+* [LoRA · epi_noiseoffset v2](https://civitai.com/models/13941/epinoiseoffset)
+* [VAE · sd-vae-ft-mse-original](https://huggingface.co/stabilityai/sd-vae-ft-mse-original)
+* See [https://huggingface.co/models?filter=stable-diffusion](https://huggingface.co/models?filter=stable-diffusion) and [https://civitai.com/](https://civitai.com/) for more.
+
+Visit the individual model pages for more information on the models and their licenses.