Skip to content

Commit 220e27f

Browse files
Changes after running make style.
1 parent 4a1c7d4 commit 220e27f

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

src/diffusers/training_utils.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -59,18 +59,18 @@ def set_seed(seed: int):
5959

6060
def compute_snr(noise_scheduler, timesteps):
6161
"""
62-
Computes SNR as per https://github.com/TiankaiHang/Min-SNR-Diffusion-Training/blob/521b624bd70c67cee4bdf49225915f5945a872e3/guided_diffusion/gaussian_diffusion.py#L847-L849
62+
Computes SNR as per https://github.com/TiankaiHang/Min-SNR-Diffusion-Training/blob/521b624bd70c67cee4bdf49225915f5945a872e3/guided_diffusion/gaussian_diffusion.py#L847-L849
6363
for the given timesteps using the provided noise scheduler.
6464
6565
Args:
6666
noise_scheduler (`NoiseScheduler`):
6767
An object containing the noise schedule parameters, specifically `alphas_cumprod`, which is used
6868
to compute the SNR values.
6969
timesteps (`torch.Tensor`):
70-
A tensor of timesteps for which the SNR is computed.
70+
A tensor of timesteps for which the SNR is computed.
7171
7272
Returns:
73-
`torch.Tensor`: A tensor containing the computed SNR values for each timestep.
73+
`torch.Tensor`: A tensor containing the computed SNR values for each timestep.
7474
"""
7575
alphas_cumprod = noise_scheduler.alphas_cumprod
7676
sqrt_alphas_cumprod = alphas_cumprod**0.5

0 commit comments

Comments
 (0)