smoothieAI
/

pipeline_animatediff_context_controlnet_v2

Model card Files Files and versions Community

smoothieAI commited on Feb 8, 2024

Commit

e68fe56

·

verified ·

1 Parent(s): c96e681

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -1191,7 +1191,7 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
                 # concatenate negative prompt embeddings with prompt embeddings on a new dimension after the first batch dimension
                 # print shape of single_prompt_embeds
                 print("single prompt embes shape ",single_prompt_embeds.shape)
-                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

                 # concatenate negative prompt embeddings with prompt embeddings on a new dimension after the first batch dimension
                 # print shape of single_prompt_embeds
                 print("single prompt embes shape ",single_prompt_embeds.shape)
+                single_prompt_embeds = torch.cat([single_negative_prompt_embeds, single_prompt_embeds])
             prompt_embeds_list.append(single_prompt_embeds)
             # print shape of single prompt embeds