This repository contains instructions for building a docker image capable of running Stable Diffusion on machines equipped with an AMD Radeon RX6000 or RX7000 graphics card. It has been tested on a RX6800 but should work on most cards in either family.
To install Stable Diffusion you will need 🐳 Docker installed. You may also need a HuggingFace account depending on where you're sourcing your ckpt and yaml files.
-
Initialize the stable-diffusion-webui submodule inside this repository.
$ git submodule update --init
-
Download the Stable Diffusion 2 checkpoint file.
The Stable Diffusion 2 checkpoint file can be found in StabilityAI's repository on 🤗 Hugging Face. Download the ckpt file from the 📄 Files and versions tab. Place the file in
Project root/stable-diffusion-webui/models/Stable-diffusion
👉 If you already have a checkpoint file for stable diffusion or one of its derivatives, you may skip this step.
-
Download the yaml configuration file for your checkpoint.
This step is only necessary if you're using the Stable Diffusion 2 checkpoint file linked above. Select and download the correct file from stable-diffusion-webui's dependency guide.
Place this file alongside your ckpt file in
Project root/stable-diffusion-webui/models/Stable-diffusion
. It should have the same name as the checkpoint, so if you're using the checkpoint768-v-ema.ckpt
that means the config file should be called768-v-ema.yaml
– you get the idea. -
Install dependencies and build the image
Once the model files are in place, you should be able to build the docker image by running
$ docker build . -t Choose a name
All relevant dependencies should be installed during the build process, and you will not have to install anything more once it is done.
⚠️ This step will take several minutes and requires 40-50 gigabytes of storage.
👉 Images aren't automatically deleted if the build process fails, so remember to delete your old images if you need to cancel the build process, or if it errors.
-
Start the Stable Diffusion image
Now that you have build the image, you can start it by running
$ ./start.sh The name you chose
-
Open the GUI in your browser
Go to http://localhost:7860 and start generating!
For tips on how to generate good images, read through the Tips & Tricks guide.
⚠️ Stable diffusion sometimes doesn't give you any indication of exactly when the web server is started. Reload the page until it's all up and running, shouldn't take more than a minute or two.
⚠️ The first generation will often take a while, subsequent generations will be considerably quicker.