Correspondance Layers in SDXL

Hello, 

Hi @Tsingularity  ! Thank you for this amazing work. Could you provide some intuitions on the applicability of SDXL and the best layer to extract features in SDXL? 

Also, do you think the method would apply to a purely transformer based architecture like SD3/DiT as well? 