(batch_size, channel * lowest_resolution, height // lowest_resolution, width // lowest_resolution).
Output can be used to condition the original UNet’s middle block activation.    The output of ControlNetModel.  FlaxControlNetModel  class diffusers.FlaxControlNetModel  < source > ( sample_size: int = 32 in_channels: int = 4 down_block_types: Tuple = ('CrossAttnDownBlock2D', 'CrossAttnDownBlock2D', 'CrossAttnDownBlock2D', 'DownBlock2D') only_cro