Where is major implementation part of guided diffusion?

SusungHong / Self-Attention-Guidance

The implementation of the paper "Improving Sample Quality of Diffusion Models Using Self-Attention Guidance" (ICCV`23)

MIT License

110 stars 14 forks source link

Where is major implementation part of guided diffusion? #9

Open john09282922 opened 9 months ago

john09282922 commented 9 months ago

I saw that 'the major implementation of our method is in ./guided_diffusion/gaussian_diffusion.py and ./guided_diffusion/unet.py.' Where is SAG part in there?

thanks,