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

fix bug with misnamed variable in diffusion prior network

Browse Source
This commit is contained in:
Phil Wang
2022-08-31 17:18:58 -07:00
parent 22019fddeb
commit 8bbc956ff1
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

@@ -1070,7 +1070,7 @@ class DiffusionPriorNetwork(nn.Module):
null_text_embeds = self.null_text_embeds.to(text_embed.dtype) null_text_embeds = self.null_text_embeds.to(text_embed.dtype)
text_embeds = torch.where( text_embed = torch.where(
text_keep_mask, text_keep_mask,
text_embed, text_embed,
null_text_embeds null_text_embeds

2
dalle2_pytorch/version.py
View File

@@ -1 +1 @@
__version__ = '1.10.3' __version__ = '1.10.4'