From 1dfdcfe93322d11fd1e1c6cf7e94a8a36cf6feed Mon Sep 17 00:00:00 2001
From: blepping <blepping@tuta.io>
Date: Tue, 6 Feb 2024 11:42:42 -0700
Subject: [PATCH] Correctly update hd in dpmpp_sde sampler

---
 py/sonar.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/py/sonar.py b/py/sonar.py
index d880d72..27ec6e4 100644
--- a/py/sonar.py
+++ b/py/sonar.py
@@ -440,6 +440,7 @@ class SonarDPMPPSDE(SonarSampler):
         diff_2 = (t - s_).expm1() * denoised
         momentum_d = (1.0 - p) * diff_2 + p * hd
         self.update_hist(momentum_d)
+        hd = self.history_d
         x_2 = (sigma_fn(s_) / sigma_fn(t)) * x - momentum_d
         x_2 = x_2 + self.noise_sampler(sigma_fn(t), sigma_fn(s)) * self.s_noise * su
         denoised_2 = self.model(x_2, sigma_fn(s) * self.s_in, **self.extra_args)