in the diffusion chain.


sample (torch.FloatTensor) —
current instance of sample being created by diffusion process.


original_image (torch.FloatTensor) —
the original image to inpaint on.


mask (torch.FloatTensor) —
the mask where 0.0 values define which part of the original image to inpaint (change).


generator (torch.Generator, optional) — random number generator.


return_dict (bool) — option for returning tuple rather than
DDPMSchedule