simplify PostprocessBatchListArgs

1 files changed, 3 insertions, 12 deletions

diff --git a/modules/processing.py b/modules/processing.py
index e9108f11..b0992ee1 100644
--- a/modules/processing.py
+++ b/modules/processing.py
@@ -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)