Skip to content

Commit 149b92d

Browse files
Apply suggestions from code review
1 parent f196e28 commit 149b92d

File tree

8 files changed

+11
-22
lines changed

8 files changed

+11
-22
lines changed

src/diffusers/schedulers/scheduling_consistency_decoder.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,8 +26,7 @@ def betas_for_alpha_bar(
2626
2727
Args:
2828
num_diffusion_timesteps (`int`): the number of betas to produce.
29-
max_beta (`float`): the maximum beta to use
30-
use values lower than 1 to
29+
max_beta (`float`): the maximum beta to use. Use values lower than 1 to
3130
prevent singularities.
3231
alpha_transform_type (`str`, *optional*, default to `cosine`): the type of noise schedule for alpha_bar.
3332
Choose from `cosine` or `exp`

src/diffusers/schedulers/scheduling_ddim.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -63,8 +63,7 @@ def betas_for_alpha_bar(
6363
6464
Args:
6565
num_diffusion_timesteps (`int`): the number of betas to produce.
66-
max_beta (`float`): the maximum beta to use
67-
use values lower than 1 to
66+
max_beta (`float`): the maximum beta to use. Use values lower than 1 to
6867
prevent singularities.
6968
alpha_transform_type (`str`, *optional*, default to `cosine`): the type of noise schedule for alpha_bar.
7069
Choose from `cosine` or `exp`
@@ -162,8 +161,7 @@ class DDIMScheduler(SchedulerMixin, ConfigMixin):
162161
`set_alpha_to_one=False` to make the last step use step 0 for the previous alpha product like in Stable
163162
Diffusion.
164163
prediction_type (`str`, defaults to `epsilon`, *optional*):
165-
Prediction type of the scheduler function
166-
can be `epsilon` (predicts the noise of the diffusion process),
164+
Prediction type of the scheduler function. Can be `epsilon` (predicts the noise of the diffusion process),
167165
`sample` (directly predicts the noisy sample`) or `v_prediction` (see section 2.4 of [Imagen
168166
Video](https://imagen.research.google/video/paper.pdf) paper).
169167
thresholding (`bool`, defaults to `False`):

src/diffusers/schedulers/scheduling_ddim_inverse.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -61,8 +61,7 @@ def betas_for_alpha_bar(
6161
6262
Args:
6363
num_diffusion_timesteps (`int`): the number of betas to produce.
64-
max_beta (`float`): the maximum beta to use
65-
use values lower than 1 to
64+
max_beta (`float`): the maximum beta to use. Use values lower than 1 to
6665
prevent singularities.
6766
alpha_transform_type (`str`, *optional*, default to `cosine`): the type of noise schedule for alpha_bar.
6867
Choose from `cosine` or `exp`
@@ -160,8 +159,7 @@ class DDIMInverseScheduler(SchedulerMixin, ConfigMixin):
160159
`set_alpha_to_one=False` to make the last step use `num_train_timesteps - 1` for the previous alpha
161160
product.
162161
prediction_type (`str`, defaults to `epsilon`, *optional*):
163-
Prediction type of the scheduler function
164-
can be `epsilon` (predicts the noise of the diffusion process),
162+
Prediction type of the scheduler function. Can be `epsilon` (predicts the noise of the diffusion process),
165163
`sample` (directly predicts the noisy sample`) or `v_prediction` (see section 2.4 of [Imagen
166164
Video](https://imagen.research.google/video/paper.pdf) paper).
167165
timestep_spacing (`str`, defaults to `"leading"`):

src/diffusers/schedulers/scheduling_ddim_parallel.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -63,8 +63,7 @@ def betas_for_alpha_bar(
6363
6464
Args:
6565
num_diffusion_timesteps (`int`): the number of betas to produce.
66-
max_beta (`float`): the maximum beta to use
67-
use values lower than 1 to
66+
max_beta (`float`): the maximum beta to use. Use values lower than 1 to
6867
prevent singularities.
6968
alpha_transform_type (`str`, *optional*, default to `cosine`): the type of noise schedule for alpha_bar.
7069
Choose from `cosine` or `exp`

src/diffusers/schedulers/scheduling_ddpm.py

Lines changed: 2 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -60,8 +60,7 @@ def betas_for_alpha_bar(
6060
6161
Args:
6262
num_diffusion_timesteps (`int`): the number of betas to produce.
63-
max_beta (`float`): the maximum beta to use
64-
use values lower than 1 to
63+
max_beta (`float`): the maximum beta to use; Use values lower than 1 to
6564
prevent singularities.
6665
alpha_transform_type (`str`, *optional*, default to `cosine`): the type of noise schedule for alpha_bar.
6766
Choose from `cosine` or `exp`
@@ -115,8 +114,7 @@ class DDPMScheduler(SchedulerMixin, ConfigMixin):
115114
clip_sample_range (`float`, defaults to 1.0):
116115
The maximum magnitude for sample clipping. Valid only when `clip_sample=True`.
117116
prediction_type (`str`, defaults to `epsilon`, *optional*):
118-
Prediction type of the scheduler function
119-
can be `epsilon` (predicts the noise of the diffusion process),
117+
Prediction type of the scheduler function. Can be `epsilon` (predicts the noise of the diffusion process),
120118
`sample` (directly predicts the noisy sample`) or `v_prediction` (see section 2.4 of [Imagen
121119
Video](https://imagen.research.google/video/paper.pdf) paper).
122120
thresholding (`bool`, defaults to `False`):

src/diffusers/schedulers/scheduling_ddpm_parallel.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -62,8 +62,7 @@ def betas_for_alpha_bar(
6262
6363
Args:
6464
num_diffusion_timesteps (`int`): the number of betas to produce.
65-
max_beta (`float`): the maximum beta to use
66-
use values lower than 1 to
65+
max_beta (`float`): the maximum beta to use. Use values lower than 1 to
6766
prevent singularities.
6867
alpha_transform_type (`str`, *optional*, default to `cosine`): the type of noise schedule for alpha_bar.
6968
Choose from `cosine` or `exp`

src/diffusers/schedulers/scheduling_ddpm_wuerstchen.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -56,8 +56,7 @@ def betas_for_alpha_bar(
5656
5757
Args:
5858
num_diffusion_timesteps (`int`): the number of betas to produce.
59-
max_beta (`float`): the maximum beta to use
60-
use values lower than 1 to
59+
max_beta (`float`): the maximum beta to use. Use values lower than 1 to
6160
prevent singularities.
6261
alpha_transform_type (`str`, *optional*, default to `cosine`): the type of noise schedule for alpha_bar.
6362
Choose from `cosine` or `exp`

src/diffusers/schedulers/scheduling_deis_multistep.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -42,8 +42,7 @@ def betas_for_alpha_bar(
4242
4343
Args:
4444
num_diffusion_timesteps (`int`): the number of betas to produce.
45-
max_beta (`float`): the maximum beta to use
46-
use values lower than 1 to
45+
max_beta (`float`): the maximum beta to use. Use values lower than 1 to
4746
prevent singularities.
4847
alpha_transform_type (`str`, *optional*, default to `cosine`): the type of noise schedule for alpha_bar.
4948
Choose from `cosine` or `exp`

0 commit comments

Comments
 (0)