/stable-diffusion-webui

Stable Diffusion web UI

Primary LanguagePythonGNU Affero General Public License v3.0AGPL-3.0

Web-based UI for Stable Diffusion

Created by sd-webui

Visit sd-webui's Discord Server Discord Server

Installation instructions for:

Want to ask a question or request a feature?

Come to our Discord Server or use Discussions.

Documentation

Documentation is located here

Want to contribute?

Check the Contribution Guide

sd-webui is:

Project Features:

  • Two great Web UI's to choose from: Streamlit or Gradio

  • No more manually typing parameters, now all you have to do is write your prompt and adjust sliders

  • Built-in image enhancers and upscalers, including GFPGAN and realESRGAN

  • Run additional upscaling models on CPU to save VRAM

  • Textual inversion 🔥: info - requires enabling, see here, script works as usual without it enabled

  • Advanced img2img editor with Mask and crop capabilities

  • Mask painting 🖌️: Powerful tool for re-generating only specific parts of an image you want to change (currently Gradio only)

  • More diffusion samplers 🔥🔥: A great collection of samplers to use, including:

    • k_euler (Default)
    • k_lms
    • k_euler_a
    • k_dpm_2
    • k_dpm_2_a
    • k_heun
    • PLMS
    • DDIM

  • Loopback ➿: Automatically feed the last generated sample back into img2img

  • Prompt Weighting 🏋️: Adjust the strength of different terms in your prompt

  • Selectable GPU usage with --gpu <id>

  • Memory Monitoring 🔥: Shows VRAM usage and generation time after outputting

  • Word Seeds 🔥: Use words instead of seed numbers

  • CFG: Classifier free guidance scale, a feature for fine-tuning your output

  • Automatic Launcher: Activate conda and run Stable Diffusion with a single command

  • Lighter on VRAM: 512x512 Text2Image & Image2Image tested working on 4GB

  • Prompt validation: If your prompt is too long, you will get a warning in the text output field

  • Copy-paste generation parameters: A text output provides generation parameters in an easy to copy-paste form for easy sharing.

  • Correct seeds for batches: If you use a seed of 1000 to generate two batches of two images each, four generated images will have seeds: 1000, 1001, 1002, 1003.

  • Prompt matrix: Separate multiple prompts using the | character, and the system will produce an image for every combination of them.

  • Loopback for Image2Image: A checkbox for img2img allowing to automatically feed output image as input for the next batch. Equivalent to saving output image, and replacing input image with it.

Stable Diffusion Web UI

A fully-integrated and easy way to work with Stable Diffusion right from a browser window.

Streamlit

Features:

  • Clean UI with an easy to use design, with support for widescreen displays.
  • Dynamic live preview of your generations
  • Easily customizable presets right from the WebUI (Coming Soon!)
  • An integrated gallery to show the generations for a prompt or session (Coming soon!)
  • Better optimization VRAM usage optimization, less errors for bigger generations.
  • Text2Video - Generate video clips from text prompts right from the WEb UI (WIP)
  • Concepts Library - Run custom embeddings others have made via textual inversion.
  • Actively being developed with new features being added and planned - Stay Tuned!
  • Streamlit is now the new primary UI for the project moving forward.
  • Currently in active development and still missing some of the features present in the Gradio Interface.

Please see the Streamlit Documentation to learn more.

Gradio

Features:

  • Older UI design that is fully functional and feature complete.
  • Has access to all upscaling models, including LSDR.
  • Dynamic prompt entry automatically changes your generation settings based on --params in a prompt.
  • Includes quick and easy ways to send generations to Image2Image or the Image Lab for upscaling.
  • Note, the Gradio interface is no longer being actively developed and is only receiving bug fixes.

Please see the Gradio Documentation to learn more.

Image Upscalers

GFPGAN

Lets you improve faces in pictures using the GFPGAN model. There is a checkbox in every tab to use GFPGAN at 100%, and also a separate tab that just allows you to use GFPGAN on any picture, with a slider that controls how strong the effect is.

If you want to use GFPGAN to improve generated faces, you need to install it separately. Download GFPGANv1.3.pth and put it into the /stable-diffusion-webui/src/gfpgan/experiments/pretrained_models directory.

RealESRGAN

Lets you double the resolution of generated images. There is a checkbox in every tab to use RealESRGAN, and you can choose between the regular upscaler and the anime version. There is also a separate tab for using RealESRGAN on any picture.

Download RealESRGAN_x4plus.pth and RealESRGAN_x4plus_anime_6B.pth. Put them into the stable-diffusion-webui/src/realesrgan/experiments/pretrained_models directory.

GoBig, LSDR, and GoLatent (Currently Gradio Only)

More powerful upscalers that uses a seperate Latent Diffusion model to more cleanly upscale images.

Download LDSR project.yaml and model last.cpkt. Rename last.ckpt to model.ckpt and place both under stable-diffusion-webui/src/latent-diffusion/experiments/pretrained_models/

Please see the Image Enhancers Documentation to learn more.

Original Information From The Stable Diffusion Repo

Stable Diffusion

Stable Diffusion was made possible thanks to a collaboration with Stability AI and Runway and builds upon our previous work:

High-Resolution Image Synthesis with Latent Diffusion Models
Robin Rombach*, Andreas Blattmann*, Dominik Lorenz, Patrick Esser, Björn Ommer

CVPR '22 Oral

which is available on GitHub. PDF at arXiv. Please also visit our Project page.

Stable Diffusion is a latent text-to-image diffusion model. Thanks to a generous compute donation from Stability AI and support from LAION, we were able to train a Latent Diffusion Model on 512x512 images from a subset of the LAION-5B database. Similar to Google's Imagen, this model uses a frozen CLIP ViT-L/14 text encoder to condition the model on text prompts. With its 860M UNet and 123M text encoder, the model is relatively lightweight and runs on a GPU with at least 10GB VRAM. See this section below and the model card.

Stable Diffusion v1

Stable Diffusion v1 refers to a specific configuration of the model architecture that uses a downsampling-factor 8 autoencoder with an 860M UNet and CLIP ViT-L/14 text encoder for the diffusion model. The model was pretrained on 256x256 images and then finetuned on 512x512 images.

*Note: Stable Diffusion v1 is a general text-to-image diffusion model and therefore mirrors biases and (mis-)conceptions that are present in its training data. Details on the training procedure and data, as well as the intended use of the model can be found in the corresponding model card.

Comments

BibTeX

@misc{rombach2021highresolution,
      title={High-Resolution Image Synthesis with Latent Diffusion Models}, 
      author={Robin Rombach and Andreas Blattmann and Dominik Lorenz and Patrick Esser and Björn Ommer},
      year={2021},
      eprint={2112.10752},
      archivePrefix={arXiv},
      primaryClass={cs.CV}
}