ut_dim (int, optional) — The dimension of the class_labels input when
class_embed_type="projection". Required when class_embed_type="projection".   class_embeddings_concat (bool, optional, defaults to False) — Whether to concatenate the time
embeddings with the class embeddings.    A conditional 2D UNet model that takes a noisy sample, conditional state, and a timestep and returns a sample
shaped output. Compared to the vanilla UNet2DConditionM