smoothieAI commited on
Commit
13ad593
·
verified ·
1 Parent(s): 15d0b5d

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +4 -0
pipeline.py CHANGED
@@ -1266,6 +1266,10 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
1266
  current_context_conditioning_frames = [conditioning_frames[c][current_context_indexes, :, :, :] for c in range(len(controlnet.nets))]
1267
  # move to device
1268
  current_context_conditioning_frames = [c.to(device) for c in current_context_conditioning_frames]
 
 
 
 
1269
  else:
1270
  # select the relevent context from the conditioning frames of shape (frame_number, channel, height, width)
1271
  current_context_conditioning_frames = conditioning_frames[current_context_indexes, :, :, :]
 
1266
  current_context_conditioning_frames = [conditioning_frames[c][current_context_indexes, :, :, :] for c in range(len(controlnet.nets))]
1267
  # move to device
1268
  current_context_conditioning_frames = [c.to(device) for c in current_context_conditioning_frames]
1269
+ # print shape of curent context conditioning frames [0]
1270
+ print("shape of current context conditioning frames", current_context_conditioning_frames[0].shape)
1271
+ # print device
1272
+ print("device of current context conditioning frames", current_context_conditioning_frames[0].device)
1273
  else:
1274
  # select the relevent context from the conditioning frames of shape (frame_number, channel, height, width)
1275
  current_context_conditioning_frames = conditioning_frames[current_context_indexes, :, :, :]