Update pipeline.py
Browse files- 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, :, :, :]
|