smoothieAI
/

pipeline_animatediff_context_controlnet_v2

Model card Files Files and versions Community

smoothieAI commited on Feb 8, 2024

Commit

d7f1509

·

verified ·

1 Parent(s): b29f051

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -1189,7 +1189,7 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
             # to avoid doing two forward passes
             if do_classifier_free_guidance:
                 # concatenate negative prompt embeddings with prompt embeddings on a new dimension after the first batch dimension
-                single_prompt_embeds = torch.stack([single_negative_prompt_embeds, single_prompt_embeds], dim=0)
             prompt_embeds_list.append(single_prompt_embeds)
             # print shape of single prompt embeds

             # to avoid doing two forward passes
             if do_classifier_free_guidance:
                 # concatenate negative prompt embeddings with prompt embeddings on a new dimension after the first batch dimension
+                single_prompt_embeds = torch.stack([single_negative_prompt_embeds, single_prompt_embeds])
             prompt_embeds_list.append(single_prompt_embeds)
             # print shape of single prompt embeds