diff --git a/dalle2_pytorch/dalle2_pytorch.py b/dalle2_pytorch/dalle2_pytorch.py index 6c8984f..9524eac 100644 --- a/dalle2_pytorch/dalle2_pytorch.py +++ b/dalle2_pytorch/dalle2_pytorch.py @@ -2496,7 +2496,7 @@ class Decoder(nn.Module): 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 + ddim_sampling_eta = 0. # can be set to 0. for deterministic sampling afaict ): super().__init__() diff --git a/dalle2_pytorch/version.py b/dalle2_pytorch/version.py index 2279cc1..666b2f7 100644 --- a/dalle2_pytorch/version.py +++ b/dalle2_pytorch/version.py @@ -1 +1 @@ -__version__ = '1.11.4' +__version__ = '1.12.0'