Skip to content

Commit 2f023d7

Browse files
authored
Fix RePaint Scheduler (#10185)
Fix repaint scheduler
1 parent e9a3911 commit 2f023d7

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/diffusers/schedulers/scheduling_repaint.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -319,7 +319,7 @@ def step(
319319
prev_unknown_part = alpha_prod_t_prev**0.5 * pred_original_sample + pred_sample_direction + variance
320320

321321
# 8. Algorithm 1 Line 5 https://arxiv.org/pdf/2201.09865.pdf
322-
prev_known_part = (alpha_prod_t_prev**0.5) * original_image + ((1 - alpha_prod_t_prev) ** 0.5) * noise
322+
prev_known_part = (alpha_prod_t_prev**0.5) * original_image + (1 - alpha_prod_t_prev) * noise
323323

324324
# 9. Algorithm 1 Line 8 https://arxiv.org/pdf/2201.09865.pdf
325325
pred_prev_sample = mask * prev_known_part + (1.0 - mask) * prev_unknown_part

0 commit comments

Comments
 (0)