Viewing a single comment thread. View all comments

jakiwjakiw OP t1_ir6m6wg wrote

Reply to comment by Serverside in [R] Introduction to Diffusion Models in JAX by jakiwjakiw

That's a pretty interesting question! Actually, I don't know. As far as I know there are currently multiple ways to do conditional generation with SGMs, depending on what your requirements are. I like the work https://arxiv.org/abs/2111.13606 on that regard. But it's also something I wanted to explore in a bit more depth.

Very happy about any input about this!

2

Serverside t1_ir71pde wrote

Yeah I've read that paper you linked, but I have not really delved into trying to implement conditional SGM code myself (I've done work with conditional generative models in terms of GANs, VAEs, etc). I am also interested in lower dimensional data than images, so your code looked like a good starting point.

After some more reading, I'll give adding conditional capabilities to your code a shot.

1

jakiwjakiw OP t1_ir960po wrote

That's great to hear, please keep me posted on your progress!

1