address #19

Author: lucidrains

rin_pytorch/rin_pytorch.py

@@ -700,8 +700,8 @@ def ddpm_sample(self, shape, time_difference = None):
 
             # get alpha sigma of time and next time
 
-            alpha, sigma = gamma_to_alpha_sigma(gamma, self.scale)
-            alpha_next, sigma_next = gamma_to_alpha_sigma(gamma_next, self.scale)
+            alpha, sigma = gamma_to_alpha_sigma(gamma)
+            alpha_next, sigma_next = gamma_to_alpha_sigma(gamma_next)
 
             # calculate x0 and noise
 
@@ -762,8 +762,8 @@ def ddim_sample(self, shape, time_difference = None):
 
             padded_gamma, padded_gamma_next = map(partial(right_pad_dims_to, img), (gamma, gamma_next))
 
-            alpha, sigma = gamma_to_alpha_sigma(padded_gamma, self.scale)
-            alpha_next, sigma_next = gamma_to_alpha_sigma(padded_gamma_next, self.scale)
+            alpha, sigma = gamma_to_alpha_sigma(padded_gamma)
+            alpha_next, sigma_next = gamma_to_alpha_sigma(padded_gamma_next)
 
             # add the time delay
 

setup.py

@@ -3,7 +3,7 @@
 setup(
   name = 'RIN-pytorch',
   packages = find_packages(exclude=[]),
-  version = '0.7.9',
+  version = '0.7.10',
   license='MIT',
   description = 'RIN - Recurrent Interface Network - Pytorch',
   author = 'Phil Wang',