Fix Scheduling

This commit is contained in:
AI-Casanova
2024-01-13 14:52:55 -06:00
parent 1edecc2389
commit cc2e3e058b
2 changed files with 7 additions and 5 deletions
+4
View File
@@ -85,6 +85,10 @@ def encode_prompts(pipe, p, prompts: list, negative_prompts: list, steps: int,
positive_schedule, scheduled = get_prompt_schedule(prompts[0], steps)
negative_schedule, neg_scheduled = get_prompt_schedule(negative_prompts[0], steps)
p.scheduled_prompt = scheduled or neg_scheduled
p.prompt_embeds = []
p.positive_pooleds = []
p.negative_embeds = []
p.negative_pooleds = []
cache = {}
for i in range(max(len(positive_schedule), len(negative_schedule))):