smoothieAI commited on
Commit
6c0b030
·
verified ·
1 Parent(s): 13ad593

Update pipeline.py

Browse files
Files changed (1) hide show
  1. 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,