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

2025-12-19 17:54:20 +01:00 · 2022-04-12 12:57:09 -07:00
parent 59b8abe09e
commit 7647be2569
1 changed files with 1 additions and 0 deletions
--- 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]