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

nevermind, do not enforce text encodings on first unet

Browse Source
This commit is contained in:
Phil Wang
2022-06-26 12:45:05 -07:00
parent 2e85e736f3
commit 032e83b0e0
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

@@ -1852,7 +1852,7 @@ class Decoder(nn.Module):
one_unet = one_unet.cast_model_parameters( one_unet = one_unet.cast_model_parameters(
lowres_cond = not is_first, lowres_cond = not is_first,
cond_on_image_embeds = not unconditional and is_first, cond_on_image_embeds = not unconditional and is_first,
cond_on_text_encodings = not unconditional and (is_first or one_unet.cond_on_text_encodings), cond_on_text_encodings = not unconditional and one_unet.cond_on_text_encodings,
channels = unet_channels, channels = unet_channels,
channels_out = unet_channels_out channels_out = unet_channels_out
) )

2
dalle2_pytorch/version.py
View File

@@ -1 +1 @@
__version__ = '0.12.0' __version__ = '0.12.1'