FastComposer

[IJCV] FastComposer: Tuning-Free Multi-Subject Image Generation with Localized Attention

AI Image EditingOpen Source AI & Machine LearningOpen Source

About

Diffusion models excel at text-to-image generation, especially in subject-driven generation for personalized images. However, existing methods are inefficient due to the subject-specific fine-tuning, which is computationally intensive and hampers efficient deployment. Moreover, existing methods struggle with multi-subject generation as they often blend features among subjects. We present FastComposer which enables efficient, personalized, multi-subject text-to-image generation without fine-tuning. FastComposer uses subject embeddings extracted by an image encoder to augment the generic text conditioning in diffusion models, enabling personalized image generation based on subject images and textual instructions with only forward passes. To address the identity blending problem in the multi-subject generation, FastComposer proposes cross-attention localization supervision during training, enforcing the attention of reference subjects localized to the correct regions in the target images. Naively conditioning on subject embeddings results in subject overfitting.

Open Source Health

Not enough history
Stars
714
Forks
43
License
MIT
Last commit
2 years ago
Python

Related Categories

Using FastComposer?

Track its cost next to the rest of your stack and get a reminder before it renews.

Add to my stack

Vendor

MIT HAN Lab

Efficient AI Computing. PI: Song Han

View vendor profile

More by MIT HAN Lab