smoothieAI commited on
Commit
f30fb3e
·
verified ·
1 Parent(s): d6564d9

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +2 -0
pipeline.py CHANGED
@@ -1261,6 +1261,8 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
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, :, :, :]
 
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
+ # move to device
1265
+ current_context_conditioning_frames = [c.to(device) for c in current_context_conditioning_frames]
1266
  else:
1267
  # select the relevent context from the conditioning frames of shape (frame_number, channel, height, width)
1268
  current_context_conditioning_frames = conditioning_frames[current_context_indexes, :, :, :]