Skip to content

GitLab

Unverified Commit c16761e9 authored by Suraj Patil's avatar Suraj Patil Committed by GitHub
Browse files

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

take the correct text embeddings
parent 7f31142c
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/community/clip_guided_stable_diffusion.py
View file @ c16761e9
...@@ -284,7 +284,7 @@ class CLIPGuidedStableDiffusion(DiffusionPipeline): ...@@ -284,7 +284,7 @@ class CLIPGuidedStableDiffusion(DiffusionPipeline):
# perform clip guidance # perform clip guidance
if clip_guidance_scale > 0: if clip_guidance_scale > 0:
text_embeddings_for_guidance = ( 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( noise_pred, latents = self.cond_fn(
latents, latents,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment