revert rho change

rho? r? its just 1/2
This commit is contained in:
yoinked 2024-08-29 18:27:13 -07:00 committed by GitHub
parent 161d678f17
commit c807049ec2
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1135,7 +1135,8 @@ def sample_dpmpp_2s_ancestral_cfg_pp(model, x, sigmas, extra_args=None, callback
else: else:
# DPM-Solver++(2S) # DPM-Solver++(2S)
t, t_next = t_fn(sigmas[i]), t_fn(sigma_down) t, t_next = t_fn(sigmas[i]), t_fn(sigma_down)
r = torch.sinh(1 + (2 - eta) * (t_next - t) / (t - t_fn(sigma_up))) # r = torch.sinh(1 + (2 - eta) * (t_next - t) / (t - t_fn(sigma_up))) works only on non-cfgpp, weird
r = 1 / 2
h = t_next - t h = t_next - t
s = t + r * h s = t + r * h
x_2 = (sigma_fn(s) / sigma_fn(t)) * (x + (denoised - temp[0])) - (-h * r).expm1() * denoised x_2 = (sigma_fn(s) / sigma_fn(t)) * (x + (denoised - temp[0])) - (-h * r).expm1() * denoised