Skip to content

GitLab

"ppdet/utils/__init__.py" did not exist on "dcc7bf4f1a243d90d6c4f7c51551cea3f256325f"
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
test_scheduler_dpm_multi.py 13.3 KB