model_output (torch.FloatTensor or np.ndarray) —
The direct output from learned diffusion model.   timestep (float or torch.FloatTensor) —
The current discrete timestep in the diffusion chain.   sample (torch.FloatTensor or np.ndarray) —
A current instance of a sample created by the diffusion process.   return_dict (bool, optional, defaults to True) —
Whether or not to return a SchedulerOutput or tuple.   s_noise (float, optional, defaults to 1