Update pipeline.py
Browse files- pipeline.py +9 -1
pipeline.py
CHANGED
@@ -1306,7 +1306,15 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
|
|
1306 |
control_model_input = control_model_input.reshape(
|
1307 |
(-1, control_model_input.shape[2], control_model_input.shape[3], control_model_input.shape[4])
|
1308 |
)
|
1309 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1310 |
down_block_res_samples, mid_block_res_sample = self.controlnet(
|
1311 |
control_model_input,
|
1312 |
t,
|
|
|
1306 |
control_model_input = control_model_input.reshape(
|
1307 |
(-1, control_model_input.shape[2], control_model_input.shape[3], control_model_input.shape[4])
|
1308 |
)
|
1309 |
+
# print device for all inputs
|
1310 |
+
try:
|
1311 |
+
print("device of control_model_input", control_model_input.device)
|
1312 |
+
print("device of controlnet_prompt_embeds", controlnet_prompt_embeds.device)
|
1313 |
+
print("device of current_context_conditioning_frames", current_context_conditioning_frames.device)
|
1314 |
+
print("device of cond_scale", cond_scale.device)
|
1315 |
+
except:
|
1316 |
+
pass
|
1317 |
+
|
1318 |
down_block_res_samples, mid_block_res_sample = self.controlnet(
|
1319 |
control_model_input,
|
1320 |
t,
|