Update pipeline.py
Browse files- pipeline.py +1 -1
pipeline.py
CHANGED
@@ -412,7 +412,7 @@ class AnimateDiffControlNetPipeline(
|
|
412 |
[single_negative_image_embeds] * num_images_per_prompt, dim=0
|
413 |
)
|
414 |
|
415 |
-
if do_classifier_free_guidance:
|
416 |
single_image_embeds = torch.cat([single_negative_image_embeds, single_image_embeds])
|
417 |
single_image_embeds = single_image_embeds.to(device)
|
418 |
|
|
|
412 |
[single_negative_image_embeds] * num_images_per_prompt, dim=0
|
413 |
)
|
414 |
|
415 |
+
if self.do_classifier_free_guidance:
|
416 |
single_image_embeds = torch.cat([single_negative_image_embeds, single_image_embeds])
|
417 |
single_image_embeds = single_image_embeds.to(device)
|
418 |
|