t from learned diffusion model. timestep
(int): current discrete timestep in the diffusion chain. sample (torch.FloatTensor or np.ndarray):
current instance of sample being created by diffusion process.
return_dict (bool): option for returning tuple rather than SchedulerOutput class


Returns

SchedulerOutput or tuple



SchedulerOutput if return_dict is True, otherwise a tuple. When
returning a tuple, the first element is the sample tensor.
