Youssef Adarrab 11f527ac0f Add Karras sigmas to HeunDiscreteScheduler (#3160) ...

* Add karras pattern to discrete heun scheduler

* Add integration test

* Fix failing CI on pytorch test on M1 (mps)

---------

Co-authored-by: Patrick von Platen <patrick.v.platen@gmail.com>

2023-04-21 19:21:04 +01:00

..

__init__.py

[Tests] Split scheduler tests (#2630)

2023-03-09 19:11:47 +01:00

test_scheduler_ddim.py

[Tests] Split scheduler tests (#2630)

2023-03-09 19:11:47 +01:00

test_scheduler_ddpm.py

ddpm custom timesteps (#3007)

2023-04-14 12:39:38 -07:00

test_scheduler_deis.py

[Tests] Split scheduler tests (#2630)

2023-03-09 19:11:47 +01:00

test_scheduler_dpm_multi.py

[2064]: Add Karras to DPMSolverMultistepScheduler (#3001)

2023-04-12 18:04:51 +05:30

test_scheduler_dpm_single.py

[Tests] Split scheduler tests (#2630)

2023-03-09 19:11:47 +01:00

test_scheduler_euler_ancestral.py

[Tests] Split scheduler tests (#2630)

2023-03-09 19:11:47 +01:00

test_scheduler_euler.py

[2905]: Add Karras pattern to discrete euler (#2956)

2023-04-06 16:10:57 +05:30

test_scheduler_flax.py

[Tests] Split scheduler tests (#2630)

2023-03-09 19:11:47 +01:00

test_scheduler_heun.py

Add Karras sigmas to HeunDiscreteScheduler (#3160)

2023-04-21 19:21:04 +01:00

test_scheduler_ipndm.py

[Tests] Split scheduler tests (#2630)

2023-03-09 19:11:47 +01:00

test_scheduler_kdpm2_ancestral.py

[Tests] Split scheduler tests (#2630)

2023-03-09 19:11:47 +01:00

test_scheduler_kdpm2_discrete.py

[Tests] Split scheduler tests (#2630)

2023-03-09 19:11:47 +01:00

test_scheduler_lms.py

[Tests] Split scheduler tests (#2630)

2023-03-09 19:11:47 +01:00

test_scheduler_pndm.py

[Tests] Split scheduler tests (#2630)

2023-03-09 19:11:47 +01:00

test_scheduler_score_sde_ve.py

[Tests] Split scheduler tests (#2630)

2023-03-09 19:11:47 +01:00

test_scheduler_unclip.py

[Tests] Split scheduler tests (#2630)

2023-03-09 19:11:47 +01:00

test_scheduler_unipc.py

tests and additional scheduler fixes

2023-04-10 12:59:33 -07:00

test_scheduler_vq_diffusion.py

[Tests] Split scheduler tests (#2630)

2023-03-09 19:11:47 +01:00

test_schedulers.py

Fix config deprecation (#3129)

2023-04-17 17:16:28 +01:00