smoothieAI commited on
Commit
98ff210
·
verified ·
1 Parent(s): 54ebd00

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +2 -2
pipeline.py CHANGED
@@ -1447,10 +1447,10 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
1447
  # Infer ControlNet only for the conditional batch.
1448
  control_model_input = latents
1449
  control_model_input = self.scheduler.scale_model_input(control_model_input, t)
1450
- controlnet_prompt_embeds = prompt_embeds[current_prompt_index].chunk(2)[1]
1451
  else:
1452
  control_model_input = latent_model_input
1453
- controlnet_prompt_embeds = prompt_embeds[current_prompt_index]
1454
  controlnet_prompt_embeds = controlnet_prompt_embeds.repeat_interleave(len(current_context_indexes), dim=0)
1455
 
1456
  if isinstance(controlnet_keep[i], list):
 
1447
  # Infer ControlNet only for the conditional batch.
1448
  control_model_input = latents
1449
  control_model_input = self.scheduler.scale_model_input(control_model_input, t)
1450
+ controlnet_prompt_embeds = prompt_embeds_list[current_prompt_index].chunk(2)[1]
1451
  else:
1452
  control_model_input = latent_model_input
1453
+ controlnet_prompt_embeds = prompt_embeds_list[current_prompt_index]
1454
  controlnet_prompt_embeds = controlnet_prompt_embeds.repeat_interleave(len(current_context_indexes), dim=0)
1455
 
1456
  if isinstance(controlnet_keep[i], list):