Skip to content

Commit 968165f

Browse files
yiyixuxuJimmy
authored and
Jimmy
committed
[schedulers] create self.sigmas during __init__ (huggingface#6006)
* fix dpm * all scheulers
1 parent 3389762 commit 968165f

File tree

5 files changed

+5
-0
lines changed

5 files changed

+5
-0
lines changed

src/diffusers/schedulers/scheduling_deis_multistep.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -162,6 +162,7 @@ def __init__(
162162
self.alpha_t = torch.sqrt(self.alphas_cumprod)
163163
self.sigma_t = torch.sqrt(1 - self.alphas_cumprod)
164164
self.lambda_t = torch.log(self.alpha_t) - torch.log(self.sigma_t)
165+
self.sigmas = ((1 - self.alphas_cumprod) / self.alphas_cumprod) ** 0.5
165166

166167
# standard deviation of the initial noise distribution
167168
self.init_noise_sigma = 1.0

src/diffusers/schedulers/scheduling_dpmsolver_multistep.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -189,6 +189,7 @@ def __init__(
189189
self.alpha_t = torch.sqrt(self.alphas_cumprod)
190190
self.sigma_t = torch.sqrt(1 - self.alphas_cumprod)
191191
self.lambda_t = torch.log(self.alpha_t) - torch.log(self.sigma_t)
192+
self.sigmas = ((1 - self.alphas_cumprod) / self.alphas_cumprod) ** 0.5
192193

193194
# standard deviation of the initial noise distribution
194195
self.init_noise_sigma = 1.0

src/diffusers/schedulers/scheduling_dpmsolver_multistep_inverse.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -184,6 +184,7 @@ def __init__(
184184
self.alpha_t = torch.sqrt(self.alphas_cumprod)
185185
self.sigma_t = torch.sqrt(1 - self.alphas_cumprod)
186186
self.lambda_t = torch.log(self.alpha_t) - torch.log(self.sigma_t)
187+
self.sigmas = ((1 - self.alphas_cumprod) / self.alphas_cumprod) ** 0.5
187188

188189
# standard deviation of the initial noise distribution
189190
self.init_noise_sigma = 1.0

src/diffusers/schedulers/scheduling_dpmsolver_singlestep.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -172,6 +172,7 @@ def __init__(
172172
self.alpha_t = torch.sqrt(self.alphas_cumprod)
173173
self.sigma_t = torch.sqrt(1 - self.alphas_cumprod)
174174
self.lambda_t = torch.log(self.alpha_t) - torch.log(self.sigma_t)
175+
self.sigmas = ((1 - self.alphas_cumprod) / self.alphas_cumprod) ** 0.5
175176

176177
# standard deviation of the initial noise distribution
177178
self.init_noise_sigma = 1.0

src/diffusers/schedulers/scheduling_unipc_multistep.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -175,6 +175,7 @@ def __init__(
175175
self.alpha_t = torch.sqrt(self.alphas_cumprod)
176176
self.sigma_t = torch.sqrt(1 - self.alphas_cumprod)
177177
self.lambda_t = torch.log(self.alpha_t) - torch.log(self.sigma_t)
178+
self.sigmas = ((1 - self.alphas_cumprod) / self.alphas_cumprod) ** 0.5
178179

179180
# standard deviation of the initial noise distribution
180181
self.init_noise_sigma = 1.0

0 commit comments

Comments
 (0)