https://github.com/google/style-aligned

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

Style Aligned Image Generation via Shared Attention

Project Page   Paper

Setup

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

Examples

• See style_aligned_sdxl notebook for generating style aligned images using SDXL.

• See style_aligned_w_controlnet notebook for generating style aligned and depth conditioned images using SDXL with ControlNet-Depth.

• style_aligned_w_multidiffusion can be used for generating style aligned panoramas using SD V2 with MultiDiffusion.

TODOs

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

Disclaimer

This is not an officially supported Google product.