smoothieAI commited on
Commit
028c64b
·
verified ·
1 Parent(s): 81a0727

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +5 -2
pipeline.py CHANGED
@@ -1311,9 +1311,11 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
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
  latent_model_input,
@@ -1324,6 +1326,7 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
1324
  guess_mode=guess_mode,
1325
  return_dict=False,
1326
  )
 
1327
  # predict the noise residual with the added controlnet residuals
1328
  noise_pred = self.unet(
1329
  latent_model_input,
 
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("shape of control_model_input", current_context_conditioning_frames.shape)
1315
  print("device of cond_scale", cond_scale.device)
1316
+ # print error
1317
+ except Exception as e:
1318
+ print("error", e)
1319
 
1320
  down_block_res_samples, mid_block_res_sample = self.controlnet(
1321
  latent_model_input,
 
1326
  guess_mode=guess_mode,
1327
  return_dict=False,
1328
  )
1329
+
1330
  # predict the noise residual with the added controlnet residuals
1331
  noise_pred = self.unet(
1332
  latent_model_input,