back to no_grad for now, also keep track and restore unet devices in one_unet_in_gpu contextmanager

This commit is contained in:
Phil Wang
2022-05-16 09:36:14 -07:00
parent bb151ca6b1
commit dab106d4e5
3 changed files with 18 additions and 14 deletions

View File

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