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

prep for classifier free guidance for the image embedding diffusion step, even though not mentioned in paper

Browse Source
This commit is contained in:
Phil Wang
2022-04-12 12:57:09 -07:00
parent 59b8abe09e
commit 7647be2569
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
dalle2_pytorch/dalle2_pytorch.py
View File

@@ -169,6 +169,7 @@ class DiffusionPriorNetwork(nn.Module):
text_encodings, text_encodings,
text_embed, text_embed,
mask = None, mask = None,
cond_drop_prob = 0.2
): ):
batch = image_embed.shape[0] batch = image_embed.shape[0]