awesome-stable-diffusion/README.md
2022-08-29 02:14:55 +08:00

5.4 KiB

Awesome Stable-Diffusion

Awesome

This is a list of Free Software and Resources for the Stable Diffusion AI Model by StabilityAI. Non-Free commercial software is listed in a separate section below.

See Contributing.

Official Repositories

CompVis/Stable Diffusion - The official release of Stable Diffusion with cli, an AI-based Safety Classifier to (detect and suppresses sexualized content and all the necessary files to get running.

Forks

All forks listed here add additional features and optimisations and are generally faster than the original release as they will keep the model in memory rather than reloading it every prompt. Most forks seem to remove the Safety Classifier, making them potentially unsuitable for chat-bots and other environments that may deploy automated detection

  • basujindal/stable-diffusion - "Optimized Stable Diffusion", fork with dramatically reduced VRAM requirements through model splitting, enabling stable diffusion on lower end video cards. Includes gradio web interface, prompt weights
  • lstein/stable-diffusion - Fork adding a conversational cli interface, basic web interface, GFPGAN, ESRGAN, prompt weights, textual-inversion
  • progrock-stable - Fork with optional Web GUI, GoBIG/ESRGAN upscaling
  • hlky/stable-diffusion - Fork with optional Gradio UI with support for txt2img, img2img inpainting, GFPGAN, ESGRAN, prompt weights and optional textual-inversion

Models and Weights

Models (.ckpt files) must be separately downloaded and are required to run Stable Diffusion.

  • The offical model card - Official Model Card 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 from the model card, required to run Stable Diffusion.
  • RealESRGAN Models - Download location for the latest RealESRGAN models required to use upscaling by many forks implementing it. Different models exist for realistic and anime content. Please refer to the fork documentation to identify the ones you need.

Online Demos & Collabs

  • HuggingFace/StabilityAI - The official demo on HuggingFace Spaces.
  • Offical Collab - The official, optimized collab for running SD on Google Cloud. Due to VRAM requirements required Collab Pro to create images.

Complimentary Models and Tools (open source)

Customisation

Upscaling

  • GFPGAN - Face Restoration GAN included in several forks for automatically fixing deformed faces commonly found in SD output.* * RealESRGAN - ESRGAN Upscaling (2x, 4x) and content restoration, useful for overcoming. Python: pip install realesrgan
  • Cupscale - Graphical User Interface to run various upscaling models, including ESRGAN and RealESRGAN
  • BasicSR - Open Source Upscaling and Restoration toolbox supporting.
  • lama-cleaner - Content aware AI inpainting tool with useful for removing unwanted objects or defects from images. Python: pip install lama-cleaner

Task Chaining

  • chaiNNer - Graphical Node Based Editor for chaining image processing tasks.

Tutorials & Comparisons

Prompt Building

  • Prompt Mania - A visual prompt construction tool.
  • Lexica.art - A searchable, visual database of images and the prompts settings to recreate them.

Discords