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