From 291377bb9c3c2528cd7a7de1b1a6bd8a9316ba8e Mon Sep 17 00:00:00 2001 From: Phil Wang Date: Tue, 19 Jul 2022 11:31:56 -0700 Subject: [PATCH] @jacobwjs reports dynamic thresholding works very well and 0.95 is a better value --- dalle2_pytorch/dalle2_pytorch.py | 2 +- dalle2_pytorch/version.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/dalle2_pytorch/dalle2_pytorch.py b/dalle2_pytorch/dalle2_pytorch.py index d6cd618..b23a652 100644 --- a/dalle2_pytorch/dalle2_pytorch.py +++ b/dalle2_pytorch/dalle2_pytorch.py @@ -2115,7 +2115,7 @@ class Decoder(nn.Module): unconditional = False, # set to True for generating images without conditioning auto_normalize_img = True, # whether to take care of normalizing the image from [0, 1] to [-1, 1] and back automatically - you can turn this off if you want to pass in the [-1, 1] ranged image yourself from the dataloader use_dynamic_thres = False, # from the Imagen paper - dynamic_thres_percentile = 0.9, + dynamic_thres_percentile = 0.95, p2_loss_weight_gamma = 0., # p2 loss weight, from https://arxiv.org/abs/2204.00227 - 0 is equivalent to weight of 1 across time - 1. is recommended p2_loss_weight_k = 1, ddim_sampling_eta = 1. # can be set to 0. for deterministic sampling afaict diff --git a/dalle2_pytorch/version.py b/dalle2_pytorch/version.py index 826d20e..f96e1c6 100644 --- a/dalle2_pytorch/version.py +++ b/dalle2_pytorch/version.py @@ -1 +1 @@ -__version__ = '0.26.0' +__version__ = '0.26.1'