bump

Author: lucidrains

dalle_pytorch/dalle_pytorch.py

@@ -74,7 +74,7 @@ def __init__(
         channels = 3,
         temperature = 0.9,
         straight_through = False,
-        kl_div_loss_weight = 1.
+        kl_div_loss_weight = 0.1
     ):
         super().__init__()
         assert log2(image_size).is_integer(), 'image size must be a power of 2'

setup.py

@@ -3,7 +3,7 @@
 setup(
   name = 'dalle-pytorch',
   packages = find_packages(),
-  version = '0.1.5',
+  version = '0.1.6',
   license='MIT',
   description = 'DALL-E - Pytorch',
   author = 'Phil Wang',