From 835a7dbf0e73c4cdf945b588d319a6c36652cbe5 Mon Sep 17 00:00:00 2001
From: AUTOMATIC1111 <16777216c@gmail.com>
Date: Wed, 26 Jul 2023 07:49:57 +0300
Subject: simplify PostprocessBatchListArgs

---
 modules/processing.py | 15 +++------------
 1 file changed, 3 insertions(+), 12 deletions(-)

(limited to 'modules/processing.py')

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)
-- 
cgit v1.2.1