smoothieAI commited on
Commit
72f3d04
·
verified ·
1 Parent(s): 1d00245

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +2 -8
pipeline.py CHANGED
@@ -1287,19 +1287,13 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
1287
  controlnet_cond_scale = controlnet_cond_scale[0]
1288
  cond_scale = controlnet_cond_scale * controlnet_keep[i]
1289
 
1290
- print("-----------------------")
1291
- print("control_model_input.shape", control_model_input.shape)
1292
 
1293
  control_model_input = torch.transpose(control_model_input, 1, 2)
1294
  control_model_input = control_model_input.reshape(
1295
  (-1, control_model_input.shape[2], control_model_input.shape[3], control_model_input.shape[4])
1296
  )
1297
- # print shapes of controlnet inputs and conditions
1298
- print("prompt_embeds.shape", prompt_embeds.shape)
1299
- print("control_model_input", control_model_input.shape)
1300
- print("controlnet_prompt_embeds", controlnet_prompt_embeds.shape)
1301
- print("current_context_conditioning_frames", current_context_conditioning_frames.shape)
1302
-
1303
  down_block_res_samples, mid_block_res_sample = self.controlnet(
1304
  control_model_input,
1305
  t,
 
1287
  controlnet_cond_scale = controlnet_cond_scale[0]
1288
  cond_scale = controlnet_cond_scale * controlnet_keep[i]
1289
 
 
 
1290
 
1291
  control_model_input = torch.transpose(control_model_input, 1, 2)
1292
  control_model_input = control_model_input.reshape(
1293
  (-1, control_model_input.shape[2], control_model_input.shape[3], control_model_input.shape[4])
1294
  )
1295
+
1296
+
 
 
 
 
1297
  down_block_res_samples, mid_block_res_sample = self.controlnet(
1298
  control_model_input,
1299
  t,