ablock_additional_residuals: Optional = None encoder_attention_mask: Optional = None return_dict: bool = True  ) → UNet2DConditionOutput or tuple  Parameters   sample (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 (