smoothieAI commited on
Commit
85420e0
·
verified ·
1 Parent(s): 6a56dfe

Update pipeline.py

Browse files
Files changed (1) hide show
  1. 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