MidBlock2DSimpleCrossAttn. If
only_cross_attention is given as a single boolean and mid_block_only_cross_attention is None, the
only_cross_attention value will be used as the value for mid_block_only_cross_attention. Else, it will
default to False.



UNet2DConditionModel is a conditional 2D UNet model that takes in a noisy sample, conditional state, and a timestep
and returns sample shaped output.
This model inherits from ModelMixin. Check the