smoothieAI commited on
Commit
041b0a3
·
verified ·
1 Parent(s): fae6ec5

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +2 -1
pipeline.py CHANGED
@@ -1118,6 +1118,7 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
1118
  elif not isinstance(control_guidance_end, list) and isinstance(control_guidance_start, list):
1119
  control_guidance_end = len(control_guidance_start) * [control_guidance_end]
1120
  elif not isinstance(control_guidance_start, list) and not isinstance(control_guidance_end, list):
 
1121
  mult = len(controlnet.nets) if isinstance(controlnet, MultiControlNetModel) else 1
1122
  control_guidance_start, control_guidance_end = (
1123
  mult * [control_guidance_start],
@@ -1426,7 +1427,7 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
1426
  latent_model_input = self.scheduler.scale_model_input(latent_model_input, t)
1427
 
1428
 
1429
- if self.controlnet != None and i < int(control_guidance_end*len(timesteps)):
1430
  contorl_start = time.time()
1431
 
1432
  current_context_conditioning_frames = conditioning_frames[current_context_indexes, :, :, :]
 
1118
  elif not isinstance(control_guidance_end, list) and isinstance(control_guidance_start, list):
1119
  control_guidance_end = len(control_guidance_start) * [control_guidance_end]
1120
  elif not isinstance(control_guidance_start, list) and not isinstance(control_guidance_end, list):
1121
+ control_end = control_guidance_end
1122
  mult = len(controlnet.nets) if isinstance(controlnet, MultiControlNetModel) else 1
1123
  control_guidance_start, control_guidance_end = (
1124
  mult * [control_guidance_start],
 
1427
  latent_model_input = self.scheduler.scale_model_input(latent_model_input, t)
1428
 
1429
 
1430
+ if self.controlnet != None and i < int(control_end*len(timesteps)):
1431
  contorl_start = time.time()
1432
 
1433
  current_context_conditioning_frames = conditioning_frames[current_context_indexes, :, :, :]