shaped output. This model inherits from ModelMixin. Check the superclass documentation for it’s generic methods implemented
for all models (such as downloading or saving).  forward  < source > ( sample: FloatTensor timestep: Union return_dict: bool = True  ) → UNet1DOutput or tuple  Parameters   sample (torch.FloatTensor) —
The noisy input tensor with the following shape (batch_size, num_channels, sample_size).   timestep (torch.FloatTensor or 