rce > ( model_output: FloatTensor sigma_hat: float sigma_prev: float sample_hat: FloatTensor sample_prev: FloatTensor derivative: FloatTensor return_dict: bool = True  ) → prev_sample (TODO)  Parameters   model_output (torch.FloatTensor) —
The direct output from learned diffusion model.   sigma_hat (float) — TODO   sigma_prev (float) — TODO   sample_hat (torch.FloatTensor) — TODO   sample_prev (torch.FloatTensor) — TODO   derivative (torch.Floa