drozbay 4d79f4f028 fix: handle substep sigmas in context window set_step (#12719) ...

Multi-step samplers (eg. dpmpp_2s_ancestral) call the model at intermediate sigma values not present in the schedule. This caused set_step to crash with "No sample_sigmas matched current timestep" when context windows were enabled.

The fix is to keep self._step from the last exact match when a substep sigma is encountered, since substeps are still logically part of their parent step and should use the same context windows.

Co-authored-by: ozbayb <17261091+ozbayb@users.noreply.github.com>

2026-03-01 09:38:30 -08:00

31 KiB

Raw Blame History

View Raw