awesome-stable-diffusion/README.md
Georg Zoeller b96b98a5f4
Update README.md
Signed-off-by: Georg Zoeller <georgzoeller@users.noreply.github.com>
2022-09-25 18:31:42 +08:00

16 KiB

Awesome Stable-Diffusion

Awesome

This is a list of software and resources for the Stable Diffusion AI model.

  • 🖊️ marks content that requires sign-up or account creation for a third party service outside GitHub.
  • 💵 marks Non-Free content: commercial content that may require any kind of payment.

Due to the fast-moving nature of the topic, entries in the list may be removed at an expedited rate until the ecosystem matures.

See Contributing.

Official Resources

  • CompVis/Stable Diffusion - The official release of Stable Diffusion including a CLI, an AI-based Safety Classifier, which detects and suppresses sexualized content, and all the necessary files to get running.
  • stability-AI/stability-sdk - The official SDK used to build python applications integrated with StabilityAI's cloud platform instead of hosting the model locally. Operation requires an API Key (🖊️💵).
  • Public Release Announcement - StabilityAI's announcement about the public release of Stable Diffusion.
  • 🖊️ Official Discord - The official Stable Diffusion Discord by StabilityAI.
  • laion-aesthetic - The dataset used train stable diffusion, useful for querying to see if a concept is represented.

Actively Maintained Forks and Containers

All forks listed here add additional features and optimisations and are generally faster than the original release, as they keep the model in memory rather than reloading it after every prompt. Most forks seem to remove the Safety Classifier which may present a risk if used to provide public-facing services, such as Discord bots.

  • basujindal/stable-diffusion - "Optimized Stable Diffusion"—a fork with dramatically reduced VRAM requirements through model splitting, enabling Stable Diffusion on lower-end graphics cards; includes a GradIO web interface and support for weighted prompts.

  • bes-dev/stable_diffusion.openvino - A fork for running the model using a CPU compatible with OpenVINO.

  • sd-webui/stable-diffusion-webui - Very active fork with optional, highly featureful Gradio UI and support for txt2img, img2img inpainting, GFPGAN, ESRGAN, weighted prompts, optimized low memory version, optional textual-inversion and more.

  • invoke-ai/InvokeAI - (formerly known as lstein/stable-diffusion) - Very active fork adding a conversational CLI, basic web interface and support for GFPGAN, ESRGAN, weighted prompts, img2img, tiling, textual-inversion as well as inference on Apple M1.

  • lowfuel/progrock-stable - Fork with optional Web GUI and a different approach to upscaling (GoBIG/ESRGAN)

    • txt2imghd - Fork of progrock diffusion that creates detailed, higher-resolution images by first generating an image from a prompt, upscaling it, then running img2img on smaller pieces of the upscaled image, and blending the results back into the original image.
  • neonsecret/stable-diffusion - Fork focusing on bigger resolutions with less vram at the expense of speed, automatically adjusting to the GPUs abilities. Also includes upscaling, facial restoration via CodeFormer and custom UI

  • replicate/copg-stable-diffusion - Cog machine learning container of SD v1.4.

  • stable-diffusion-jupyterlab-docker - A Docker setup ready to go with Jupyter notebooks for Stable Diffusion.

Models and Weights

Models (.ckpt files) must be separately downloaded and are required to run Stable Diffusion. The latest model release is v1.4.

  • 🖊️ Official Model Card - Official Model Card on Hugging Face with all versions of the model. Download requires sign-in and acceptance of terms of service.
  • stable-diffusion-v-1-4-original.chkpt - The latest model's card
  • RealESRGAN Models - Download location for the latest RealESRGAN models required to use the upscaling features implemented by many forks. Different models exist for realistic and anime content. Please refer to the fork documentation to identify the ones you need.

Online Demos and Notebooks

Complementary Models and Tools

Tools and models for use in conjuction with Stable Diffusion

Img2Img

  • Prompt to Prompt - Unofficial Implementation of Cross-attention-control for prompt to prompt image editing.

Customisation

GUIS

