mirror of
https://github.com/finegrain-ai/refiners.git
synced 2024-11-21 21:58:47 +00:00
black
This commit is contained in:
parent
e70dee987e
commit
fc71e900a0
|
@ -56,9 +56,7 @@ class DDPM(Scheduler):
|
|||
else tensor(1, device=self.device)
|
||||
)
|
||||
current_factor = current_cumulative_factor / previous_cumulative_scale_factor
|
||||
estimated_denoised_data = (
|
||||
x - (1 - current_cumulative_factor) ** 0.5 * noise
|
||||
) / current_cumulative_factor**0.5
|
||||
estimated_denoised_data = (x - (1 - current_cumulative_factor) ** 0.5 * noise) / current_cumulative_factor**0.5
|
||||
estimated_denoised_data = estimated_denoised_data.clamp(-1, 1)
|
||||
original_data_coeff = (previous_cumulative_scale_factor**0.5 * (1 - current_factor)) / (
|
||||
1 - current_cumulative_factor
|
||||
|
|
Loading…
Reference in a new issue