address #51

lucidrains · Nov 6, 2023 · 3df2a5e · 3df2a5e
1 parent fd6b5c3
commit 3df2a5e
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
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'