We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a73f8b7 commit feaa732Copy full SHA for feaa732
src/diffusers/schedulers/scheduling_ddpm.py
@@ -133,6 +133,10 @@ def __init__(
133
elif beta_schedule == "squaredcos_cap_v2":
134
# Glide cosine schedule
135
self.betas = betas_for_alpha_bar(num_train_timesteps)
136
+ elif beta_schedule == "sigmoid":
137
+ # GeoDiff sigmoid schedule
138
+ betas = torch.linspace(-6, 6, num_train_timesteps)
139
+ self.betas = torch.sigmoid(betas) * (beta_end - beta_start) + beta_start
140
else:
141
raise NotImplementedError(f"{beta_schedule} does is not implemented for {self.__class__}")
142
0 commit comments