backwards pass is not recommended under the autocast context, per pytorch docs

This commit is contained in:
Phil Wang
2022-05-14 18:26:19 -07:00
parent aee92dba4a
commit 4ec6d0ba81
2 changed files with 5 additions and 5 deletions

View File

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