e dimension of the attention heads.


resnet_time_scale_shift (str, optional, defaults to "default") — Time scale shift config
for resnet blocks, see ResnetBlock2D. Choose from default or scale_shift.


class_embed_type (str, optional, defaults to None) — The type of class embedding to use which is ultimately
summed with the time embeddings. Choose from None, "timestep", or "identity".



UNet2DConditionModel is a conditional 2D UNet model that