fix for self conditioning in diffusion prior network https://github.com/lucidrains/DALLE2-pytorch/issues/273

dalle2_pytorch/dalle2_pytorch.py

@@ -1124,7 +1124,7 @@ class DiffusionPriorNetwork(nn.Module):
         learned_queries = repeat(self.learned_query, 'd -> b 1 d', b = batch)
 
         if self.self_cond:
-            learned_queries = torch.cat((image_embed, self_cond), dim = -2)
+            learned_queries = torch.cat((self_cond, learned_queries), dim = -2)
 
         tokens = torch.cat((
             text_encodings,

dalle2_pytorch/version.py

@@ -1 +1 @@
-__version__ = '1.12.0'
+__version__ = '1.12.1'