Class-Conditional using Unet2DModel

Hello everyone.

I want to try create simpe model that can generate images from noise and class vector, but i can understant how to create this. I saw an article somewhere where it was said that Unet2DModel can be used for these purposes, but I don’t understand what parameters in the model configuration need to be configured for everything to work.

Thanks for any kind of help!