From 9bfb5a8c9b97709e8f4fa4ab957eb958e70c467e Mon Sep 17 00:00:00 2001 From: Kosinkadink Date: Thu, 19 Sep 2024 07:50:17 +0900 Subject: [PATCH] Fix print_schedule throwing error in cases where prompt schedule is extending prompts after the last provided index --- animatediff/scheduling.py | 1 + 1 file changed, 1 insertion(+) diff --git a/animatediff/scheduling.py b/animatediff/scheduling.py index 2c4e9cb..b223d34 100644 --- a/animatediff/scheduling.py +++ b/animatediff/scheduling.py @@ -372,6 +372,7 @@ def handle_prompt_interpolation(pairs: list[InputPair], length: int, clip: CLIP, prev_holder = CondHolder(idx=i, prompt=real_prompt, raw_prompt=prev_holder.raw_prompt, cond=cond, pooled=pooled, hold=prev_holder.hold) real_cond[i] = prev_holder.cond real_pooled[i] = prev_holder.pooled + real_holders[i] = prev_holder else: prev_holder = real_holders[i] pbar.update(1)