earned diffusion model.   timestep (float) —
The current discrete timestep in the diffusion chain.   sample (torch.FloatTensor) —
A current instance of a sample created by the diffusion process.   return_dict (bool) —
Whether or not to return a SchedulerOutput or tuple.   Returns 
SchedulerOutput or tuple
  
If return_dict is True, SchedulerOutput is returned, otherwise a
tuple is returned where the first element is the sample tensor.
  Predict