tas for stepping the model. Choose from
linear, scaled_linear, or squaredcos_cap_v2.   trained_betas (np.ndarray, optional) —
Pass an array of betas directly to the constructor to bypass beta_start and beta_end.   skip_prk_steps (bool, defaults to False) —
Allows the scheduler to skip the Runge-Kutta steps defined in the original paper as being required before
PLMS steps.   set_alpha_to_one (bool, defaults to False) —
Each diffusion step uses t