fix(edm2-sample): multistep_corrector_update crashes for (+1 more)

[andrewwhitecdw]

Small fixes in EDM2_sample.py:

fix: multistep_corrector_update crashes for order=1 with IndexError

Fix: Replace:

        indexes = [-i - 1 for i in range(n + 1)]

with:

        indexes = [-i - 1 for i in range(max(n + 1, 2))]

fix: DPM_Solver_v3 references missing total_N when t_end is None

Fix: Replace:

        t_0 = 1.0 / self.noise_schedule.total_N if t_end is None else t_end

with:

        assert t_end is not None, "t_end must be specified for NoiseScheduleEDM"
        t_0 = t_end

Files changed

• EDM2_sample.py

[andrewwhitecdw]

Closing this sweep-generated PR: PR has 3 commits; sweep requires exactly one commit per PR. It does not meet the sweep requirements (single signed-off commit).