omriav/blended-diffusion

Would you open the training code?

kings-rgb opened this issue · 1 comments

kings-rgb commented

It's a very interesting job. We want to know how to train the GAN with opened dataset. So, would you open the training code?

omriav commented

Hi,

Thank you for your interest in our project!

  1. The underlying generative model is not a GAN but a diffusion model.
  2. If you want to train your own custom diffusion model and then guide it using text, you can find the training code in OpenAI's repository and then put the trained model in the chekpoints directory in our project.

Good luck!