Skip to content

Commit db8dc50

Browse files
committed
merge schedulers into one file
1 parent 1d0060e commit db8dc50

File tree

9 files changed

+46
-140
lines changed

9 files changed

+46
-140
lines changed

docs/source/_toctree.yml

Lines changed: 0 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -29,22 +29,6 @@
2929
- local: pipelines/glide
3030
title: "Glide"
3131
title: Pipelines
32-
- sections:
33-
- local: schedulers_doc/ddpm
34-
title: "DDPM"
35-
- local: schedulers_doc/ddim
36-
title: "DDIM"
37-
- local: schedulers_doc/kerras_ve
38-
title: "KerrasVE"
39-
- local: schedulers_doc/lms_discrete
40-
title: "LMS-Discrete"
41-
- local: schedulers_doc/pndm
42-
title: "PNDM"
43-
- local: schedulers_doc/sde_ve
44-
title: "SDE-VE"
45-
- local: schedulers_doc/sde_vp
46-
title: "SDE-VP"
47-
title: Schedulers
4832
- sections:
4933
- local: models/unet
5034
title: "Unet"

docs/source/schedulers.mdx

Lines changed: 46 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -38,9 +38,50 @@ with a `set_format(...)` method.
3838

3939
[[autodoc]] SchedulerMixin
4040

41-
## Examples
41+
## Existing Classes
4242

43-
- The ['DDPMScheduler'] was proposed in [Denoising Diffusion Probabilistic Models](https://arxiv.org/abs/2006.11239) and can be found in [scheduling_ddpm.py](https://github.com/huggingface/diffusers/blob/main/src/diffusers/schedulers/scheduling_ddpm.py).
44-
An example of how to use this scheduler can be found in [pipeline_ddpm.py](https://github.com/huggingface/diffusers/blob/main/src/diffusers/pipelines/pipeline_ddpm.py).
45-
- The ['DDIMScheduler'] was proposed in [Denoising Diffusion Implicit Models](https://arxiv.org/abs/2010.02502) and can be found in [scheduling_ddim.py](https://github.com/huggingface/diffusers/blob/main/src/diffusers/schedulers/scheduling_ddim.py). An example of how to use this scheduler can be found in [pipeline_ddim.py](https://github.com/huggingface/diffusers/blob/main/src/diffusers/pipelines/pipeline_ddim.py).
46-
- The ['PNDMScheduler'] was proposed in [Pseudo Numerical Methods for Diffusion Models on Manifolds](https://arxiv.org/abs/2202.09778) and can be found in [scheduling_pndm.py](https://github.com/huggingface/diffusers/blob/main/src/diffusers/schedulers/scheduling_pndm.py). An example of how to use this scheduler can be found in [pipeline_pndm.py](https://github.com/huggingface/diffusers/blob/main/src/diffusers/pipelines/pipeline_pndm.py).
43+
### DDIM
44+
45+
DDIM is a scheduler.
46+
47+
[[autodoc]] schedulers.scheduling_ddim.DDIMScheduler
48+
49+
### DDPM
50+
51+
DDPM is a scheduler.
52+
53+
[[autodoc]] schedulers.scheduling_ddpm.DDPMScheduler
54+
55+
### KerrasVE
56+
57+
KerrasVE is a scheduler.
58+
59+
[[autodoc]] schedulers.scheduling_karras_ve.KarrasVeScheduler
60+
61+
### LMSDiscrete
62+
63+
LMSDiscrete is a scheduler.
64+
65+
[[autodoc]] schedulers.scheduling_lms_discrete.LMSDiscreteScheduler
66+
67+
### PNDM
68+
69+
PNDMDiscrete is a scheduler.
70+
71+
[[autodoc]] schedulers.scheduling_pndm.PNDMScheduler
72+
73+
### Score SDE-VE
74+
75+
Score SDE-VE is a scheduler.
76+
77+
[[autodoc]] schedulers.scheduling_sde_ve.ScoreSdeVeScheduler
78+
79+
### Score SDE-VP
80+
81+
<Tip warning={true}>
82+
83+
Score SDE-VP is under construction.
84+
85+
</Tip>
86+
87+
[[autodoc]] schedulers.scheduling_sde_vp.ScoreSdeVpScheduler

docs/source/schedulers_doc/ddim.mdx

Lines changed: 0 additions & 17 deletions
This file was deleted.

docs/source/schedulers_doc/ddpm.mdx

Lines changed: 0 additions & 17 deletions
This file was deleted.

docs/source/schedulers_doc/kerras_ve.mdx

Lines changed: 0 additions & 17 deletions
This file was deleted.

docs/source/schedulers_doc/lms_discrete.mdx

Lines changed: 0 additions & 17 deletions
This file was deleted.

docs/source/schedulers_doc/pndm.mdx

Lines changed: 0 additions & 17 deletions
This file was deleted.

docs/source/schedulers_doc/sde_ve.mdx

Lines changed: 0 additions & 17 deletions
This file was deleted.

docs/source/schedulers_doc/sde_vp.mdx

Lines changed: 0 additions & 17 deletions
This file was deleted.

0 commit comments

Comments
 (0)