
correct_steps: int = 1

)


Parameters 

num_train_timesteps (int) — number of diffusion steps used to train the model.


snr (float) —
coefficient weighting the step from the model_output sample (from the network) to the random noise.


sigma_min (float) —
initial noise scale for sigma sequence in sampling procedure. The minimum sigma should mirror the
distribution of the data.


sigma_max (float) — maximum value used for the range of continu