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