smoothieAI commited on
Commit
f788402
·
verified ·
1 Parent(s): 4288e03

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +2 -3
pipeline.py CHANGED
@@ -1259,9 +1259,8 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
1259
  if self.controlnet != None:
1260
  # if we are using multiple controlnets, select the context window for each controlnet
1261
  if isinstance(controlnet, MultiControlNetModel):
1262
- current_context_conditioning_frames = []
1263
- for c in range(len(controlnet.nets)):
1264
- current_context_conditioning_frames[c] = conditioning_frames[c][current_context_indexes, :, :, :]
1265
  else:
1266
  # select the relevent context from the conditioning frames of shape (frame_number, channel, height, width)
1267
  current_context_conditioning_frames = conditioning_frames[current_context_indexes, :, :, :]
 
1259
  if self.controlnet != None:
1260
  # if we are using multiple controlnets, select the context window for each controlnet
1261
  if isinstance(controlnet, MultiControlNetModel):
1262
+ print("lengt of conditioning_frames", len(conditioning_frames))
1263
+ current_context_conditioning_frames = [conditioning_frames[c][current_context_indexes, :, :, :] for c in range(len(controlnet.nets))]
 
1264
  else:
1265
  # select the relevent context from the conditioning frames of shape (frame_number, channel, height, width)
1266
  current_context_conditioning_frames = conditioning_frames[current_context_indexes, :, :, :]