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