Refactor SchedulerOutput and add pred_original_sample in `DPMSolverSDE`, `Heun`, `KDPM2Ancestral` and `KDPM2`#9650

Merged

yiyixuxu merged 2 commits intohuggingface:mainfrom

hlky:refactor-scheduleroutput

Oct 15, 2024

Commits on Oct 11, 2024

Refactor SchedulerOutput and add pred_original_sample
hlky
committed

Commits on Oct 15, 2024

Merge branch 'main' into refactor-scheduleroutput
yiyixuxu
authored