address issue with resolutions in generator for calculating attention placement, thanks to @inspirit #44

Author: lucidrains

gigagan_pytorch/gigagan_pytorch.py

@@ -919,7 +919,7 @@ def __init__(
         num_layers = int(log2(image_size) - 1)
         self.num_layers = num_layers
 
-        resolutions = image_size / ((2 ** torch.arange(num_layers)))
+        resolutions = image_size / ((2 ** torch.arange(num_layers).flip(0)))
         resolutions = resolutions.long().tolist()
 
         dim_layers = (2 ** (torch.arange(num_layers) + 1)) * dim_capacity

gigagan_pytorch/version.py

@@ -1 +1 @@
-__version__ = '0.2.17'
+__version__ = '0.2.18'