or: Optional = None return_dict: bool = True  ) → RePaintSchedulerOutput or tuple  Parameters   model_output (torch.FloatTensor) —
The direct output from learned diffusion model.   timestep (int) —
The current discrete timestep in the diffusion chain.   sample (torch.FloatTensor) —
A current instance of a sample created by the diffusion process.   original_image (torch.FloatTensor) —
The original image to inpaint on.   mask (torch.FloatTensor) 