Update pipeline.py
Browse files- pipeline.py +2 -1
pipeline.py
CHANGED
@@ -1427,7 +1427,8 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
|
|
1427 |
latent_model_input = torch.cat([current_context_latents] * 2) if do_classifier_free_guidance else current_context_latents
|
1428 |
latent_model_input = self.scheduler.scale_model_input(latent_model_input, t)
|
1429 |
|
1430 |
-
|
|
|
1431 |
if self.controlnet != None and i < int(control_end*num_inference_steps):
|
1432 |
|
1433 |
torch.cuda.synchronize() # Synchronize GPU
|
|
|
1427 |
latent_model_input = torch.cat([current_context_latents] * 2) if do_classifier_free_guidance else current_context_latents
|
1428 |
latent_model_input = self.scheduler.scale_model_input(latent_model_input, t)
|
1429 |
|
1430 |
+
control_end_step = int(control_end*num_inference_steps)
|
1431 |
+
print("control_end_step", control_end_step)
|
1432 |
if self.controlnet != None and i < int(control_end*num_inference_steps):
|
1433 |
|
1434 |
torch.cuda.synchronize() # Synchronize GPU
|