Update pipeline.py
Browse files- 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(
|
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, :, :, :]
|