odel outputs (most often the predicted noise).

step_correct

<
source
>
(
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) — direct output from learned diffusion model.


sigma_hat (float) — TODO


sigma_prev (float) — TODO


sample_hat (torch.FloatTensor) — TODO


