kwargs: typing.Union[typing.Dict[str, typing.Any], NoneType] = 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


return_dict (bool, optional, defaults to True) —
