address lucidrains#292

siddancha · Feb 15, 2024 · 7a77b45 · 7a77b45
1 parent 840d3ff
commit 7a77b45
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/denoising_diffusion_pytorch/learned_gaussian_diffusion.py b/denoising_diffusion_pytorch/learned_gaussian_diffusion.py
@@ -81,7 +81,7 @@ def __init__(
 
         self.vb_loss_weight = vb_loss_weight
 
-    def model_predictions(self, x, t, clip_x_start = False):
+    def model_predictions(self, x, t, x_self_cond = None, clip_x_start = False, rederive_pred_noise = False):
         model_output = self.model(x, t)
         model_output, pred_variance = model_output.chunk(2, dim = 1)
 

diff --git a/denoising_diffusion_pytorch/version.py b/denoising_diffusion_pytorch/version.py
@@ -1 +1 @@
-__version__ = '1.10.9'
+__version__ = '1.10.10'