Tensor) — 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.


return_dict (bool) — option for returning tuple rather than SchedulerOutput class


Returns

~scheduling_utils.SchedulerOutput or tuple



~scheduling_utils.SchedulerOutput if return_dict is
True, otherwise a tuple. When returnin