Commit 79d4e819 authored by AUTOMATIC1111's avatar AUTOMATIC1111

fix processing error that happens if batch_size is not a multiple of how many...

fix processing error that happens if batch_size is not a multiple of how many prompts/negative prompts there are #12509
parent 7e77a38c
......@@ -382,13 +382,18 @@ class StableDiffusionProcessing:
def setup_prompts(self):
if type(self.prompt) == list:
self.all_prompts = self.prompt
elif type(self.negative_prompt) == list:
self.all_prompts = [self.prompt] * len(self.negative_prompt)
else:
self.all_prompts = self.batch_size * self.n_iter * [self.prompt]
if type(self.negative_prompt) == list:
self.all_negative_prompts = self.negative_prompt
else:
self.all_negative_prompts = self.batch_size * self.n_iter * [self.negative_prompt]
self.all_negative_prompts = [self.negative_prompt] * len(self.all_prompts)
if len(self.all_prompts) != len(self.all_negative_prompts):
raise RuntimeError(f"Received a different number of prompts ({len(self.all_prompts)}) and negative prompts ({len(self.all_negative_prompts)})")
self.all_prompts = [shared.prompt_styles.apply_styles_to_prompt(x, self.styles) for x in self.all_prompts]
self.all_negative_prompts = [shared.prompt_styles.apply_negative_styles_to_prompt(x, self.styles) for x in self.all_negative_prompts]
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment