diff --git a/dalle2_pytorch/dalle2_pytorch.py b/dalle2_pytorch/dalle2_pytorch.py index 17384d9..ab18a19 100644 --- a/dalle2_pytorch/dalle2_pytorch.py +++ b/dalle2_pytorch/dalle2_pytorch.py @@ -1075,14 +1075,14 @@ class LowresConditioner(nn.Module): if self.training and self.downsample_first and exists(downsample_image_size): cond_fmap = resize_image_to(cond_fmap, target_image_size, mode = self.cond_upsample_mode) - cond_fmap = resize_image_to(cond_fmap, target_image_size, mode = self.cond_upsample_mode) - if self.training: # when training, blur the low resolution conditional image blur_sigma = default(blur_sigma, self.blur_sigma) blur_kernel_size = default(blur_kernel_size, self.blur_kernel_size) cond_fmap = gaussian_blur2d(cond_fmap, cast_tuple(blur_kernel_size, 2), cast_tuple(blur_sigma, 2)) + cond_fmap = resize_image_to(cond_fmap, target_image_size, mode = self.cond_upsample_mode) + return cond_fmap class Decoder(nn.Module): diff --git a/setup.py b/setup.py index 406d421..71a3e49 100644 --- a/setup.py +++ b/setup.py @@ -10,7 +10,7 @@ setup( 'dream = dalle2_pytorch.cli:dream' ], }, - version = '0.0.32', + version = '0.0.33', license='MIT', description = 'DALL-E 2', author = 'Phil Wang',