address #51

lucidrains · lucidrains · commit 3df2a5e23ee6 · 2023-11-06T09:54:00.000-08:00
diff --git a/gigagan_pytorch/unet_upsampler.py b/gigagan_pytorch/unet_upsampler.py
@@ -399,7 +399,6 @@ def __init__(
         num_resolutions = len(in_out)
 
         for ind, ((dim_in, dim_out), layer_full_attn, layer_cross_attn, layer_attn_depth) in enumerate(zip(in_out, full_attn, cross_attn, attn_depths)):
-            ind >= (num_resolutions - 1)
 
             should_not_downsample = ind < num_layer_no_downsample
             has_cross_attn = not self.unconditional and layer_cross_attn
diff --git a/gigagan_pytorch/version.py b/gigagan_pytorch/version.py
@@ -1 +1 @@
-__version__ = '0.2.19'
+__version__ = '0.2.20'
