Commit a0a51eb

authored

Kandinsky5 No cfg fix (#12527)

fix

1 parent a5a0ccf commit a0a51ebCopy full SHA for a0a51eb

File tree

-7

lines changed

-7

lines changed

Lines changed: 9 additions & 7 deletions

Original file line number	Diff line number	Diff line change
`@@ -744,11 +744,13 @@ def __call__(`
`744`	`744`	`)`
`745`	`745`
`746`	`746`	`if negative_prompt_embeds_qwen is None:`
`747`		`- negative_prompt_embeds_qwen, negative_prompt_embeds_clip, negative_cu_seqlens = self.encode_prompt(`
`748`		`- prompt=negative_prompt,`
`749`		`- max_sequence_length=max_sequence_length,`
`750`		`- device=device,`
`751`		`- dtype=dtype,`
	`747`	`+ negative_prompt_embeds_qwen, negative_prompt_embeds_clip, negative_prompt_cu_seqlens = (`
	`748`	`+ self.encode_prompt(`
	`749`	`+ prompt=negative_prompt,`
	`750`	`+ max_sequence_length=max_sequence_length,`
	`751`	`+ device=device,`
	`752`	`+ dtype=dtype,`
	`753`	`+ )`
`752`	`754`	`)`
`753`	`755`
`754`	`756`	`# 4. Prepare timesteps`
`@@ -780,8 +782,8 @@ def __call__(`
`780`	`782`	`text_rope_pos = torch.arange(prompt_cu_seqlens.diff().max().item(), device=device)`
`781`	`783`
`782`	`784`	`negative_text_rope_pos = (`
`783`		`- torch.arange(negative_cu_seqlens.diff().max().item(), device=device)`
`784`		`- if negative_cu_seqlens is not None`
	`785`	`+ torch.arange(negative_prompt_cu_seqlens.diff().max().item(), device=device)`
	`786`	`+ if negative_prompt_cu_seqlens is not None`
`785`	`787`	`else None`
`786`	`788`	`)`
`787`	`789`

Comments

(0)