Skip to content

Commit 7e39045

Browse files
Nathan LambertPrathik Rao
Nathan Lambert
authored and
Prathik Rao
committed
add sigmoid betas (huggingface#777)
* add sigmoid betas * convert to torch * add comment on source
1 parent 455689a commit 7e39045

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/diffusers/schedulers/scheduling_ddpm.py

+4
Original file line numberDiff line numberDiff line change
@@ -133,6 +133,10 @@ def __init__(
133133
elif beta_schedule == "squaredcos_cap_v2":
134134
# Glide cosine schedule
135135
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
136140
else:
137141
raise NotImplementedError(f"{beta_schedule} does is not implemented for {self.__class__}")
138142

0 commit comments

Comments
 (0)