on chain.


sample (torch.FloatTensor) —
current instance of sample being created by diffusion process.
order — coefficient for multi-step inference.


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


Returns

~schedulers.scheduling_utils.LMSDiscreteSchedulerOutput or tuple



~schedulers.scheduling_utils.LMSDiscreteSchedulerOutput if return_dict is True, otherwise a tuple.
When returning a tuple, t