for all models (such as downloading or saving).  forward  < source > ( sample: FloatTensor timestep: Union class_labels: Optional = None return_dict: bool = True  ) → ~models.unet_2d.UNet2DOutput or tuple  Parameters   sample (torch.FloatTensor) —