mple (torch.FloatTensor) —
The noisy input tensor with the following shape (batch, channel, height, width).   timestep (torch.FloatTensor or float or int) — The number of timesteps to denoise an input.   encoder_hidden_states (torch.FloatTensor) —
The encoder hidden states with shape (batch, sequence_length, feature_dim).   encoder_attention_mask (torch.Tensor) —
A cross-attention mask of shape (batch, sequence_length) is applied to encoder_hid