search

LingxiaoYang2023 / DSG2024

Official pytorch repository for “Guidance with Spherical Gaussian Constraint for Conditional Diffusion”
43 stars 2 forks source link

Add DSG to other codes #4

Open zmy-08 opened 5 months ago

zmy-08 commented 5 months ago

This is a very surprising job! How can I add DSG to other codes? Where is the specific modification? I want to test the function of DSG in my own diffusion model of text to image generation, hoping to get your answer.

LingxiaoYang2023 commented 5 months ago

Thanks so much for your interest in our work! The main algorithm of DSG is in _/guided_diffusion/conditionmethods.py, Line 100 - 136. You may also track _/guided_diffusion/gaussiandiffusion.py, as we have made a minor modification to include a new parameter, _x_tmean.

Maguire1999 commented 4 months ago

Hello, I am interested in investigating the impact of incorporating DSG into other models.