Fix prompt matrix script

This commit is contained in:
Lukas Meller 2022-09-09 21:55:54 +02:00 committed by AUTOMATIC1111
parent 89f4bb3ca1
commit fbdec2ef20
2 changed files with 2 additions and 1 deletions

View File

@ -176,7 +176,7 @@ def process_images(p: StableDiffusionProcessing) -> Processed:
all_prompts = p.batch_size * p.n_iter * [p.prompt] all_prompts = p.batch_size * p.n_iter * [p.prompt]
if type(p.seed) == list: if type(p.seed) == list:
all_seeds = int(p.seed) all_seeds = p.seed
else: else:
all_seeds = [int(p.seed + x) for x in range(len(all_prompts))] all_seeds = [int(p.seed + x) for x in range(len(all_prompts))]

View File

@ -51,6 +51,7 @@ class Script(scripts.Script):
def run(self, p, put_at_start): def run(self, p, put_at_start):
modules.processing.fix_seed(p) modules.processing.fix_seed(p)
seed = p.seed
original_prompt = p.prompt[0] if type(p.prompt) == list else p.prompt original_prompt = p.prompt[0] if type(p.prompt) == list else p.prompt