) — number of diffusion steps used to train the model. beta_start (float): the


starting beta value of inference. beta_end (float) — the final beta value. beta_schedule (str):
the beta schedule, a mapping from a beta range to a sequence of betas for stepping the model. Choose from
linear or scaled_linear.


trained_betas (np.ndarray, optional) —
option to pass an array of betas directly to the constructor to bypass beta_start, beta_end etc.
op