typing.Optional[torch.Tensor] = None
encoder_attention_mask: typing.Optional[torch.Tensor] = None
return_dict: bool = True

)
→
UNet2DConditionOutput or tuple

Parameters 

sample (torch.FloatTensor) — (batch, channel, height, width) noisy inputs tensor


timestep (torch.FloatTensor or float or int) — (batch) timesteps


encoder_hidden_states (torch.FloatTensor) — (batch, sequence_length, feature_dim) encoder hidden states


encoder_attention_m