Update pipeline.py
Browse files- 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)
|