image: FloatTensor
mask: FloatTensor
generator: typing.Optional[torch._C.Generator] = None
return_dict: bool = True

)
→
~schedulers.scheduling_utils.RePaintSchedulerOutput or tuple

Parameters 

model_output (torch.FloatTensor) — direct output from learned
diffusion model.


timestep (int) — current discrete timestep in the diffusion chain.


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


original