Skip to content

Commit 236eaa2

Browse files
Apply suggestions from code review
1 parent 6425924 commit 236eaa2

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

src/diffusers/schedulers/scheduling_deis_multistep.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -95,8 +95,7 @@ class DEISMultistepScheduler(SchedulerMixin, ConfigMixin):
9595
The DEIS order which can be `1` or `2` or `3`. It is recommended to use `solver_order=2` for guided
9696
sampling, and `solver_order=3` for unconditional sampling.
9797
prediction_type (`str`, defaults to `epsilon`):
98-
Prediction type of the scheduler function
99-
can be `epsilon` (predicts the noise of the diffusion process),
98+
Prediction type of the scheduler function. Can be `epsilon` (predicts the noise of the diffusion process),
10099
`sample` (directly predicts the noisy sample`) or `v_prediction` (see section 2.4 of [Imagen
101100
Video](https://imagen.research.google/video/paper.pdf) paper).
102101
thresholding (`bool`, defaults to `False`):

0 commit comments

Comments
 (0)