smoothieAI
/

pipeline_animatediff_context_controlnet

Model card Files Files and versions Community

smoothieAI commited on Jan 27, 2024

Commit

fae6ec5

·

verified ·

1 Parent(s): 2eb1fa8

Update pipeline.py

Files changed (1) hide show

pipeline.py +1 -1

pipeline.py CHANGED Viewed

@@ -1426,7 +1426,7 @@ class AnimateDiffPipeline(DiffusionPipeline, TextualInversionLoaderMixin, IPAdap
                     latent_model_input = self.scheduler.scale_model_input(latent_model_input, t)
-                    if self.controlnet != None and i < 4:
                         contorl_start = time.time()
                         current_context_conditioning_frames = conditioning_frames[current_context_indexes, :, :, :]

                     latent_model_input = self.scheduler.scale_model_input(latent_model_input, t)
+                    if self.controlnet != None and i < int(control_guidance_end*len(timesteps)):
                         contorl_start = time.time()
                         current_context_conditioning_frames = conditioning_frames[current_context_indexes, :, :, :]