From 7647be25698879afe70654b241ff1bc4db671b0b Mon Sep 17 00:00:00 2001 From: Phil Wang Date: Tue, 12 Apr 2022 12:57:09 -0700 Subject: [PATCH] prep for classifier free guidance for the image embedding diffusion step, even though not mentioned in paper --- dalle2_pytorch/dalle2_pytorch.py | 1 + 1 file changed, 1 insertion(+) diff --git a/dalle2_pytorch/dalle2_pytorch.py b/dalle2_pytorch/dalle2_pytorch.py index fad469a..fb6770d 100644 --- a/dalle2_pytorch/dalle2_pytorch.py +++ b/dalle2_pytorch/dalle2_pytorch.py @@ -169,6 +169,7 @@ class DiffusionPriorNetwork(nn.Module): text_encodings, text_embed, mask = None, + cond_drop_prob = 0.2 ): batch = image_embed.shape[0]