A simple way to install and use [Stable Diffusion](https://replicate.com/stability-ai/stable-diffusion) on your local computer. Provides a browser UI for generating images from text prompts. Just enter your text prompt, and see the generated image. 🎉 **New!** `img2img` is now supported! You can provide an image to generate new images based on it (and an optional text prompt). You can also use the generated image as the new input image in 1-click, to refine it further. # What does this do? Two things: 1. Automatically downloads and installs Stable Diffusion on your local computer (no need to mess with conda or environments) 2. Gives you a simple browser-based UI to talk to your local Stable Diffusion. Enter text prompts and view the generated image. No API keys required. All the processing will happen on your local computer, it does not transmit your prompts or process on any remote server. Screenshot of tool # System Requirements 1. Computer capable of running Stable Diffusion. 2. Linux or Windows 11 (with [WSL](https://docs.microsoft.com/en-us/windows/wsl/install)) or Windows 10 v2004+ (Build 19041+) with [WSL](https://docs.microsoft.com/en-us/windows/wsl/install). 3. Requires (a) [Docker](https://docs.docker.com/engine/install/), (b) [docker-compose v1.29](https://docs.docker.com/compose/install/), and (c) [nvidia-container-toolkit](https://stackoverflow.com/a/58432877). **Important:** If you're using Windows, please install docker inside your [WSL](https://docs.microsoft.com/en-us/windows/wsl/install)'s Linux. Install docker for the Linux distro in your WSL. **Don't install Docker for Windows.** # Installation 1. Clone this repository: `git clone https://github.com/cmdr2/stable-diffusion-ui.git` or [download the zip file](https://github.com/cmdr2/stable-diffusion-ui/archive/refs/heads/main.zip) and unzip. 2. Open your terminal, and in the project directory run: `docker-compose up &` (warning: this will take some time during the first run, since it'll download Stable Diffusion's [docker image](https://replicate.com/stability-ai/stable-diffusion), nearly 17 GiB) 3. Open http://localhost:8000 in your browser. That's it! If you're getting errors, please check the [Troubleshooting](#troubleshooting) section below. To stop the server, please run `docker-compose down` # Usage Open http://localhost:8000 in your browser (after running `docker-compose up &` from step 2 previously). ## With a text description 1. Enter a text prompt, like `a photograph of an astronaut riding a horse` in the textbox. 2. Press `Make Image`. This will take some time, depending on your system's processing power. 3. See the image generated using your prompt. ## With an image 1. Click `Browse..` next to `Initial Image`. Select your desired image. 2. An optional text prompt can help you further describe the kind of image you want to generate. 3. Press `Make Image`. See the image generated using your prompt. **Pro tip:** You can also click `Use as Input` on a generated image, to use it as the input image for your next generation. This can be useful for sequentially refining the generated image with a single click. **Another tip:** Images with the same aspect ratio of your generated image work best. E.g. 1:1 if you're generating images sized 512x512. ## Problems? Please [file an issue](https://github.com/cmdr2/stable-diffusion-ui/issues) if this did not work for you (after trying the common [troubleshooting](#troubleshooting) steps)! # Advanced Settings You can also set the configuration like `seed`, `width`, `height`, `num_outputs`, `num_inference_steps` and `guidance_scale` using the 'show' button next to 'Advanced settings'. Use the same `seed` number to get the same image for a certain prompt. This is useful for refining a prompt without losing the basic image design. Enable the `random images` checkbox to get random images. ![Screenshot of advanced settings](media/config-v2.jpg?raw=true) # Troubleshooting ## './docker-compose.yml' is invalid: > ERROR: The Compose file './docker-compose.yml' is invalid because: > services.stability-ai.deploy.resources.reservations value Additional properties are not allowed ('devices' was unexpected) Please ensure you have `docker-compose` version 1.29 or higher. Check `docker-compose --version`, and if required [update it to 1.29](https://docs.docker.com/compose/install/). (Thanks [HVRyan](https://github.com/HVRyan)) ## RuntimeError: Found no NVIDIA driver on your system: If you have an NVIDIA GPU and the latest [NVIDIA driver](http://www.nvidia.com/Download/index.aspx), please ensure that you've installed [nvidia-container-toolkit](https://stackoverflow.com/a/58432877). (Thanks [u/exintrovert420](https://www.reddit.com/user/exintrovert420/)) ## Some other process is already running at port 8000 / port 8000 could not be bound You can override the port used. Please change `docker-compose.yml` inside the project directory, and update the line `8000:8000` to `1337:8000` (or where 1337 is whichever port number you want). After doing this, please restart your server, by running `docker-compose down` and then `docker-compose up &`. After this, you can access the server at `http://localhost:1337` (where 1337 is the new port you specified earlier). # Behind the scenes This project is a quick way to get started with Stable Diffusion. You do not need to have Stable Diffusion already installed, and do not need any API keys. This project will automatically download Stable Diffusion's docker image, the first time it is run. This project runs Stable Diffusion in a docker container behind the scenes, using Stable Diffusion's [Docker image](https://replicate.com/stability-ai/stable-diffusion) on replicate.com. # Bugs reports and code contributions welcome If there are any problems or suggestions, please feel free to [file an issue](https://github.com/cmdr2/stable-diffusion-ui/issues). Also, please feel free to submit a pull request, if you have any code contributions in mind. # Disclaimer The authors of this project are not responsible for any content generated using this interface. This license of this software forbids you from sharing any content that violates any laws, produce any harm to a person, disseminate any personal information that would be meant for harm, spread misinformation and target vulnerable groups. For the full list of restrictions please read [the license](LICENSE).