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

makes more sense to keep this as True as default, for stability

Browse Source
This commit is contained in:
Phil Wang
2022-05-02 10:50:55 -07:00
parent 2d25c89f35
commit 11469dc0c6
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
dalle2_pytorch/dalle2_pytorch.py
View File

@@ -599,7 +599,7 @@ class CausalTransformer(nn.Module):
dim_head = 64, dim_head = 64,
heads = 8, heads = 8,
ff_mult = 4, ff_mult = 4,
norm_out = False, norm_out = True,
attn_dropout = 0., attn_dropout = 0.,
ff_dropout = 0., ff_dropout = 0.,
final_proj = True final_proj = True

2
setup.py
View File

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