 the learned model outputs (most often the predicted noise). —
model_output (torch.FloatTensor or np.ndarray): direct output from learned diffusion model. timestep
(int): current discrete timestep in the diffusion chain. sample (torch.FloatTensor or np.ndarray):
current instance of sample being created by diffusion process.
return_dict (bool): option for returning tuple rather than SchedulerOutput class


Returns

SchedulerOutput or tuple



Sc