ax: float = 1348.0 sampling_eps: float = 1e-05 correct_steps: int = 1  )   Parameters   num_train_timesteps (int, defaults to 1000) —
The number of diffusion steps to train the model.   snr (float, defaults to 0.15) —
A coefficient weighting the step from the model_output sample (from the network) to the random noise.   sigma_min (float, defaults to 0.01) —
The initial noise scale for the sigma sequence in the sampling procedure. The minimum si