NVlabs / edm

Elucidating the Design Space of Diffusion-Based Generative Models (EDM)
Other
1.29k stars 137 forks source link

Using EDM for end-to-end training of latent diffusion models #28

Open pandekan opened 1 month ago

pandekan commented 1 month ago

It would be great if I could please get some guidance on how to adapt this framework for end-to-end training of the parameters of VAE and diffusion models on the VAE's latent space as in this LSGM paper by Vahdat et al. https://github.com/NVlabs/LSGM.