/style-aligned

Official code for "Style Aligned Image Generation via Shared Attention"

Primary LanguagePythonApache License 2.0Apache-2.0

Style Aligned Image Generation via Shared Attention

Project PagePaper

Setup

This code was tested with Python 3.11, Pytorch 2.1 and Diffusers 0.16.

Examples

alt text

alt text

alt text

alt text

TODOs

  • Adding demo.
  • StyleAligned from an input image.
  • Multi-style with MultiDiffusion.
  • StyleAligned with DreamBooth

Disclaimer

This is not an officially supported Google product.