typing.Union[torch.FloatTensor, numpy.ndarray]
generator: typing.Optional[torch._C.Generator] = None
return_dict: bool = True

)
→
SchedulerOutput or tuple

Parameters 

Predict the sample at the previous timestep by reversing the SDE. Core function to propagate the diffusion —


process from the learned model outputs (most often the predicted noise). —
model_output (torch.FloatTensor or np.ndarray): direct output from learned diffusion model. 