From 981d407792e26cabc348363f1d38b562588d6673 Mon Sep 17 00:00:00 2001 From: Phil Wang Date: Fri, 12 Aug 2022 11:41:23 -0700 Subject: [PATCH] comment --- dalle2_pytorch/dalle2_pytorch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dalle2_pytorch/dalle2_pytorch.py b/dalle2_pytorch/dalle2_pytorch.py index 31ce157..5375a24 100644 --- a/dalle2_pytorch/dalle2_pytorch.py +++ b/dalle2_pytorch/dalle2_pytorch.py @@ -1700,7 +1700,7 @@ class Unet(nn.Module): attn_heads = 16, lowres_cond = False, # for cascading diffusion - https://cascaded-diffusion.github.io/ lowres_noise_cond = False, # for conditioning on low resolution noising, based on Imagen - self_cond = False, + self_cond = False, # set this to True to use the self-conditioning technique from - https://arxiv.org/abs/2208.04202 sparse_attn = False, cosine_sim_cross_attn = False, cosine_sim_self_attn = False,