Skip to content

Commit 70611a1

Browse files
authored
Fix static typing and doc typos (#8807)
* Fix static typing and doc typos * Fix more same type hint typos with make fix-copies
1 parent 9838867 commit 70611a1

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

src/diffusers/schedulers/scheduling_ddim.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -377,7 +377,7 @@ def step(
377377
Whether or not to return a [`~schedulers.scheduling_ddim.DDIMSchedulerOutput`] or `tuple`.
378378
379379
Returns:
380-
[`~schedulers.scheduling_utils.DDIMSchedulerOutput`] or `tuple`:
380+
[`~schedulers.scheduling_ddim.DDIMSchedulerOutput`] or `tuple`:
381381
If return_dict is `True`, [`~schedulers.scheduling_ddim.DDIMSchedulerOutput`] is returned, otherwise a
382382
tuple is returned where the first element is the sample tensor.
383383

src/diffusers/schedulers/scheduling_ddpm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -194,7 +194,7 @@ def __init__(
194194
sample_max_value: float = 1.0,
195195
timestep_spacing: str = "leading",
196196
steps_offset: int = 0,
197-
rescale_betas_zero_snr: int = False,
197+
rescale_betas_zero_snr: bool = False,
198198
):
199199
if trained_betas is not None:
200200
self.betas = torch.tensor(trained_betas, dtype=torch.float32)

src/diffusers/schedulers/scheduling_ddpm_parallel.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -202,7 +202,7 @@ def __init__(
202202
sample_max_value: float = 1.0,
203203
timestep_spacing: str = "leading",
204204
steps_offset: int = 0,
205-
rescale_betas_zero_snr: int = False,
205+
rescale_betas_zero_snr: bool = False,
206206
):
207207
if trained_betas is not None:
208208
self.betas = torch.tensor(trained_betas, dtype=torch.float32)

0 commit comments

Comments
 (0)