Michael 687bc27727 add TCD Scheduler (#7174) ...

* add: support TCD scheduler


---------

Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>

2024-03-04 19:43:34 -10:00

..

__init__.py

[Tests] Split scheduler tests (#2630)

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

test_scheduler_consistency_model.py

Add test_full_loop_with_noise tests to all scheduler with add_nosie function (#5184)

2023-09-27 13:08:37 +02:00

test_scheduler_ddim_inverse.py

Remove unused parameters and fixed FutureWarning (#6317)

2023-12-26 22:09:10 +01:00

test_scheduler_ddim_parallel.py

change to 2024 in the license (#6902)

2024-02-08 08:19:31 -10:00

test_scheduler_ddim.py

Add test_full_loop_with_noise tests to all scheduler with add_nosie function (#5184)

2023-09-27 13:08:37 +02:00

test_scheduler_ddpm_parallel.py

change to 2024 in the license (#6902)

2024-02-08 08:19:31 -10:00

test_scheduler_ddpm.py

Add rescale_betas_zero_snr Argument to DDPMScheduler (#6305)

2023-12-26 17:54:30 +01:00

test_scheduler_deis.py

Add test_full_loop_with_noise tests to all scheduler with add_nosie function (#5184)

2023-09-27 13:08:37 +02:00

test_scheduler_dpm_multi_inverse.py

refactor DPMSolverMultistepScheduler using sigmas (#4986)

2023-09-19 11:21:49 -10:00

test_scheduler_dpm_multi.py

DPMSolverMultistep add rescale_betas_zero_snr (#7097)

2024-02-27 11:37:34 -10:00

test_scheduler_dpm_sde.py

Lazy Import for Diffusers (#4829)

2023-09-11 09:56:22 +02:00

test_scheduler_dpm_single.py

fix DPM Scheduler with use_karras_sigmas option (#6477)

2024-01-19 07:08:22 -10:00

test_scheduler_edm_dpmsolver_multistep.py

add DPM scheduler with EDM formulation (#7120)

2024-02-27 23:39:38 +05:30

test_scheduler_edm_euler.py

Add EDMEulerScheduler (#7109)

2024-02-27 17:51:19 +05:30

test_scheduler_euler_ancestral.py

EulerAncestral add rescale_betas_zero_snr (#6187)

2023-12-20 13:09:25 +05:30

test_scheduler_euler.py

EulerDiscreteScheduler add rescale_betas_zero_snr (#6024)

2023-12-06 21:51:04 -10:00

test_scheduler_flax.py

change to 2024 in the license (#6902)

2024-02-08 08:19:31 -10:00

test_scheduler_heun.py

Add test_full_loop_with_noise tests to all scheduler with add_nosie function (#5184)

2023-09-27 13:08:37 +02:00

test_scheduler_ipndm.py

add a step_index counter (#4347)

2023-08-23 10:49:54 -10:00

test_scheduler_kdpm2_ancestral.py

Add test_full_loop_with_noise tests to all scheduler with add_nosie function (#5184)

2023-09-27 13:08:37 +02:00

test_scheduler_kdpm2_discrete.py

Add test_full_loop_with_noise tests to all scheduler with add_nosie function (#5184)

2023-09-27 13:08:37 +02:00

test_scheduler_lcm.py

Add Custom Timesteps Support to LCMScheduler and Supported Pipelines (#5874)

2023-11-27 12:39:14 +01:00

test_scheduler_lms.py

Add test_full_loop_with_noise tests to all scheduler with add_nosie function (#5184)

2023-09-27 13:08:37 +02:00

test_scheduler_pndm.py

[Tests] Split scheduler tests (#2630)

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

test_scheduler_sasolver.py

add Sa-Solver (#5975)

2024-01-21 21:37:44 -10:00

test_scheduler_score_sde_ve.py

[Tests] Split scheduler tests (#2630)

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

test_scheduler_tcd.py

add TCD Scheduler (#7174)

2024-03-04 19:43:34 -10:00

test_scheduler_unclip.py

[Tests] Split scheduler tests (#2630)

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

test_scheduler_unipc.py

Update UniPC to support 1D diffusion. (#5199)

2023-10-02 19:17:46 +02:00

test_scheduler_vq_diffusion.py

[Tests] Split scheduler tests (#2630)

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

test_schedulers.py

Add EDMEulerScheduler (#7109)

2024-02-27 17:51:19 +05:30