make sure vqgan-vae trainer supports mixed precision

This commit is contained in:
Phil Wang
2022-05-06 10:44:16 -07:00
parent 28e944f328
commit 3676ef4d49
2 changed files with 29 additions and 18 deletions

View File

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