[CLIPGuidedStableDiffusion] take the correct text embeddings (#667)

patil-suraj · web-flow · commit c16761e9d94a · 2022-09-28T17:41:34.000+02:00
take the correct text embeddings
diff --git a/examples/community/clip_guided_stable_diffusion.py b/examples/community/clip_guided_stable_diffusion.py
@@ -284,7 +284,7 @@ def __call__(
             # perform clip guidance
             if clip_guidance_scale > 0:
                 text_embeddings_for_guidance = (
-                    text_embeddings.chunk(2)[0] if do_classifier_free_guidance else text_embeddings
+                    text_embeddings.chunk(2)[1] if do_classifier_free_guidance else text_embeddings
                 )
                 noise_pred, latents = self.cond_fn(
                     latents,

Original file line number	Diff line number	Diff line change
`@@ -284,7 +284,7 @@ def __call__(`
`284`	`284`	`# perform clip guidance`
`285`	`285`	`if clip_guidance_scale > 0:`
`286`	`286`	`text_embeddings_for_guidance = (`
`287`		`- text_embeddings.chunk(2)[0] if do_classifier_free_guidance else text_embeddings`
	`287`	`+ text_embeddings.chunk(2)[1] if do_classifier_free_guidance else text_embeddings`
`288`	`288`	`)`
`289`	`289`	`noise_pred, latents = self.cond_fn(`
`290`	`290`	`latents,`