Skip to content

Commit

Permalink
Fix RePaint Scheduler (#10185)
Browse files Browse the repository at this point in the history
Fix repaint scheduler
  • Loading branch information
hlky authored Dec 16, 2024
1 parent e9a3911 commit 2f023d7
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/diffusers/schedulers/scheduling_repaint.py
Original file line number Diff line number Diff line change
Expand Up @@ -319,7 +319,7 @@ def step(
prev_unknown_part = alpha_prod_t_prev**0.5 * pred_original_sample + pred_sample_direction + variance

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

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

0 comments on commit 2f023d7

Please sign in to comment.