How to train the Unet

[shenwenxiao]

What an amazing job! This work can help me a lot in my study.

I did not find the training code for the Unet. If I want to train the Unet for my data, how should I do? You mention that this work is based on "guided-diffusion". Is the Unet is trained by the code of guided-diffusion?

[omriav]

Hi,

Thank you very much.
I am recommending using Diffusers Stable Diffusion implementation of our follow-up work Blended Latent Diffusion.

Then, you can just train your Stable Diffusion model using the Diffusers tutorial and use the trained model with the aforementioned code.

Best,
Omri