ck2D, ~models.unet_2d_blocks.CrossAttnUpBlock2D,
~models.unet_2d_blocks.UNetMidBlock2DCrossAttn.    A conditional 2D UNet model that takes a noisy sample, conditional state, and a timestep and returns a sample
shaped output. This model inherits from ModelMixin. Check the superclass documentation for it’s generic methods implemented
for all models (such as downloading or saving). reverse_transformer_layers_per_block : (Tuple[Tuple], optional, de