smoothieAI commited on
Commit
c96e681
·
verified ·
1 Parent(s): d7f1509

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +2 -0
pipeline.py CHANGED
@@ -1189,6 +1189,8 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
1189
  # to avoid doing two forward passes
1190
  if do_classifier_free_guidance:
1191
  # concatenate negative prompt embeddings with prompt embeddings on a new dimension after the first batch dimension
 
 
1192
  single_prompt_embeds = torch.stack([single_negative_prompt_embeds, single_prompt_embeds])
1193
 
1194
  prompt_embeds_list.append(single_prompt_embeds)
 
1189
  # to avoid doing two forward passes
1190
  if do_classifier_free_guidance:
1191
  # concatenate negative prompt embeddings with prompt embeddings on a new dimension after the first batch dimension
1192
+ # print shape of single_prompt_embeds
1193
+ print("single prompt embes shape ",single_prompt_embeds.shape)
1194
  single_prompt_embeds = torch.stack([single_negative_prompt_embeds, single_prompt_embeds])
1195
 
1196
  prompt_embeds_list.append(single_prompt_embeds)