amples with a pre-trained model.



Sets the continuous timesteps used for the diffusion chain. Supporting function to be run before inference.

step

<
source
>
(
model_output: FloatTensor
sigma_hat: float
sigma_prev: float
sample_hat: FloatTensor
return_dict: bool = True

)
→
KarrasVeOutput or tuple

Parameters 

model_output (torch.FloatTensor) — direct output from learned diffusion model.


sigma_hat (float) — TODO


sigma_prev (float) — TO