trust the conclusions of https://arxiv.org/abs/2208.03641

lucidrains · Nov 7, 2022 · f8cc75f · f8cc75f
1 parent 8ec3321
commit f8cc75f
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 2 deletions.
diff --git a/imagen_pytorch/imagen_video/imagen_video.py b/imagen_pytorch/imagen_video/imagen_video.py
@@ -564,7 +564,10 @@ def forward(self, x):
 
 def Downsample(dim, dim_out = None):
     dim_out = default(dim_out, dim)
-    return Conv2d(dim, dim_out, 4, 2, 1)
+    return nn.Sequential(
+        Rearrange('b c f (h p1) (w p2) -> b (c p1 p2) f h w', p1 = 2, p2 = 2),
+        Conv2d(dim * 4, dim_out, 1)
+    )
 
 class SinusoidalPosEmb(nn.Module):
     def __init__(self, dim):

diff --git a/imagen_pytorch/version.py b/imagen_pytorch/version.py
@@ -1 +1 @@
-__version__ = '1.12.1'
+__version__ = '1.14.0'