From b693e0be034b70ab81f0086d465151074612b900 Mon Sep 17 00:00:00 2001 From: Phil Wang Date: Mon, 30 May 2022 10:06:48 -0700 Subject: [PATCH] default number of resnet blocks per layer in unet to 2 (in imagen it was 3 for base 64x64) --- dalle2_pytorch/dalle2_pytorch.py | 2 +- setup.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/dalle2_pytorch/dalle2_pytorch.py b/dalle2_pytorch/dalle2_pytorch.py index 0bb86b5..a26daf1 100644 --- a/dalle2_pytorch/dalle2_pytorch.py +++ b/dalle2_pytorch/dalle2_pytorch.py @@ -1347,7 +1347,7 @@ class Unet(nn.Module): init_dim = None, init_conv_kernel_size = 7, resnet_groups = 8, - num_resnet_blocks = 1, + num_resnet_blocks = 2, init_cross_embed_kernel_sizes = (3, 7, 15), cross_embed_downsample = False, cross_embed_downsample_kernel_sizes = (2, 4), diff --git a/setup.py b/setup.py index fb2deb3..fc32ff8 100644 --- a/setup.py +++ b/setup.py @@ -10,7 +10,7 @@ setup( 'dream = dalle2_pytorch.cli:dream' ], }, - version = '0.6.0', + version = '0.6.1', license='MIT', description = 'DALL-E 2', author = 'Phil Wang',