Ad

stable-diffusion: Latent text-to-image diffusion model

Stable Diffusion generates high-resolution images from text prompts. It's a lightweight diffusion model requiring 10GB VRAM and is available for research use with licensing restrictions.
Screenshot of CompVis/stable-diffusion homepage

Stable Diffusion is a latent text-to-image diffusion model developed in collaboration with Stability AI and Runway. It leverages a frozen CLIP ViT-L/14 text encoder to condition a 860M UNet, enabling efficient image generation with a minimum of 10GB VRAM. The model was pretrained on LAION-5B and finetuned on high-resolution images, providing a balance of speed and quality.

Stable Diffusion distinguishes itself through its lightweight architecture, achieving quality comparable to larger models with significantly reduced computational resources. Its use of a frozen CLIP ViT-L/14 text encoder contributes to faster inference. The project emphasizes responsible AI through a CreativeML OpenRAIL M license and provides tools for mitigating biases and harmful outputs.

  • Core Functionality: Generates high-resolution images from text prompts using a latent diffusion process.
  • Supported Platforms: Primarily designed for GPU execution, with integrations available for various frameworks.
  • Configuration/Extensibility: Offers flexible parameters like guidance scale, sampling steps, and various samplers for controlling image generation.
  • Performance/Scalability: Efficient resource usage (10GB VRAM) enables deployment on consumer GPUs.
  • Developer Experience: Provides a reference script and a diffusers integration for ease of use and experimentation.
  • Image Resolution: Supports image generation at resolutions from 256x256 to 512x512 and beyond.
  • Watermarking: Includes optional invisible watermarking to identify machine-generated images.

Stable Diffusion is a mature project with active development and a strong community presence, evidenced by a high number of stars and forks. Regular model checkpoints are released with improved performance and features. Detailed documentation, including a model card with limitations and biases, guides users on responsible usage. The project adheres to responsible AI licensing practices.

Stable Diffusion benefits researchers and developers seeking a powerful yet accessible text-to-image generation model. It is valuable for creating visual content, prototyping image-based applications, and conducting research on generative models. It provides a feasible alternative to more computationally expensive models, while also prompting careful consideration of responsible deployment due to inherent biases in its training data.

Summarize:
Share:
Stars
73,169
Forks
10,589
Issues
616
Created
3 years ago
Commit
2 years ago
License
OTHER
Archived
No
Updated 16 days ago

Similar Repositories