Add `timestep_spacing` and `steps_offset` to schedulers - #3947

Merged

patrickvonplaten merged 30 commits into

mainfrom

schedulers-spacing-main

Jul 5, 2023

Commits on Jul 4, 2023

Add timestep_spacing to DDPM, LMSDiscrete, PNDM.
pcuenca
committed
326172d
Remove spurious line.
pcuenca
committed
9ac281b
More easy schedulers.
pcuenca
committed
77075af
Add `linspace` to DDIM
pcuenca
committed
5e3c4f7
Noise sigma for `trailing`.
pcuenca
committed
c1e93d2
Add timestep_spacing to DEISMultistepScheduler.
pcuenca
committed
cd4cc28
Fix: remove line used to debug.
pcuenca
committed
02cdd87
Support timestep_spacing in DPMSolverMultistep, DPMSolverSDE, UniPC
pcuenca
committed
30e3a5d
Fix: convert to numpy.
pcuenca
committed
8375d0a
Use sched. defaults when instantiating from_config
pcuenca
committed
eb60db4
Apply suggestions from code review

pcuenca
and
patrickvonplaten
committed
81c5a1e
Missing args in DPMSolverMultistep
pcuenca
committed
a38021f
Test: default args not in config
pcuenca
committed
8f9c73c
Style
pcuenca
committed
d833276
Fix scheduler name in test
pcuenca
committed
3de3909
Remove duplicated entries
pcuenca
committed
5080d7f
Add test for solver_type
pcuenca
committed
083c8aa
UniPC: use same default for solver_type
pcuenca
committed
bef7669

Commits on Jul 5, 2023

do not save use default values
patrickvonplaten
committed
21d7eea
fix more
patrickvonplaten
committed
cb1109f
fix all
patrickvonplaten
committed
46c5c8f
fix schedulers
patrickvonplaten
committed
dc8c70b
fix more
patrickvonplaten
committed
6470353
finish for real
patrickvonplaten
committed
72949cb
finish for real
patrickvonplaten
committed
45328ef
flaky tests
patrickvonplaten
committed
8f4c443
Update tests/pipelines/stable_diffusion/test_stable_diffusion_pix2pix_zero.py
patrickvonplaten
authored
cedcb1b
Default steps_offset to 0.
pcuenca
committed
a8bdf06
Add missing docstrings
pcuenca
committed
f5a1d2c
Apply suggestions from code review
patrickvonplaten
authored
910d609