aljaz/DALLE2-pytorch
1
1
Fork 0
mirror of https://github.com/lucidrains/DALLE2-pytorch.git synced 2025-12-24 03:54:19 +01:00
Code Issues Packages Projects Releases Wiki Activity

use some magic just this once to remove the need for researchers to think

Browse Source
This commit is contained in:
Phil Wang
2022-04-18 12:40:38 -07:00
parent 7214df472d
commit 960a79857b
3 changed files with 27 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
README.md
View File

@@ -218,7 +218,6 @@ unet1 = Unet(
unet2 = Unet(
dim = 16,
image_embed_dim = 512,
lowres_cond = True, # subsequent unets must have this turned on (and first unet must have this turned off)
cond_dim = 128,
channels = 3,
dim_mults = (1, 2, 4, 8, 16)
@@ -349,8 +348,7 @@ unet2 = Unet(
image_embed_dim = 512,
cond_dim = 128,
channels = 3,
dim_mults = (1, 2, 4, 8, 16),
lowres_cond = True
dim_mults = (1, 2, 4, 8, 16)
).cuda()
decoder = Decoder(