Skip to content

GitLab

Switch branch/tag
Find file BlameHistoryPermalink
  • Beinsezii's avatar
    DPMSolverMultistep add `rescale_betas_zero_snr` (#7097) · 2e31a759
    Beinsezii authored 
    * DPMMultistep rescale_betas_zero_snr

* DPM upcast samples in step()

* DPM rescale_betas_zero_snr UT

* DPMSolverMulti move sample upcast after model convert

Avoids having to re-use the dtype.

* Add a newline for Ruff
    2e31a759
scheduling_dpmsolver_multistep.py 46.2 KB