aljaz/DALLE2-pytorch
1
1
Fork 0
mirror of https://github.com/lucidrains/DALLE2-pytorch.git synced 2026-02-15 03:24:27 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix decoder trainer optimizer loading (since there are multiple for each unet), also save and load step number correctly

Browse Source
This commit is contained in:
Phil Wang
2022-05-22 15:21:00 -07:00
parent 271a376eaf
commit 49de72040c
2 changed files with 24 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
setup.py
View File

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