mirror of
https://github.com/finegrain-ai/refiners.git
synced 2024-11-09 23:12:02 +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)
|
else tensor(1, device=self.device)
|
||||||
)
|
)
|
||||||
current_factor = current_cumulative_factor / previous_cumulative_scale_factor
|
current_factor = current_cumulative_factor / previous_cumulative_scale_factor
|
||||||
estimated_denoised_data = (
|
estimated_denoised_data = (x - (1 - current_cumulative_factor) ** 0.5 * noise) / current_cumulative_factor**0.5
|
||||||
x - (1 - current_cumulative_factor) ** 0.5 * noise
|
|
||||||
) / current_cumulative_factor**0.5
|
|
||||||
estimated_denoised_data = estimated_denoised_data.clamp(-1, 1)
|
estimated_denoised_data = estimated_denoised_data.clamp(-1, 1)
|
||||||
original_data_coeff = (previous_cumulative_scale_factor**0.5 * (1 - current_factor)) / (
|
original_data_coeff = (previous_cumulative_scale_factor**0.5 * (1 - current_factor)) / (
|
||||||
1 - current_cumulative_factor
|
1 - current_cumulative_factor
|
||||||
|
|
Loading…
Reference in a new issue