michael/stable-diffusion-webui
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Fix prompt matrix script

Browse source
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
Show stats Download patch file Download diff file Expand all files Collapse all files

2
modules/processing.py
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))]

1
scripts/prompt_matrix.py
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