aim-uofa / FreeCustom

[CVPR 2024] Official PyTorch implementation of FreeCustom: Tuning-Free Customized Image Generation for Multi-Concept Composition
https://aim-uofa.github.io/FreeCustom/
MIT License
62 stars 0 forks source link

Can reference images be processed invidually ? #3

Open wzic opened 2 weeks ago

wzic commented 2 weeks ago

Hi! In the code, it seems that the ref latents are parallelly concatenated with the own latent and being denoised simultaneously ? May I know if it is equivalent to first invidually get the self-attention key and value of each ref invidually, and then concat them together during denoising ?

dingangui commented 2 weeks ago

yes, you can get the self-attention key and value of each ref invidually first, and then concat them together during denoising, it will save your GPU memory

wzic commented 2 weeks ago

Thanks