make sure learned variance can work for any number of unets in the decoder, defaults to first unet, as suggested was used in the paper

This commit is contained in:
Phil Wang
2022-05-12 14:18:15 -07:00
parent 28b58e568c
commit 2277b47ffd
2 changed files with 130 additions and 19 deletions

View File

@@ -10,7 +10,7 @@ setup(
'dream = dalle2_pytorch.cli:dream'
],
},
version = '0.2.12',
version = '0.2.14',
license='MIT',
description = 'DALL-E 2',
author = 'Phil Wang',