mirrors/stable-diffusion-webui
1
Fork 0
mirror of https://github.com/AUTOMATIC1111/stable-diffusion-webui.git synced 2025-12-27 20:32:35 -08:00
Code Issues Projects Releases Packages Wiki Activity

simplify PostprocessBatchListArgs

Browse source
This commit is contained in:
AUTOMATIC1111 2023-07-26 07:49:57 +03:00
parent 7c22bbd3ad
commit 835a7dbf0e
2 changed files with 9 additions and 22 deletions
Download patch file Download diff file Expand all files Collapse all files

15
modules/processing.py
View file

@ -807,21 +807,12 @@ def process_images_inner(p: StableDiffusionProcessing) -> Processed:
if p.scripts is not None:
p.scripts.postprocess_batch(p, x_samples_ddim, batch_number=n)
batch_params = scripts.PostprocessBatchListArgs(
list(x_samples_ddim),
p.all_prompts[n * p.batch_size:(n + 1) * p.batch_size],
p.all_negative_prompts[n * p.batch_size:(n + 1) * p.batch_size],
p.seeds,
p.subseeds,
)
p.prompts = p.all_prompts[n * p.batch_size:(n + 1) * p.batch_size]
p.negative_prompts = p.all_negative_prompts[n * p.batch_size:(n + 1) * p.batch_size]
batch_params = scripts.PostprocessBatchListArgs(list(x_samples_ddim))
p.scripts.postprocess_batch_list(p, batch_params, batch_number=n)
x_samples_ddim = batch_params.images
p.prompts = batch_params.prompts
p.negative_prompts = batch_params.negative_prompts
p.seeds = batch_params.seeds
p.subseeds = batch_params.subseeds
def infotext(index=0, use_main_prompt=False):
return create_infotext(p, p.prompts, p.seeds, p.subseeds, use_main_prompt=use_main_prompt, index=index, all_negative_prompts=p.negative_prompts)