Commit3b2cf7b

committed

fix for self conditioning in diffusion prior networklucidrains#273

1 parent984d62a commit3b2cf7bCopy full SHA for 3b2cf7b

File tree

-2

lines changed

-2

lines changed

Lines changed: 1 addition & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -1124,7 +1124,7 @@ def forward(`
`1124`	`1124`	`learned_queries=repeat(self.learned_query,'d -> b 1 d',b=batch)`
`1125`	`1125`
`1126`	`1126`	`ifself.self_cond:`
`1127`		`-learned_queries=torch.cat((image_embed,self_cond),dim=-2)`
	`1127`	`+learned_queries=torch.cat((self_cond,learned_queries),dim=-2)`
`1128`	`1128`
`1129`	`1129`	`tokens=torch.cat((`
`1130`	`1130`	`text_encodings,`

Lines changed: 1 addition & 1 deletion

Comments

(0)