Moves "Auto-SD" from bottom of list to top. Signed-off-by: Gadi Cohen <dragon@wastelands.net>
13 KiB
Awesome Stable-Diffusion
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.
-
lstein/stable-diffusion - Very active fork adding a conversational CLI, basic web interface and support for GFPGAN, ESRGAN, weighted prompts, img2img, 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.
-
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
- HuggingFace/StabilityAI - The official demo on HuggingFace Spaces.
- 🖊️💵 Offical Colab - The official, optimized colab for running SD on Google Cloud. Due to VRAM requirements required Colab Pro to create images.
- andreasjansson/stable-diffusion-animation - Animate between prompts.
- 🖊️ Stable Diffusion Interpolation - AA simple implementation of generating N interpolated images (Colab)
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
- textual-inversion - Addition of personalized content to Stable Diffusion without retraining the model (Paper, Paper2).
GUIS
- 🖊️💵 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.
- 🖊️ DiffusionUI - web UI made with Vue.js inspired by Dall-e using diffusers, perfect for inpainting. Video demo
- 🖊️ 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
- sd-webui/stable-diffusion-webui - GradIO based web UI included in an active fork.
- Stable Diffusion GRisk GUI - Windows GUI binary for SD.
Upscaling
- RealESRGAN - ESRGAN Upscaling (2x, 4x) and content restoration. Python:
pip install realesrgan
- Cupscale - GUI for running various upscaling models, including ESRGAN and RealESRGAN.
- BasicSR - Open-source upscaling and restoration toolbox supporting several models.
- BSRGAN - BSRGAN—another upscaling solution specialized in upscaling degraded images.
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.
- PromptoMania - A visual prompt construction tool.
- Lexica.art - A searchable, visual database of images and the prompts settings used to create them.
- 🖊️Phraser - A visual prompt builder drawing on a database of examples. (Requires account creation)
- ai-art.com/modifiers - A visual reference guide for keywords.
- pharmapsychotic/clip-interrogator - Jupyter notebook uses CLIP models to suggest a prompt for images similar to a given image (Demo).
- rom1504/clip-retrieval - Searches for prompt keywords in the datasets used in training Stable Diffusion and other models (Online GUI).
Tutorials and Comparisons
Tutorials and high quality educational resources
Getting Up and Running
- Stable Diffusion How To - A basic tutorial on getting Stable Diffusion up and running.
- Installing on Windows - A guide on installing and runing Stable Diffusion on Windows.
- Running on M1 Apple Silicon - Reddit thread with instructions on running Stable Diffusion on Apple M1 CPU and GPU.
- Easy CPU-only Stable Diffusion - A guide on setting up CPU-only Stable Diffusion for GNU/Linux without littering the system with dependencies.
- "Ultimate GUI Retard Guide" - Tutorial for installing the sd-webui fork.
Learning and Mastering
- Stable Diffusion Akashic Records - A comprehensive curated list of guides, studies, keywords, prompts and artists.
- Sunny's Tips & Tricks - Sunny's SD Tips & Tricks Google Doc with lots of visual comparisons and useful information.
- AI Image Generator Comparison - A visual comparison between Dall-e, Stable Diffusion and Midjourney by PetaPixel.com.
- Getting great results at Stable Diffusion - A guide on generating images that don't suck.
Studies
- Modifier Studies - Visual study of popular modifiers/keywords.
- Artist Studies - Visual study of various artists.
Extending Functionality
- Building a SD Discord Bot - A tutorial on building a Stable Diffusion Discord bot using Python.
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.
Social Media
- r/StableDiffusion - Stable Diffusion Subreddit.
Online Services 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).