Add `timestep_spacing` and `steps_offset` to schedulers #3947
patrickvonplaten
merged
30 commits into
main
from
schedulers-spacing-main
Add timestep_spacing to DDPM, LMSDiscrete, PNDM.
326172d1
Remove spurious line.
9ac281b0
More easy schedulers.
77075af9
Add `linspace` to DDIM
5e3c4f7f
Noise sigma for `trailing`.
c1e93d2d
Add timestep_spacing to DEISMultistepScheduler.
cd4cc28c
Fix: remove line used to debug.
02cdd877
Support timestep_spacing in DPMSolverMultistep, DPMSolverSDE, UniPC
30e3a5db
Fix: convert to numpy.
8375d0af
Use sched. defaults when instantiating from_config
eb60db4b
Apply suggestions from code review
81c5a1e9
Missing args in DPMSolverMultistep
a38021fc
Test: default args not in config
8f9c73c1
Style
d833276c
Fix scheduler name in test
3de39093
Remove duplicated entries
5080d7fa
Add test for solver_type
083c8aab
UniPC: use same default for solver_type
bef7669b
pcuenca
changed the title 
do not save use default values
21d7eea5
fix more
cb1109f0
fix all
46c5c8f5
pcuenca
commented
on 2023-07-05
pcuenca
commented
on 2023-07-05
fix schedulers
dc8c70b0
fix more
64703530
finish for real
72949cb6
finish for real
45328ef6
flaky tests
8f4c443b
patrickvonplaten
approved these changes
on 2023-07-05
patrickvonplaten
commented
on 2023-07-05
Update tests/pipelines/stable_diffusion/test_stable_diffusion_pix2pix…
cedcb1b4
patrickvonplaten
commented
on 2023-07-05
Default steps_offset to 0.
a8bdf065
Add missing docstrings
f5a1d2c4
patrickvonplaten
commented
on 2023-07-05
Apply suggestions from code review
910d609b
Login to write a write a comment.
Login via GitHub