step: typing.Union[float, torch.FloatTensor]
sample: FloatTensor
order: int = 4
return_dict: bool = True

)
→
~schedulers.scheduling_utils.LMSDiscreteSchedulerOutput or tuple

Parameters 

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


timestep (float) — current timestep in the diffusion chain.


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