google / style-aligned

Official code for "Style Aligned Image Generation via Shared Attention"
Apache License 2.0
1.19k stars 86 forks source link

Custom train weights #18

Closed kimtaehyeong closed 8 months ago

kimtaehyeong commented 8 months ago

Very great work!

I have one question. Does style-aligned work well when I have custom learned diffuser weights?

Thanks.

anvoynov commented 8 months ago

Hi! We tested it on different checkpoints of general text-to-image models and on dreambooth-tuned, it works in both cases so I guess it should work. If you will try – it would be very interesting to see your results.