Skip to content

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

License

Notifications You must be signed in to change notification settings

google/style-aligned

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

25 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Style Aligned Image Generation via Shared Attention

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

Demos

Thanks to @yvrjsharma for preparing the demos: style aligned text to image, ControlNet + StyleAligned and MultiDiffusion + StyleAligned

To start a demo locally, simply run

python <demo file name>.py  

and enter the demo in your browser using the provided url.

An online demo of ControlNet + StyleAligned is available here.

TODOs

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

Disclaimer

This is not an officially supported Google product.