sion of the learnable embedding matrix to be projected to time_embed_dim, when performing
class conditioning with class_embed_type equal to None.



UNet2DModel is a 2D UNet model that takes in a noisy sample and a timestep and returns sample shaped output.
This model inherits from ModelMixin. Check the superclass documentation for the generic methods the library
implements for all the model (such as downloading or saving, etc.)

forward

<
sou