follow algorithm 2

lucidrains · Aug 18, 2022 · d03521c · d03521c
1 parent 6eaf639
commit d03521c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/bit_diffusion/bit_diffusion.py b/bit_diffusion/bit_diffusion.py
@@ -444,7 +444,7 @@ def ddpm_sample(self, shape):
 
             # add the time delay
 
-            time_next += self.sample_time_delay
+            time_next = F.relu(time_next - self.sample_time_delay)
 
             noise_cond = self.log_snr(time)
 
@@ -508,7 +508,7 @@ def ddim_sample(self, shape):
 
             # add the time delay
 
-            times_next += self.sample_time_delay
+            times_next = F.relu(times_next - self.sample_time_delay)
 
             x_start = self.model(img, log_snr, x_start)
 

diff --git a/setup.py b/setup.py
@@ -3,7 +3,7 @@
 setup(
   name = 'bit-diffusion',
   packages = find_packages(exclude=[]),
-  version = '0.0.7',
+  version = '0.0.9',
   license='MIT',
   description = 'Bit Diffusion - Pytorch',
   author = 'Phil Wang',