get image_size from RIN

Author: lucidrains

rin_pytorch/rin_pytorch.py

@@ -349,6 +349,7 @@ def __init__(
         assert divisible_by(image_size, patch_size)
         dim_latent = default(dim_latent, dim)
 
+        self.image_size = image_size
         self.channels = channels # times 2 due to self-conditioning
 
         patch_height_width = image_size // patch_size
@@ -508,7 +509,6 @@ def __init__(
         self,
         model: RIN,
         *,
-        image_size,
         timesteps = 1000,
         use_ddim = True,
         noise_schedule = 'sigmoid',
@@ -525,7 +525,7 @@ def __init__(
         assert objective in {'x0', 'eps'}, 'objective must be either predict x0 or noise'
         self.objective = objective
 
-        self.image_size = image_size
+        self.image_size = model.image_size
 
         if noise_schedule == "linear":
             self.gamma_schedule = simple_linear_schedule

setup.py

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