nsor timestep: Union class_labels: Optional = None return_dict: bool = True  ) → ~models.unet_2d.UNet2DOutput 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.   class_labels (torch.FloatTensor, optional, defaults to None) —
Optional class labels for conditioning. Their 