Skip to content

schedulers/dpmsolver_multistep: add missing sde-dpmsolver branch in third-order update#13778

Open
Dev-X25874 wants to merge 1 commit into
huggingface:mainfrom
Dev-X25874:fix/dpmsolver-multistep-third-order-sde-branch
Open

schedulers/dpmsolver_multistep: add missing sde-dpmsolver branch in third-order update#13778
Dev-X25874 wants to merge 1 commit into
huggingface:mainfrom
Dev-X25874:fix/dpmsolver-multistep-third-order-sde-branch

Conversation

@Dev-X25874
Copy link
Copy Markdown
Contributor

This happens silently after the two lower-order warm-up steps complete (i.e. on step 3+),
making it a hard runtime crash for any user who configures sde-dpmsolver with
solver_order=3.

The fix adds the missing elif self.config.algorithm_type == "sde-dpmsolver" branch,
derived from the same factor-of-2 doubling pattern that governs the sde-dpmsolver
formulation relative to dpmsolver at every other order (per Appendix A of the
DPMSolver paper), and using the identical
noise term sigma_t * sqrt(exp(2h) - 1) * noise already present in the 1st- and
2nd-order sde-dpmsolver branches.

Fixes # (issue)

Before submitting

Who can review?

@yiyixuxu

@github-actions github-actions Bot added schedulers size/S PR with diff < 50 LOC labels May 21, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

schedulers size/S PR with diff < 50 LOC

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant