Where exactly is the spatial attention between ReferenceNet and DenoisingUnet in code ?

guoqincode / Open-AnimateAnyone

Unofficial Implementation of Animate Anyone

2.85k stars 230 forks source link

Where exactly is the spatial attention between ReferenceNet and DenoisingUnet in code ? #92

Open mayank64ce opened 6 months ago

mayank64ce commented 6 months ago

I am very new to these kinds of models so please help me understand where where in the code the output of ReferenceNet's spatial attention and cross-attention going in the DenoisingUnet ?

Thanks!