Most of these GUIS, unless mentioned otherwise in their documentation, include stable-diffusion.

  • AUTOMATIC1111/stable-diffusion-webui - Gradio based UI with extensive features such as in and outpainting, previews, xy plots, upscaling and more. Comes with a handy install script that takes care of most dependencies and addons.
  • 🖊️💵 Auto SD Workflow - A UI for lstein/stable-diffusion's dream.py with optimized UX for large-scale/production workflow around image synthesis. Video Walkthrough.
  • cmdr2/stable-diffusion-ui - Another, simple to use UI for windows and Linux
  • 🖊️ DiffusionUI - web UI made with Vue.js inspired by Dall-e using diffusers, perfect for inpainting. Video demo
  • 💵 NMKD GUI - Windows UI, fully featured. Closed source. Pick your own price.
  • 🖊️ SD-MUI - mobile-first PWA made with React + MaterialUI. Run free locally or use free & paid credits on the live site. (Source Code) MIT License TypeScript
  • Stable Diffusion GRisk GUI - Windows GUI binary for SD. Closed source so use at your own risk.
  • Stable Diffusion Infinity - A proof of concept for outpainting with an infinite canvas interface. (requires powerful GPU).
  • Unstable Fusion - A Stable Diffusion desktop frontend with inpainting, img2img and more

Upscaling

  • BasicSR - Open-source upscaling and restoration toolbox supporting several models.
  • BSRGAN - BSRGAN—another upscaling solution specialized in upscaling degraded images.
  • Cupscale - GUI for running various upscaling models, including ESRGAN and RealESRGAN.
  • RealESRGAN - ESRGAN Upscaling (2x, 4x) and content restoration. Python: pip install realesrgan
  • jquesnelle/txt2imghd - A port of the GOBIG mode from progrockdiffusion, providing high quality upscaling on top of txt2img.

Content Restoration

  • lama-cleaner - Content aware AI inpainting tool useful for removing unwanted objects or defects from images. Python: pip install lama-cleaner
  • GFPGAN - Face Restoration GAN included in several forks for automatically fixing the face deformation commonly found in SD output.
  • CodeFormer - Another Face Restoration model (Paper).

Task Chaining

  • chaiNNer - Graphical node-based editor for chaining image processing tasks.
  • ai-art-generator - AI art generation suite combining Stable Diffusion and other models for high volume art generation.

Prompt Building

Prompts are the instructions given to diffusion models to manipulate their output.

  • ai-art.com/modifiers - A visual reference guide for keywords.
  • krea.ai - Prompt search engine that also recommends similar prompts to the one that you click on.
    • 🖊️ With account creation, you can like and save prompts in your own collections.
  • Lexica.art - A searchable, visual database of images and the prompts settings used to create them.
  • pharmapsychotic/clip-interrogator - Jupyter notebook uses CLIP models to suggest a prompt for images similar to a given image (Demo).
  • 🖊️Phraser - A visual prompt builder drawing on a database of examples. (Requires account creation)
  • 🖊️Prompthero = Another visual prompt builder and reference library.
  • PromptoMania - A visual prompt construction tool.
  • rom1504/clip-retrieval - Searches for prompt keywords in the datasets used in training Stable Diffusion and other models (Online GUI).
  • Stable Diffusion Prompt Generator - Gives suggestions for improving a given text prompt.

Specialized Usecases

  • dream-textures - A blender addon leveraging stable diffusion for texture creation.

Tutorials and Comparisons

Tutorials and high quality educational resources

Getting Up and Running

Learning and Mastering

Courses

Studies

Extending Functionality

Community Resources

  • 1 week of Stable Diffusion - A curated list of Stable Diffusion services, adaptations, user interfaces and integrations.
  • pharmapsychotic.com/tools - A curated list of Tools and Resources for AI Art, including but not limited to Stable Diffusion.
  • Stable Diffusion Resources - A thorough resource for answering pressing questions about Stable Diffusion, including guides, tutorials, and best software.

Social Media

SaaS implementing Stable Diffusion

  • 🖊️💵 Dream Studio - Online art generation service by StabilityAI, the creators of Stable Diffusion. Similar to services like DALL-E or Midjourney, this operates on a credit model with a free allowance of credits given to signed up users on a monthly basis.
  • 🖊️💵 dream.ai - Online art generation service by Wombo.ai (mobile apps available).
  • 🖊️💵 Image Computer - Easy-to-use service aimed at non-technical people (comes with free trial credits)
  • 🖊️💵 replicate.com stable diffusion - Another SaaS offering for Stable Diffusion.