/StyleAligned

StyleAligned

Primary LanguagePython

Style Aligned Image Generation via Shared Attention

Setup

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

Examples

使用SDXLBase的结果: alt text alt text alt text SDXLBase的结果在T-Sne上的可视化: alt text 使用ICON的结果: alt text ICON的结果在T-Sne上的可视化: alt text

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.